#338 rework startup logic

Author: arthanson

netbox_custom_objects/__init__.py

@@ -100,20 +100,11 @@ def ready(self):
                 super().ready()
                 return
 
-            try:
-                with transaction.atomic():
-                    qs = CustomObjectType.objects.all()
-                    for obj in qs:
-                        model = obj.get_model()
-                        get_serializer_class(model)
-            except (DatabaseError, OperationalError, ProgrammingError):
-                # Only suppress exceptions during tests when schema may not match model
-                # During normal operation, re-raise to alert of actual problems
-                if "test" in sys.argv:
-                    # The transaction.atomic() block will automatically rollback
-                    pass
-                else:
-                    raise
+            with transaction.atomic():
+                qs = CustomObjectType.objects.all()
+                for obj in qs:
+                    model = obj.get_model()
+                    get_serializer_class(model)
 
         super().ready()
 
@@ -170,27 +161,17 @@ def get_models(self, include_auto_created=False, include_swapped=False):
             # Add custom object type models
             from .models import CustomObjectType
 
-            try:
-                with transaction.atomic():
-                    custom_object_types = CustomObjectType.objects.all()
-                    for custom_type in custom_object_types:
-                        model = custom_type.get_model()
-                        if model:
-                            yield model
-
-                            # If include_auto_created is True, also yield through models
-                            if include_auto_created and hasattr(model, '_through_models'):
-                                for through_model in model._through_models:
-                                    yield through_model
-            except (DatabaseError, OperationalError, ProgrammingError):
-                # Only suppress exceptions during tests when schema may not match model
-                # (e.g., cache_timestamp column doesn't exist yet during test setup)
-                # During normal operation, re-raise to alert of actual problems
-                if "test" in sys.argv:
-                    # The transaction.atomic() block will automatically rollback
-                    pass
-                else:
-                    raise
+            with transaction.atomic():
+                custom_object_types = CustomObjectType.objects.all()
+                for custom_type in custom_object_types:
+                    model = custom_type.get_model()
+                    if model:
+                        yield model
+
+                        # If include_auto_created is True, also yield through models
+                        if include_auto_created and hasattr(model, '_through_models'):
+                            for through_model in model._through_models:
+                                yield through_model
 
 
 config = CustomObjectsPluginConfig