check apps model

Author: arthanson

netbox_custom_objects/field_types.py

@@ -987,11 +987,11 @@ def create_m2m_table(self, instance, model, field_name):
         # Register the model with Django's app registry
         apps = model._meta.apps
 
-        try:
-            through_model = apps.get_model(APP_LABEL, instance.through_model_name)
-        except LookupError:
+        app_models = apps.all_models[APP_LABEL]
+        if instance.through_model_name not in app_models:
             apps.register_model(APP_LABEL, through)
-            through_model = through
+        else:
+            through_model = apps.all_models[APP_LABEL][instance.through_model_name]
 
         # Update the M2M field's through model and target model
         field.remote_field.through = through_model

netbox_custom_objects/models.py

@@ -615,6 +615,8 @@ def wrapped_post_through_setup(self, cls):
         model_name = model._meta.model_name
         if model_name not in app_models:
             apps.register_model(APP_LABEL, model)
+        else:
+            model = apps.all_models[APP_LABEL][model_name]
 
         if not manytomany_models:
             self._after_model_generation(attrs, model)