Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor to a more sensible test.

Browse Source 
reduce cyclomatic complexity of if clause
This commit is contained in:
EspenAlbert
2020-11-24 18:52:51 +00:00
parent 0679bb4158
commit 6f9b32b575
2 changed files with 40 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ormar/models/modelproxy.py
View File

@ -70,14 +70,15 @@ class ModelTableProxy:
f"model without pk set!"
)
model_dict[field] = pk_value
elif isinstance(field_value, list):
targets = [target.get(target_pkname) for target in field_value]
if targets:
model_dict[field] = targets
elif field_value: # nested dict
if isinstance(field_value, list):
model_dict[field] = [
target.get(target_pkname) for target in field_value
]
else:
model_dict.pop(field)
else: # nested dict
model_dict[field] = field_value.get(target_pkname)
model_dict[field] = field_value.get(target_pkname)
else:
model_dict.pop(field)
return model_dict
@classmethod