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

Add test for CustomFieldManager.get_for_model()

Browse Source
This commit is contained in:
Jeremy Stretch
2020-05-08 12:18:08 -04:00
parent e3be5f8468
commit 745c9a9c2b
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
netbox/extras/models/customfields.py
View File

@@ -62,13 +62,8 @@ class CustomFieldManager(models.Manager):
"""
Return all CustomFields assigned to the given model.
"""
model = model._meta.concrete_model
# Fetch from the database if the model's CustomFields have not been cached
content_type = ContentType.objects.get_for_model(model)
customfields = CustomField.objects.filter(obj_type=content_type)
return customfields
content_type = ContentType.objects.get_for_model(model._meta.concrete_model)
return self.get_queryset().filter(obj_type=content_type)
class CustomField(models.Model):