Refactor get_object

Author: arthanson

netbox_custom_objects/__init__.py

@@ -52,6 +52,31 @@ class CustomObjectsPluginConfig(PluginConfig):
     required_settings = []
     template_extensions = "template_content.template_extensions"
 
+    def ready(self):
+        from .models import CustomObjectType
+        from netbox_custom_objects.api.serializers import get_serializer_class
+
+        # Suppress warnings about database calls during app initialization
+        with warnings.catch_warnings():
+            warnings.filterwarnings(
+                "ignore", category=RuntimeWarning, message=".*database.*"
+            )
+            warnings.filterwarnings(
+                "ignore", category=UserWarning, message=".*database.*"
+            )
+            
+            # Skip database calls if running during migration or if table doesn't exist
+            if is_running_migration() or not check_custom_object_type_table_exists():
+                super().ready()
+                return
+
+            qs = CustomObjectType.objects.all()
+            for obj in qs:
+                model = obj.get_model()
+                get_serializer_class(model)
+
+        super().ready()
+
     def get_model(self, model_name, require_ready=True):
         if "table" in model_name.lower() and "model" in model_name.lower():
             is_custom_object_model = True

netbox_custom_objects/models.py

@@ -493,7 +493,6 @@ def register_custom_object_search_index(self, model):
     def get_model(
         self,
         fields=None,
-        app_label=None,
         skip_object_fields=False,
         no_cache=False,
         _generating_models=None,
@@ -504,10 +503,6 @@ def get_model(
 
         :param fields: Extra table field instances that need to be added the model.
         :type fields: list
-        :param app_label: In some cases with related fields, the related models must
-            have the same app_label. If passed along in this parameter, then the
-            generated model will use that one instead of generating a unique one.
-        :type app_label: Optional[String]
         :param skip_object_fields: Don't add object or multiobject fields to the model
         :type skip_object_fields: bool
         :param no_cache: Don't cache the generated model or attempt to pull from cache
@@ -517,8 +512,6 @@ def get_model(
         :return: The generated model.
         :rtype: Model
         """
-        if app_label is None:
-            app_label = APP_LABEL
 
         # Check if we have a cached model for this CustomObjectType
         model_name = self.get_table_model_name(self.pk).lower()
@@ -634,10 +627,10 @@ def wrapped_post_through_setup(self, cls):
         from netbox_custom_objects.api.serializers import get_serializer_class
 
         get_serializer_class(model)
-        '''
 
         # Register the global SearchIndex for this model
         self.register_custom_object_search_index(model)
+        '''
 
         # Clean up: remove this model from the set of models being generated
         if _generating_models is not None:

netbox_custom_objects/utilities.py

@@ -8,6 +8,7 @@
     "AppsProxy",
     "generate_model",
     "get_viewname",
+    "get_sub_models",
 )
 
 
@@ -108,3 +109,67 @@ def generate_model(*args, **kwargs):
             apps.clear_cache = apps.clear_cache
 
     return model
+
+
+def get_sub_models(custom_object, visited=None):
+    """
+    Recursively find all related custom object models for a given CustomObject instance.
+    
+    This function traverses through the custom object type's fields and finds all
+    related custom object types that are referenced through OBJECT or MULTIOBJECT fields.
+    
+    Args:
+        custom_object: An instance of a dynamically created CustomObject class
+        visited: Set of custom object type IDs already visited (used for cycle detection)
+        
+    Returns:
+        list: A list of model classes (including the original model) that are related
+              through custom object relationships
+    """
+    from netbox_custom_objects.models import CustomObjectType
+    models = [custom_object]
+    if visited is None:
+        visited = set()
+
+    # Get the custom object type from the instance
+    custom_object_type = custom_object.custom_object_type
+
+    # If we've already visited this type, return empty list to prevent infinite recursion
+    if custom_object_type.id in visited:
+        return []
+
+    # Add this type to visited set
+    visited.add(custom_object_type.id)
+
+    # Get all fields of type OBJECT or MULTIOBJECT
+    # Using string constants to avoid import issues
+    object_fields = custom_object_type.fields.filter(
+        type__in=["object", "multiobject"],
+        related_object_type__isnull=False,
+        related_object_type__app_label=APP_LABEL
+    )
+
+    # For each object field, check if it references another custom object
+    for field in object_fields:
+        related_object_type = field.related_object_type
+
+        # Get the related custom object type
+        related_custom_object_type = CustomObjectType.objects.get(
+            object_type=related_object_type
+        )
+
+        # Get the model for the related custom object type
+        related_model = related_custom_object_type.get_model()
+
+        # Recursively get sub-models from the related type
+        sub_models = get_sub_models(related_model(), visited.copy())
+
+        # Add the related model and its sub-models to our list
+        if related_model not in models:
+            models.append(related_model)
+
+        for sub_model in sub_models:
+            if sub_model not in models:
+                models.append(sub_model)
+
+    return models