api-datamanager/Modules/File/FileImpl.py

import os

from Utils.ObjUtil import SpecObject
from Utils.ValidateUtil import ValidateAttr


class FileObjImpl(object):
    """对象文件创建"""

    @staticmethod
    def get_project_abs_path(**kwargs):
        file_abs_path = os.path.abspath(os.path.dirname(__file__))
        project_abs_path = file_abs_path[:file_abs_path.find('wide_rating') + len('wide_rating')]
        name = kwargs['name']
        pdf_relative_path = '/Modules/File/static/{}'.format(name)
        pdf_abs_path = os.path.abspath(project_abs_path + pdf_relative_path)
        return pdf_abs_path

    def new_file_obj(self, **kwargs):
        """
        新建对象文件
        1.根据参数动态创建class类
        2.将代码写入.py文件保存
        """
        data = kwargs['data']

        name = data['class_name']
        file_name = '{}.py'.format(name)
        parameter = data['parameter']
        filename = self.get_project_abs_path(name=file_name)

        with open(filename, 'w', encoding="utf-8") as f:
            f.write('from Utils.ObjUtil import SpecObject')
            f.write('\n')
            f.write('from Utils.ValidateUtil import ValidateAttr, Validate')
            f.write('\n')
            f.write('\n')
            f.write('\n')
            f.write('class {}(SpecObject):'.format(name))
            f.write('\n')

            def handle_fields(**args):
                fields = args['param']
                fields_name = args['class_name']
                level = args['levels']
                kind = args['kind']

                # 判断类型(list)
                if fields['types'] == 'classmethod' and kind == 'list':
                    list_instance = 'instance_list={}'.format(fields_name)
                else:
                    list_instance = ''

                # 判断类型(dict)
                if len(list_instance) == 0:
                    if fields['types'] == 'classmethod' and kind == 'dict':
                        types = 'type={}'.format(fields_name)
                    else:
                        if fields['types'] == 'date':
                            types = 'func=Validate.date_format'
                        elif fields['types'] == 'time':
                            types = 'func=Validate.time_format'
                        else:
                            types = 'type={}'.format(fields['types'])
                else:
                    types = ''

                # 判断长度
                if 'length' in fields:
                    length = 'length={}, '.format(fields['length'])
                else:
                    length = ''

                # 包含列表
                if 'in_list' in fields:
                    in_list = 'in_list={}, '.format(fields['in_list'])
                else:
                    in_list = ''

                # 类名
                class_names = fields['name']
                if level == 'frist':
                    temp = f"\t{class_names} = ValidateAttr(field='{class_names}', {length}{list_instance}{in_list}{types})"
                else:
                    temp = f"\t\t{class_names} = ValidateAttr(field='{class_names}', {length}{list_instance}{in_list}{types})"
                f.write(temp)
                f.write('\n')

            # 写入子类
            def hanlde_special(temp_data, temp_type):
                f.write('\n')
                f.write('\tclass {}(SpecObject):'.format(temp_data['class_name']))
                f.write('\n')
                temp_dict = dict()
                for list_param in temp_data['parameter']:
                    # fields_map映射数据
                    temp_dict[list_param['name']] = list_param['chinese']
                    # 写入每个fields
                    handle_fields(param=list_param, class_name=temp_data['class_name'], levels='second', kind=temp_type)
                temp_fields = '\t\tfields_map = {}'.format(str(temp_dict))
                f.write('\n')
                f.write(temp_fields)
                f.write('\n')

            # 第一次遍历，创建参数中的子类对象
            for param in parameter:
                # 为instance_list新建对象
                if 'instance_list' in param:
                    instance_list = param['instance_list']
                    hanlde_special(instance_list, 'list')
                # 为instance_dict新建对象
                if 'instance_dict' in param:
                    instance_dict = param['instance_dict']
                    hanlde_special(instance_dict, 'dict')

            f.write('\n')

            first_dict = dict()
            for param in parameter:
                first_dict[param['name']] = param['chinese']
                class_name = None
                temp_type = None
                if 'instance_list' in param:
                    instance_list = param['instance_list']
                    class_name = instance_list['class_name']
                    temp_type = 'list'

                if 'instance_dict' in param:
                    instance_dict = param['instance_dict']
                    class_name = instance_dict['class_name']
                    temp_type = 'dict'

                handle_fields(param=param, class_name=class_name, levels='frist', kind=temp_type)
            f.write('\n')
            frist_fields = '\tfields_map = {}'.format(str(first_dict))
            f.write(frist_fields)
            f.write('\n')
            return '新建成功'