def get_module_nodes(self): modules = list( set(self._field_nodes.keys()) | set(self._type_nodes.keys())) result = {} for module in modules: self.add_import_statement(module, "commercetools", "types") self.add_import_statement(module, "marshmallow") type_nodes = reorder_class_definitions(self._type_nodes[module]) global_nodes = [ ast.Assign( targets=[ast.Name(id="__all__")], value=ast.List(elts=[ ast.Str(s=node.name, kind=None) for node in sorted(type_nodes, key=operator.attrgetter("name")) ]), ) ] all_nodes = (self._import_nodes[module] + global_nodes + self._field_nodes[module] + type_nodes) value = ast.Module(body=all_nodes) result[module] = value result["__init__"] = self.generate_init_module(result.keys()) return result
def get_module_nodes(self): modules = list(self._type_nodes.keys()) result = {} for module in modules: self.add_import_statement(module, "attr") self.add_import_statement(module, "typing") self.add_import_statement(module, "datetime") type_nodes = reorder_class_definitions(self._type_nodes[module]) import_nodes = merge_imports(self._import_nodes[module]) global_nodes = [ ast.Assign( targets=[ast.Name(id="__all__")], value=ast.List(elts=[ ast.Str(s=node.name) for node in sorted(type_nodes, key=operator.attrgetter("name")) ]), ) ] all_nodes = (import_nodes + self.get_typing_imports(module) + global_nodes + type_nodes) value = ast.Module(body=all_nodes) result[module] = value result["__init__"] = self.generate_init_module(result.keys()) return result