Skip to content

Commit 2cb9ad0

Browse files
committed
NPL-286 add get_action_url override
1 parent ad0472d commit 2cb9ad0

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

netbox_custom_objects/models.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,18 @@ def get_list_url(self):
133133
kwargs={"custom_object_type": self.custom_object_type.name.lower()},
134134
)
135135

136+
@classmethod
137+
def _get_viewname(cls, action=None, rest_api=False):
138+
if rest_api:
139+
return f"plugins-api:netbox_custom_objects:api:customobject_{action}"
140+
return f"plugins:netbox_custom_objects:customobject_{action}"
141+
142+
@classmethod
143+
def get_action_url(cls, action=None, rest_api=False, kwargs=None):
144+
if kwargs:
145+
kwargs["custom_object_type"] = cls.custom_object_type.name.lower()
146+
return reverse(cls._get_viewname(action, rest_api), kwargs=kwargs)
147+
136148

137149
class CustomObjectType(PrimaryModel):
138150
# Class-level cache for generated models

0 commit comments

Comments
 (0)