Fixing associated signal tables (#2999)

Netflix · Feb 15, 2023 · 55fe565 · 55fe565
1 parent f3426d8
commit 55fe565
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/dispatch/signal/models.py b/src/dispatch/signal/models.py
@@ -42,7 +42,11 @@ class RuleMode(DispatchEnum):
 assoc_signal_instance_tags = Table(
     "assoc_signal_instance_tags",
     Base.metadata,
-    Column("signal_instance_id", UUID, ForeignKey("signal_instance.id", ondelete="CASCADE")),
+    Column(
+        "signal_instance_id",
+        UUID(as_uuid=True),
+        ForeignKey("signal_instance.id", ondelete="CASCADE"),
+    ),
     Column("tag_id", Integer, ForeignKey("tag.id", ondelete="CASCADE")),
     PrimaryKeyConstraint("signal_instance_id", "tag_id"),
 )
@@ -58,7 +62,11 @@ class RuleMode(DispatchEnum):
 assoc_signal_instance_entities = Table(
     "assoc_signal_instance_entities",
     Base.metadata,
-    Column("signal_instance_id", UUID, ForeignKey("signal_instance.id", ondelete="CASCADE")),
+    Column(
+        "signal_instance_id",
+        UUID(as_uuid=True),
+        ForeignKey("signal_instance.id", ondelete="CASCADE"),
+    ),
     Column("entity_id", Integer, ForeignKey("entity.id", ondelete="CASCADE")),
     PrimaryKeyConstraint("signal_instance_id", "entity_id"),
 )