def gen_file()

in uberpoet/filegen.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def gen_file(self, class_count, function_count, import_list=None):
        if import_list is None:
            import_list = []
        imports_out = "\n".join(["import {}".format(i if type(i) is str else i.keys()[0]) for i in import_list])
        func_out, func_nums = self.gen_func(function_count, "7")
        class_out, class_nums = self.gen_class(class_count, 5, import_list)

        chunks = [uber_poet_header, imports_out, func_out, class_out]

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