From 070ccec23f08ddae903bb5cb11045628b4d00271 Mon Sep 17 00:00:00 2001 From: leigh-mil Date: Fri, 15 Feb 2019 14:42:29 -0500 Subject: [PATCH] Add migration for LOA array type --- .../db161adbafdf_update_loa_to_array_type.py | 28 +++++++++++++++++++ atst/models/task_order.py | 2 +- 2 files changed, 29 insertions(+), 1 deletion(-) create mode 100644 alembic/versions/db161adbafdf_update_loa_to_array_type.py diff --git a/alembic/versions/db161adbafdf_update_loa_to_array_type.py b/alembic/versions/db161adbafdf_update_loa_to_array_type.py new file mode 100644 index 00000000..afe3ae6d --- /dev/null +++ b/alembic/versions/db161adbafdf_update_loa_to_array_type.py @@ -0,0 +1,28 @@ +"""Update LOA to Array Type + +Revision ID: db161adbafdf +Revises: b3a1a07cf30b +Create Date: 2019-02-15 14:28:33.181136 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = 'db161adbafdf' +down_revision = 'b3a1a07cf30b' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.execute("ALTER TABLE task_orders ALTER COLUMN loa TYPE varchar[] USING array[loa]") + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.execute("ALTER TABLE task_orders ALTER COLUMN loa TYPE varchar USING loa[1]") + # ### end Alembic commands ### diff --git a/atst/models/task_order.py b/atst/models/task_order.py index 73e090f9..c265cbe0 100644 --- a/atst/models/task_order.py +++ b/atst/models/task_order.py @@ -90,7 +90,7 @@ class TaskOrder(Base, mixins.TimestampsMixin): pdf_attachment_id = Column(ForeignKey("attachments.id")) _pdf = relationship("Attachment", foreign_keys=[pdf_attachment_id]) number = Column(String, unique=True) # Task Order Number - loa = Column(String) # Line of Accounting (LOA) + loa = Column(ARRAY(String)) # Line of Accounting (LOA) custom_clauses = Column(String) # Custom Clauses signer_dod_id = Column(String) signed_at = Column(DateTime)