def gen_class()

in uberpoet/filegen.py [0:0]

• 15 lines of code
• 2 McCabe index (conditional complexity)

    def gen_class(self, class_count, func_per_class_count, import_list):
        out = []
        class_nums = {}

        for _ in xrange(class_count):
            num = seed()
            swift_only_func_out, swift_only_func_nums = self.gen_func(func_per_class_count, "x", indent=4)
            swift_objc_friendly_func_out, swift_objc_friendly_func_nums = self.gen_objc_friendly_func(indent=4)
            func_out = swift_only_func_out + "\n" + swift_objc_friendly_func_out
            func_call_out = get_import_func_calls(self.language(), import_list, indent=8)
            out.append(swift_class_template.format(num, func_out, func_call_out))

            class_nums[num] = {
                FuncType.SWIFT_ONLY: swift_only_func_nums,
                FuncType.OBJC_FRIENDLY: swift_objc_friendly_func_nums
            }

        return "\n".join(out), class_nums