NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-25 18:08:38 -06:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Extend has_feature() to accept a model or OT/CT

Browse Source
This commit is contained in:
Jeremy Stretch 2025-07-24 09:29:27 -04:00
parent 9f2ef2b909
commit 943f98e86d
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/netbox/models/features.py
View File

@ -654,14 +654,19 @@ def get_model_features(model):
]
def has_feature(model, feature):
def has_feature(model_or_ct, feature):
"""
Returns True if the model supports the specified feature.
"""
# Resolve a ContentType to its model class
if type(model) is ContentType:
model = model.model_class()
ot = ObjectType.objects.get_for_model(model)
# If an ObjectType was passed, we can use it directly
if type(model_or_ct) is ObjectType:
ot = model_or_ct
# If a ContentType was passed, resolve its model class
elif type(model_or_ct) is ContentType:
ot = ObjectType.objects.get_for_model(model_or_ct.model_class())
# For anything else, look up the ObjectType
else:
ot = ObjectType.objects.get_for_model(model_or_ct)
return feature in ot.features