get_model in form

Author: arthanson

netbox_custom_objects/views.py

@@ -421,22 +421,26 @@ class CustomObjectEditView(generic.ObjectEditView):
 
     def setup(self, request, *args, **kwargs):
         super().setup(request, *args, **kwargs)
+        model = self.get_model(**kwargs)
         self.object = self.get_object()
-        model = self.object._meta.model
+        # model = self.object._meta.model
         self.form = self.get_form(model)
 
     def get_queryset(self, request):
         model = self.object._meta.model
         return model.objects.all()
 
-    def get_object(self, **kwargs):
-        if self.object:
-            return self.object
+    def get_model(self, **kwargs):
         custom_object_type = self.kwargs.pop("custom_object_type", None)
         object_type = get_object_or_404(
             CustomObjectType, slug=custom_object_type
         )
-        model = object_type.get_model_with_serializer()
+        return object_type.get_model_with_serializer()
+
+    def get_object(self, **kwargs):
+        model = self.get_model(**kwargs)
+        if self.object:
+            return self.object
 
         if not self.kwargs.get("pk", None):
             # We're creating a new object