Rename resolve_permission_ct() to resolve_permission_type()

netbox/utilities/permissions.py

@@ -7,7 +7,7 @@ __all__ = (
     'permission_is_exempt',
     'qs_filter_from_constraints',
     'resolve_permission',
-    'resolve_permission_ct',
+    'resolve_permission_type',
 )
 
 
@@ -42,9 +42,9 @@ def resolve_permission(name):
     return app_label, action, model_name
 
 
-def resolve_permission_ct(name):
+def resolve_permission_type(name):
     """
-    Given a permission name, return the relevant ContentType and action. For example, "dcim.view_site" returns
+    Given a permission name, return the relevant ObjectType and action. For example, "dcim.view_site" returns
     (Site, "view").
 
     :param name: Permission name in the format <app_label>.<action>_<model>
@@ -52,7 +52,7 @@ def resolve_permission_ct(name):
     from core.models import ObjectType
     app_label, action, model_name = resolve_permission(name)
     try:
-        object_type = ObjectType.objects.get(app_label=app_label, model=model_name)
+        object_type = ObjectType.objects.get_by_natural_key(app_label=app_label, model=model_name)
     except ObjectType.DoesNotExist:
         raise ValueError(_("Unknown app_label/model_name for {name}").format(name=name))
 

netbox/utilities/testing/base.py

@@ -12,7 +12,7 @@ from taggit.managers import TaggableManager
 
 from core.models import ObjectType
 from users.models import ObjectPermission
-from utilities.permissions import resolve_permission_ct
+from utilities.permissions import resolve_permission_type
 from utilities.utils import content_type_identifier
 from .utils import extract_form_failures
 
@@ -44,11 +44,11 @@ class TestCase(_TestCase):
         Assign a set of permissions to the test user. Accepts permission names in the form <app>.<action>_<model>.
         """
         for name in names:
-            ct, action = resolve_permission_ct(name)
+            object_type, action = resolve_permission_type(name)
             obj_perm = ObjectPermission(name=name, actions=[action])
             obj_perm.save()
             obj_perm.users.add(self.user)
-            obj_perm.object_types.add(ct)
+            obj_perm.object_types.add(object_type)
 
     #
     # Custom assertions