17
17
)
18
18
19
19
20
- def from_maybe_list (value : Union [Any , List [Any ]]) -> Tuple [type , List [Any ]]:
20
+ def _from_maybe_list (value : Union [Any , List [Any ]]) -> Tuple [type , List [Any ]]:
21
21
return_type = type (value )
22
22
if not isinstance (value , list ):
23
23
value = [value ]
24
24
return (return_type , value )
25
25
26
26
27
- def return_maybe (_type : type , _values : List [Any ]) -> Union [Any , List [Any ]]:
27
+ def _return_maybe (_type : type , _values : List [Any ]) -> Union [Any , List [Any ]]:
28
28
if _type is list :
29
29
return _values
30
30
else :
31
31
return _values [0 ]
32
32
33
33
34
- def override_only_unset (__dict : dict , __target : dict ):
34
+ def _override_only_unset (__dict : dict , __target : dict ):
35
35
unset_fields = set (__target .keys ()) - set (__dict .keys ())
36
36
new_target = dict ()
37
37
for field in unset_fields :
@@ -62,7 +62,7 @@ def __init__(self, *arg, **kwargs):
62
62
'updated_at' : now ,
63
63
self .__records__ : dict (),
64
64
}
65
- self .__dict__ = override_only_unset (self .__dict__ , defaults )
65
+ self .__dict__ = _override_only_unset (self .__dict__ , defaults )
66
66
67
67
def __getitem__ (self , key : str ) -> Any :
68
68
try :
@@ -110,9 +110,9 @@ def metadata(self) -> dict:
110
110
return meta
111
111
112
112
def get (self , key : Union [str , List [str ]], default = None ) -> Union [Any , List [Any ]]:
113
- _type , _keys = from_maybe_list (key )
113
+ _type , _keys = _from_maybe_list (key )
114
114
values = [self .records .get (k , default ) for k in _keys ]
115
- return return_maybe (_type , values )
115
+ return _return_maybe (_type , values )
116
116
117
117
def update (self , mapping : Union [dict , tuple ] = (), ** kwargs ) -> None :
118
118
return self .records .update (mapping , ** kwargs )
@@ -122,8 +122,8 @@ def modify(
122
122
id : Union [str , List [str ]],
123
123
value : Union [Any , List [Any ]]
124
124
) -> None :
125
- type_id , ids = from_maybe_list (id )
126
- type_value , values = from_maybe_list (value )
125
+ type_id , ids = _from_maybe_list (id )
126
+ type_value , values = _from_maybe_list (value )
127
127
if len (ids ) != len (values ):
128
128
raise ValueError ('Can not match ids and values. please check type and length of them' )
129
129
for index in range (len (ids )):
@@ -134,20 +134,20 @@ def modify(
134
134
self .records .update (target )
135
135
136
136
def add (self , item : Union [Any , List [Any ]]) -> Union [str , List [str ]]:
137
- _type , _items = from_maybe_list (item )
137
+ _type , _items = _from_maybe_list (item )
138
138
139
139
ids = []
140
140
for i in _items :
141
141
uid = shortuuid .uuid ()
142
142
self .records [uid ] = i
143
143
ids .append (uid )
144
144
145
- return return_maybe (_type , ids )
145
+ return _return_maybe (_type , ids )
146
146
147
147
def remove (self , key : Union [str , List [str ]]) -> Union [str , List [str ]]:
148
- _type , _keys = from_maybe_list (key )
148
+ _type , _keys = _from_maybe_list (key )
149
149
popped = [self .records .pop (key ) for key in _keys ]
150
- return return_maybe (_type , popped )
150
+ return _return_maybe (_type , popped )
151
151
152
152
def all (self ) -> List [Any ]:
153
153
return self .records .values ()
@@ -163,10 +163,10 @@ def find(self, func: Callable[..., bool]) -> List[str]:
163
163
return ids
164
164
165
165
def has (self , key : Union [str , List [str ]]) -> Union [bool , List [bool ]]:
166
- _type , _keys = from_maybe_list (key )
166
+ _type , _keys = _from_maybe_list (key )
167
167
key_set = set (self .records .keys ())
168
168
values = [k in key_set for k in _keys ]
169
- return return_maybe (_type , values )
169
+ return _return_maybe (_type , values )
170
170
171
171
def count (self ) -> int :
172
172
"""
@@ -213,8 +213,8 @@ async def save(
213
213
keyword arguments for `json.dumps(**kwargs)`.
214
214
Defaults to `(sort_keys=True)`.
215
215
"""
216
- file_kwds = override_only_unset (file_kwds , _defaults_save_file_kwargs )
217
- json_kwds = override_only_unset (json_kwds , _defaults_save_json_kwargs )
216
+ file_kwds = _override_only_unset (file_kwds , _defaults_save_file_kwargs )
217
+ json_kwds = _override_only_unset (json_kwds , _defaults_save_json_kwargs )
218
218
219
219
async with aiofiles .open (self .filepath , ** file_kwds ) as f :
220
220
dict_out = dict (self .__dict__ )
0 commit comments