60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
|
|
class SpecObject(object):
|
|
"""自定义类"""
|
|
|
|
fields_map = {}
|
|
|
|
def dict_to_show(self, **kwargs):
|
|
"""显示对象"""
|
|
_dict_ = dict()
|
|
|
|
for key in self.__dict__.keys():
|
|
if type(self.__dict__[key]).__name__ in ['str', 'int', 'float', 'dict', 'bool', 'tuple']:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
elif type(self.__dict__[key]).__name__ == 'list':
|
|
if len(self.__dict__[key]) == 0:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
elif type(self.__dict__[key][0]).__name__ in ['str', 'int', 'float', 'dict', 'bool', 'tuple']:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
else:
|
|
_dict_[self.fields_map[key]] = [item.dict_to_save() for item in self.__dict__[key]]
|
|
elif self.__dict__[key] is None:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
else:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key].dict_to_show()
|
|
|
|
if 'columns' in kwargs:
|
|
_dict_ = {key: _dict_[key] for key in kwargs['columns']}
|
|
|
|
return _dict_
|
|
|
|
def dict_to_save(self, **kwargs):
|
|
"""存储对象"""
|
|
_dict_ = dict()
|
|
|
|
for key in self.__dict__.keys():
|
|
if type(self.__dict__[key]).__name__ in ['str', 'int', 'float', 'dict', 'bool', 'tuple']:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
elif type(self.__dict__[key]).__name__ == 'list':
|
|
if len(self.__dict__[key]) == 0:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
elif type(self.__dict__[key][0]).__name__ in ['str', 'int', 'float', 'dict', 'bool', 'tuple']:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
else:
|
|
_dict_[self.fields_map[key]] = [item.dict_to_save() for item in self.__dict__[key]]
|
|
elif self.__dict__[key] is None:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key]
|
|
else:
|
|
_dict_[self.fields_map[key]] = self.__dict__[key].dict_to_save()
|
|
|
|
if 'columns' in kwargs:
|
|
_dict_ = {key: _dict_[key] for key in kwargs['columns']}
|
|
|
|
return _dict_
|
|
|
|
# def dict_to_post(self, **kwargs):
|
|
# if kwargs.__contains__('data'):
|
|
# fields_map = dict([v, k] for k, v in self.fields_map.items())
|
|
# for field in list(kwargs['data'].keys()):
|
|
# self.__setattr__(fields_map[field], kwargs['data'][field])
|