From 2d40c5b64cff8964166344294b4f1c2018e0565e Mon Sep 17 00:00:00 2001
From: kevgliss <kevgliss@gmail.com>
Date: Mon, 13 Sep 2021 12:07:10 -0700
Subject: [PATCH] Ensures new incident roles match all priorities (#1616)

* Ensures new incident roles match all priorities

* removing unneeded import

* Fixing id
---
 .../versions/2021-08-09_b73416df5744.py       | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/dispatch/database/revisions/tenant/versions/2021-08-09_b73416df5744.py b/src/dispatch/database/revisions/tenant/versions/2021-08-09_b73416df5744.py
index 59a073b0dc73..742e208c50f1 100644
--- a/src/dispatch/database/revisions/tenant/versions/2021-08-09_b73416df5744.py
+++ b/src/dispatch/database/revisions/tenant/versions/2021-08-09_b73416df5744.py
@@ -44,6 +44,11 @@ class IncidentType(Base):
     liaison_service = relationship("Service", foreign_keys=[liaison_service_id])
 
 
+class IncidentPriority(Base):
+    __tablename__ = "incident_priority"
+    id = Column(Integer, primary_key=True)
+
+
 assoc_incident_roles_incident_types = Table(
     "incident_role_incident_type",
     Base.metadata,
@@ -52,6 +57,14 @@ class IncidentType(Base):
     PrimaryKeyConstraint("incident_role_id", "incident_type_id"),
 )
 
+assoc_incident_roles_incident_priorities = Table(
+    "incident_role_incident_priorities",
+    Base.metadata,
+    Column("incident_role_id", Integer, ForeignKey("incident_role.id")),
+    Column("incident_priority_id", Integer, ForeignKey("incident_priority.id")),
+    PrimaryKeyConstraint("incident_role_id", "incident_priority_id"),
+)
+
 
 class IncidentRole(Base):
     __tablename__ = "incident_role"
@@ -67,6 +80,9 @@ class IncidentRole(Base):
 
     # Relationships
     incident_types = relationship("IncidentType", secondary=assoc_incident_roles_incident_types)
+    incident_priorities = relationship(
+        "IncidentPriority", secondary=assoc_incident_roles_incident_priorities
+    )
 
     service_id = Column(Integer, ForeignKey("service.id"))
     service = relationship("Service")
@@ -155,12 +171,15 @@ def upgrade():
         if i_type.liaison_service_id:
             roles[(i_type.project_id, i_type.liaison_service_id, "Liaison")].append(i_type)
 
+    incident_priorities = session.query(IncidentPriority).all()
+
     for k, v in roles.items():
         project_id, service_id, role = k
         session.add(
             IncidentRole(
                 project_id=project_id,
                 incident_types=v,
+                incident_priorities=incident_priorities,
                 role=role,
                 service_id=service_id,
             )