Enable unique constraints for unique fields

piket_server/alembic/versions/6a5989118ee3_enable_unique_constraints.py

@@ -0,0 +1,36 @@
+"""Enable unique constraints
+
+Revision ID: 6a5989118ee3
+Revises: cca57457a0a6
+Create Date: 2019-09-22 17:04:01.945713
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = "6a5989118ee3"
+down_revision = "cca57457a0a6"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    with op.batch_alter_table("consumption_types") as batch_op:
+        batch_op.create_unique_constraint("uc_consumption_types_name", ["name"])
+
+    with op.batch_alter_table("people") as batch_op2:
+        batch_op2.create_unique_constraint("uc_people_aardbei_id", ["aardbei_id"])
+        batch_op2.create_unique_constraint("uc_people_full_name", ["full_name"])
+        batch_op2.create_unique_constraint("uc_people_display_name", ["display_name"])
+
+
+def downgrade():
+    with op.batch_alter_table("people") as batch_op2:
+        batch_op2.drop_constraint("uc_people_display_name", type_="unique")
+        batch_op2.drop_constraint("uc_people_full_name", type_="unique")
+        batch_op2.drop_constraint("uc_people_aardbei_id", type_="unique")
+
+    with op.batch_alter_table("consumption_types") as batch_op:
+        batch_op.drop_constraint("uc_consumption_types_name", type_="unique")

piket_server/models.py

@@ -18,9 +18,9 @@ class Person(db.Model):
     __tablename__ = "people"
 
     person_id = db.Column(db.Integer, primary_key=True)
-    full_name = db.Column(db.String, nullable=False)
-    display_name = db.Column(db.String, nullable=True)
-    aardbei_id = db.Column(db.Integer, nullable=True)
+    full_name = db.Column(db.String, nullable=False, unique=True)
+    display_name = db.Column(db.String, nullable=True, unique=True)
+    aardbei_id = db.Column(db.Integer, nullable=True, unique=True)
     active = db.Column(db.Boolean, nullable=False, default=False)
 
     consumptions = db.relationship("Consumption", backref="person", lazy=True)
@@ -194,7 +194,7 @@ class ConsumptionType(db.Model):
     __tablename__ = "consumption_types"
 
     consumption_type_id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String, nullable=False)
+    name = db.Column(db.String, nullable=False, unique=True)
     icon = db.Column(db.String)
     active = db.Column(db.Boolean, default=True)