From c43efa4b10cc2bb5c040d20451292e4ac3c9980a Mon Sep 17 00:00:00 2001
From: Charlotte Van Petegem <charlotte.vanpetegem@ugent.be>
Date: Fri, 20 May 2022 21:34:18 +0200
Subject: [PATCH] Migration

---
 .../5c8378aa4dff_add_user_associations.py     | 28 +++++++++++++++++++
 app/models/order.py                           |  1 +
 2 files changed, 29 insertions(+)
 create mode 100644 app/migrations/versions/5c8378aa4dff_add_user_associations.py

diff --git a/app/migrations/versions/5c8378aa4dff_add_user_associations.py b/app/migrations/versions/5c8378aa4dff_add_user_associations.py
new file mode 100644
index 0000000..4cc135d
--- /dev/null
+++ b/app/migrations/versions/5c8378aa4dff_add_user_associations.py
@@ -0,0 +1,28 @@
+"""Add user associations
+
+Revision ID: 5c8378aa4dff
+Revises: 55013fe95bea
+Create Date: 2022-05-20 21:32:47.786340
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '5c8378aa4dff'
+down_revision = '55013fe95bea'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('order', sa.Column('association', sa.String(length=120), nullable=True))
+    op.add_column('user', sa.Column('associations', sa.String(length=120), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('user', 'associations')
+    op.drop_column('order', 'association')
+    # ### end Alembic commands ###
diff --git a/app/models/order.py b/app/models/order.py
index 6e5391c..7729d42 100644
--- a/app/models/order.py
+++ b/app/models/order.py
@@ -27,6 +27,7 @@ class Order(db.Model):
     stoptime = db.Column(db.DateTime)
     public = db.Column(db.Boolean, default=True)
     slug = db.Column(db.String(7), default=generate_slug, unique=True)
+    association = db.Column(db.String(120))
 
     items = db.relationship("OrderItem", backref="order", lazy="dynamic")