diff --git a/python_modules/marshal.py b/python_modules/marshal.py index a09b614..e9130f3 100644 --- a/python_modules/marshal.py +++ b/python_modules/marshal.py @@ -2,13 +2,14 @@ from . import ptypes from . import codegen import re +import os def write_includes(writer): writer.header.writeln("#include ") writer.header.writeln('#include "common/marshaller.h"') writer.header.newline() if writer.header.has_option("dest_file"): - src = writer.header.options["dest_file"] + src = os.path.basename(writer.header.options["dest_file"]) else: src = "generated_headers.h" src = re.sub(r'(?i)[^a-z0-9]+', '_', src)