mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

model_to_dict(): Remove fields that start with an underscore

Browse Source
This commit is contained in:
Jeremy Stretch
2020-01-31 12:14:51 -05:00
parent 6a17be740b
commit a208cbdf0b
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
netbox/utilities/testing/utils.py
View File

@ -9,6 +9,7 @@ def model_to_dict(instance, fields=None, exclude=None):
""" """
Customized wrapper for Django's built-in model_to_dict(). Does the following: Customized wrapper for Django's built-in model_to_dict(). Does the following:
- Excludes the instance ID field - Excludes the instance ID field
- Exclude any fields prepended with an underscore
- Convert any assigned tags to a comma-separated string - Convert any assigned tags to a comma-separated string
""" """
_exclude = ['id'] _exclude = ['id']
@ -17,6 +18,10 @@ def model_to_dict(instance, fields=None, exclude=None):
model_dict = _model_to_dict(instance, fields=fields, exclude=_exclude) model_dict = _model_to_dict(instance, fields=fields, exclude=_exclude)
for key in list(model_dict.keys()):
if key.startswith('_'):
del model_dict[key]
if 'tags' in model_dict: if 'tags' in model_dict:
model_dict['tags'] = ','.join(sorted([tag.name for tag in model_dict['tags']])) model_dict['tags'] = ','.join(sorted([tag.name for tag in model_dict['tags']]))