diff options
Diffstat (limited to 'codegen/codegen.py')
| -rw-r--r-- | codegen/codegen.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/codegen/codegen.py b/codegen/codegen.py index fa4417a..ab9153c 100644 --- a/codegen/codegen.py +++ b/codegen/codegen.py @@ -1374,9 +1374,10 @@ typedef intobjargproc ssizeobjargproc; def write_imports(self): self.fp.write('/* ---------- types from other modules ---------- */\n') - for module, pyname, cname in self.overrides.get_imports(): - self.fp.write('static PyTypeObject *_%s;\n' % cname) - self.fp.write('#define %s (*_%s)\n' % (cname, cname)) + for module, pyname, cname, importing_for in self.overrides.get_imports(): + if importing_for is None or is_registered_object(importing_for): + self.fp.write('static PyTypeObject *_%s;\n' % cname) + self.fp.write('#define %s (*_%s)\n' % (cname, cname)) self.fp.write('\n\n') def write_type_declarations(self): @@ -1485,8 +1486,9 @@ typedef intobjargproc ssizeobjargproc; return bymod = {} - for module, pyname, cname in imports: - bymod.setdefault(module, []).append((pyname, cname)) + for module, pyname, cname, importing_for in imports: + if importing_for is None or is_registered_object(importing_for): + bymod.setdefault(module, []).append((pyname, cname)) self.fp.write(' PyObject *module;\n\n') for module in bymod: self.fp.write( @@ -1625,6 +1627,9 @@ typedef intobjargproc ssizeobjargproc; _objects = {} +def is_registered_object(c_name): + return c_name in _objects + def get_object_by_name(c_name): global _objects return _objects[c_name] |