Create contenttypes entry on CustomObjectType save

Author: bctiemann

netbox_custom_objects/forms.py

@@ -70,7 +70,7 @@ def object_type_name(self, object_type, include_app=True):
         except AttributeError:
             if object_type.app_label == 'netbox_custom_objects':
                 try:
-                    model_name = self.custom_object_types.get(slug=object_type.model).name
+                    model_name = self.custom_object_types.get(name__iexact=object_type.model).name
                     return f'Custom Objects > {model_name} (Custom)'
                 except CustomObjectType.DoesNotExist:
                     pass

netbox_custom_objects/models.py

@@ -65,6 +65,10 @@ class Meta:
     def __str__(self):
         return self.name
 
+    def save(self, *args, **kwargs):
+        super().save(*args, **kwargs)
+        content_type, _ = ContentType.objects.get_or_create(app_label='netbox_custom_objects', model=self.name.lower())
+
     @property
     def formatted_schema(self):
         result = '<ul>'
@@ -632,7 +636,7 @@ def deserialize(self, value):
     def related_custom_object_type(self):
         if self.related_object_type.app_label != 'netbox_custom_objects':
             return None
-        return CustomObjectType.objects.get(slug=self.related_object_type.model)
+        return CustomObjectType.objects.get(name__iexact=self.related_object_type.model)
 
     @property
     def related_object_queryset(self):