Add Semester Enum to Co-Op Table

Author: mbillow

conditional/models/models.py

@@ -191,13 +191,14 @@ class CurrentCoops(db.Model):
     __tablename__ = 'current_coops'
     id = Column(Integer, primary_key=True)
     uid = Column(String(32), nullable=False)
-    active = Column(Boolean, nullable=False)
     date_created = Column(Date, nullable=False)
+    semester = Column(Enum('Fall', 'Spring', name="co_op_enum"), nullable=False)
 
-    def __init__(self, uid):
+    def __init__(self, uid, semester):
         self.uid = uid
         self.active = True
         self.date_created = datetime.now()
+        self.semester = semester
 
 
 class OnFloorStatusAssigned(db.Model):

migrations/versions/2c3193839c9d_.py

@@ -0,0 +1,31 @@
+"""Add Semester Enum to Co-Op Table
+
+Revision ID: 2c3193839c9d
+Revises: 6ae578b76143
+Create Date: 2017-05-24 00:18:22.645256
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '2c3193839c9d'
+down_revision = '6ae578b76143'
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+
+def upgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    semester = postgresql.ENUM('Fall', 'Spring', 'Neither', name='co_op_enum')
+    semester.create(op.get_bind())
+    op.add_column('current_coops', sa.Column('semester', sa.Enum('Fall', 'Spring', 'Neither', name='co_op_enum'), server_default='Neither', nullable=False))
+    op.drop_column('current_coops', 'active')
+    ### end Alembic commands ###
+
+
+def downgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('current_coops', sa.Column('active', sa.BOOLEAN(), server_default=sa.sql.expression.true(), autoincrement=False, nullable=False))
+    op.drop_column('current_coops', 'semester')
+    ### end Alembic commands ###