def gen_file()

in uberpoet/filegen.py [0:0]

• 13 lines of code
• 5 McCabe index (conditional complexity)

    def gen_file(self, class_count, function_count, import_list=None):
        if import_list is None:
            import_list = []
        imports = []
        for i in import_list:
            if type(i) is str:
                imports.append('#import \"{}\"'.format(i))
            elif type(i) is dict:
                imports.append('@import {};'.format(i.keys()[0]))
        imports_out = "\n".join(imports)
        class_out, class_nums = self.gen_class(class_count, 5, import_list)

        chunks = [uber_poet_header, objc_system_import_template, imports_out, class_out]

        return FileResult("\n".join(chunks), [], class_nums)