NPL-386 fix up journal entry

Author: arthanson

netbox_custom_objects/migrations/0006_customobjectbase.py

@@ -0,0 +1,35 @@
+# Generated by Django 5.2.4 on 2025-08-06 16:45
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("netbox_custom_objects", "0005_alter_customobjecttype_description"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="CustomObjectBase",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True, primary_key=True, serialize=False
+                    ),
+                ),
+                (
+                    "custom_object_type",
+                    models.ForeignKey(
+                        blank=True,
+                        editable=False,
+                        null=True,
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to="netbox_custom_objects.customobjecttype",
+                    ),
+                ),
+            ],
+        ),
+    ]

netbox_custom_objects/models.py

@@ -52,6 +52,15 @@
 USER_TABLE_DATABASE_NAME_PREFIX = "custom_objects_"
 
 
+class CustomObjectBase(models.Model):
+    """
+    Base class for all custom object models.
+    """
+    custom_object_type = models.ForeignKey(
+        'CustomObjectType', on_delete=models.CASCADE, editable=False, null=True, blank=True
+    )
+
+
 class CustomObject(
     BookmarksMixin,
     ChangeLoggingMixin,
@@ -134,6 +143,10 @@ def get_list_url(self):
             kwargs={"custom_object_type": self.custom_object_type.name.lower()},
         )
 
+    @classmethod
+    def _get_viewname(cls, action=None, rest_api=False):
+        return f"plugins:netbox_custom_objects:customobject_{action}"
+
 
 class CustomObjectType(PrimaryModel):
     # Class-level cache for generated models
@@ -403,8 +416,6 @@ def get_model(
         attrs = {
             "Meta": meta,
             "__module__": "database.models",
-            "custom_object_type": self,
-            "custom_object_type_id": self.id,
         }
 
         field_attrs = self._fetch_and_generate_field_attrs(fields)
@@ -428,9 +439,10 @@ def wrapped_post_through_setup(self, cls):
         try:
             model = type(
                 str(model_name),
-                (CustomObject, models.Model),
+                (CustomObject, CustomObjectBase),
                 attrs,
             )
+            model.custom_object_type = self
         finally:
             # Restore the original method
             TM.post_through_setup = original_post_through_setup

netbox_custom_objects/tables.py

@@ -178,15 +178,15 @@ class Meta(NetBoxTable.Meta):
             "pk",
             "id",
             "name",
-            "custom_object_type",
+            # "custom_object_type",
             "created",
             "last_updated",
         )
         default_columns = (
             "pk",
             "id",
             "name",
-            "custom_object_type",
+            # "custom_object_type",
             "created",
             "last_updated",
         )

netbox_custom_objects/urls.py

@@ -65,7 +65,7 @@
         name="customobject_delete",
     ),
     path(
-        "<str:custom_object_type>/<int:pk>/journal/",
+        "custom_object/<int:pk>/journal/",
         views.CustomObjectJournalView.as_view(),
         name="customobject_journal",
     ),