fix: runtime exception for module duplicate uuid

Author: nampereira

arts-main/pubsub/handlers.py

@@ -1,6 +1,7 @@
 """MQTT message handlers."""
 
 import uuid
+from django.db import IntegrityError
 
 from arts_core.models import Runtime, Module, File, FileType
 from arts_core.serializers import ModuleSerializer, RuntimeSerializer
@@ -108,7 +109,13 @@ def __create_module(self, msg, send_wasm=False):
             module = self.__object_from_dict(Module, data)
             module.source = self.__create_or_get_file(msg)
             module.parent = self.__get_runtime_or_schedule(msg, module)
-            module.save()
+
+            try:
+                module.save()
+            except IntegrityError as e:
+                if 'UNIQUE constraint' in str(e):
+                    raise messages.DuplicateUUID(data, obj_type='module')
+
             self.profiler.register_module(
                 msg.get('data', 'uuid'), msg.get('data', 'name'))
 

arts-main/pubsub/messages.py

@@ -100,6 +100,12 @@ def __init__(self, obj, obj_type="runtime"):
         super().__init__(
             {"desc": "invalid {} UUID".format(obj_type), "data": obj})
 
+class DuplicateUUID(ARTSException):
+    """Exception for runtime/module when trying to create a duplicate UUID (expected in per-scene instantiated modules)."""
+
+    def __init__(self, obj, obj_type="runtime"):
+        super().__init__(
+            {"desc": "duplicate {} UUID; request ignored".format(obj_type), "data": obj})
 
 class InvalidArgument(ARTSException):
     """Exception for invalid argument value."""