Cleanup table name resolution

arthanson · arthanson · commit d32128035bf2 · 2025-08-15T14:24:31.000-07:00
diff --git a/netbox_custom_objects/__init__.py b/netbox_custom_objects/__init__.py
@@ -99,9 +99,8 @@ def get_model(self, model_name, require_ready=True):
             )
 
         from .models import CustomObjectType
-        from .utilities import get_custom_object_type_from_content_type
 
-        custom_object_type_id = get_custom_object_type_from_content_type(model_name)
+        custom_object_type_id = model_name.replace("table", "").replace("model", "")
 
         try:
             obj = CustomObjectType.objects.get(pk=custom_object_type_id)

Original file line number	Diff line number	Diff line change
`@@ -99,9 +99,8 @@ def get_model(self, model_name, require_ready=True):`
`99`	`99`	`)`
`100`	`100`
`101`	`101`	`from .models import CustomObjectType`
`102`		`- from .utilities import get_custom_object_type_from_content_type`
`103`	`102`
`104`		`- custom_object_type_id = get_custom_object_type_from_content_type(model_name)`
	`103`	`+ custom_object_type_id = model_name.replace("table", "").replace("model", "")`
`105`	`104`
`106`	`105`	`try:`
`107`	`106`	`obj = CustomObjectType.objects.get(pk=custom_object_type_id)`