api-datamanager/Modules/File/FileImpl.py

129 lines
4.9 KiB
Python

from Utils.ObjUtil import SpecObject
from Utils.ValidateUtil import ValidateAttr
class FileObjImpl(object):
"""对象文件创建"""
@staticmethod
def new_file_obj(**kwargs):
"""
新建对象文件
1.根据参数动态创建class类
2.将代码写入.py文件保存
"""
data = kwargs['data']
name = data['class_name']
parameter = data['parameter']
filename = './Modules/File/static/{}.py'.format(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 '新建成功'