From 34391c8972380833789e6eed6e1828543b155dac Mon Sep 17 00:00:00 2001 From: Joao Paulo Ramos Date: Fri, 28 Jul 2023 18:07:17 -0300 Subject: [PATCH] fix(RHINENG-1284): Add constrains to AssignmentRule model --- app/models.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/app/models.py b/app/models.py index 7d7c304d39..46faabd69e 100644 --- a/app/models.py +++ b/app/models.py @@ -480,7 +480,11 @@ def __init__( class AssignmentRule(db.Model): __tablename__ = "assignment_rules" - __table_args__ = (Index("idxassrulesorgid", "org_id"),) + __table_args__ = ( + Index("idxassrulesorgid", "org_id"), + UniqueConstraint("org_id", "name", name="assignment_rules_org_id_name_key"), + UniqueConstraint("group_id", name="assignment_rules_unique_group_id"), + ) def __init__( self, @@ -524,7 +528,7 @@ def update(self, input_ar): account = db.Column(db.String(10)) name = db.Column(db.String(255), nullable=False) description = db.Column(db.String(255)) - group_id = db.Column(UUID(as_uuid=True), ForeignKey("groups.id")) + group_id = db.Column(UUID(as_uuid=True), ForeignKey("groups.id"), primary_key=True) filter = db.Column(JSONB, nullable=False) enabled = db.Column(db.Boolean(), default=True) created_on = db.Column(db.DateTime(timezone=True), default=_time_now)