py.lib

Yohn Y. 2022-08-13 Parent:4186c3b229fa Child:84b54a8a6d4c

33:57f63bf31fd8 Go to Latest

py.lib/dataclass_utils.py

* С разбегу влетел в type hinting, пришлось обкладывать костылями. И этот процесс судя по всему длительный, и место больное будет долго...

Download raw file
View source Diff to previous Annotate
History
1 # coding: utf-8
2
3 from dataclasses import fields, dataclass, is_dataclass
4 from typing import Union, Dict, Any
5
6
7 def _dict_has(obj: Dict[str, Any], key: str) -> bool:
8 return key in obj
9
10
11 def _dict_get(obj: Dict[str, Any], key: str) -> Any:
12 return obj[key]
13
14
15 def _obj_has(obj: object, key: str) -> bool:
16 return hasattr(obj, key)
17
18
19 def _obj_get(obj: object, key: str) -> Any:
20 return getattr(obj, key)
21
22
23 def difficult_type_recognizer(type_obj, value):
24 """\
25 Магия борющаяся с костылями type hinting
26 """
27 if type(type_obj).__name__ == '_GenericAlias':
28 if (
29 hasattr(type_obj, '__args__')
30 and type(type_obj.__args__) == tuple
31 and type_obj.__args__
32 ):
33 if type_obj.__name__ == 'List':
34 return list(map(type_obj.__args__[0], value))
35
36 elif type_obj.__name__ == 'Tuple':
37 return tuple(map(type_obj.__args__[0], value))
38
39 else:
40 raise ValueError('Неизвестный тип')
41
42 else:
43 ValueError('Неизвестный тип')
44
45 elif type(type_obj).__name__ == '_UnionGenericAlias':
46 if type_obj.__name__ not in ('Union', 'Optional'):
47 raise TypeError(f'Неизвестный подтип _UnionGenericAlias: {type_obj.__name__}')
48
49 if not (
50 hasattr(type_obj, '__args__')
51 and type(type_obj.__args__) == tuple
52 and type_obj.__args__
53 ):
54 raise TypeError(f'Не ясно как работать с типом не вижу аргументов: {type_obj.__name__}')
55
56 for _t in type_obj.__args__:
57 if _t.__name__ == 'NoneType':
58 continue
59
60 try:
61 return _t(value)
62
63 except (TypeError, ValueError):
64 continue
65
66 raise ValueError('Не удалось привести значение к типу')
67
68
69 def dataobj_extract(obj: Union[object, Dict[str, Any]], dataclass_type: type) -> dataclass:
70 """\
71 Извлекает объект данных из предоставленного объекта, путём получения из него
72 указанных в классе данных аттрибутов и поиска их в данном объекте.
73 """
74
75 params = {}
76
77 if isinstance(obj, dict):
78 _has = _dict_has
79 _get = _dict_get
80
81 else:
82 _has = _obj_has
83 _get = _obj_get
84
85 if not is_dataclass(dataclass_type):
86 raise ValueError(f'Не относится к классам данных: {dataclass_type.__name__}')
87
88 for fld in fields(dataclass_type):
89 if _has(obj, fld.name):
90 val = _get(obj, fld.name)
91 if val is not None and not isinstance(val, fld.type):
92 try:
93 val = fld.type(val)
94
95 except (ValueError, TypeError) as e:
96 try:
97 val = difficult_type_recognizer(fld.type, val)
98
99 except (ValueError, TypeError):
100 raise ValueError(f'Аттрибут {fld.name} не может быть получен из значения "{val}"'
101 f' с типом {type(val).__name__} поскольку не может быть преобразован в'
102 f' тип {fld.type.__name__}, заданный в классе данных: {e}')
103
104 params[fld.name] = val
105
106 try:
107 res = dataclass_type(**params)
108
109 except (ValueError, TypeError) as e:
110 _params = ', '.join(map(lambda x: f'{x[0]}="{x[1]}"', params.items()))
111 raise ValueError(f'Не удалось получить объект'
112 f' класс {dataclass_type.__name__}'
113 f' из параметров: {_params}'
114 f' ошибка: {e}')
115
116 return res