diff --git a/alembic/versions/55ba973d08b9_add_fundz_models.py b/alembic/versions/55ba973d08b9_add_fundz_models.py new file mode 100644 index 00000000..097532f4 --- /dev/null +++ b/alembic/versions/55ba973d08b9_add_fundz_models.py @@ -0,0 +1,38 @@ +"""add_fundz_models + +Revision ID: 55ba973d08b9 +Revises: 4ea5917e7781 +Create Date: 2018-07-30 14:43:34.099799 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = '55ba973d08b9' +down_revision = '4ea5917e7781' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.create_table('pe_number', + sa.Column('number', sa.String(), nullable=False), + sa.Column('description', sa.String(), nullable=True), + sa.PrimaryKeyConstraint('number') + ) + op.create_table('task_order', + sa.Column('id', sa.Integer(), nullable=False), + sa.Column('number', sa.String(), nullable=True), + sa.PrimaryKeyConstraint('id') + ) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.drop_table('task_order') + op.drop_table('pe_number') + # ### end Alembic commands ### diff --git a/atst/models/__init__.py b/atst/models/__init__.py index fef82003..1d6daae1 100644 --- a/atst/models/__init__.py +++ b/atst/models/__init__.py @@ -8,3 +8,5 @@ from .permissions import Permissions from .role import Role from .user import User from .workspace_role import WorkspaceRole +from .pe_number import PENumber +from .task_order import TaskOrder diff --git a/atst/models/pe_number.py b/atst/models/pe_number.py new file mode 100644 index 00000000..61a0755e --- /dev/null +++ b/atst/models/pe_number.py @@ -0,0 +1,13 @@ +from sqlalchemy import String, Column + +from atst.models import Base + +class PENumber(Base): + __tablename__ = "pe_number" + + number = Column(String, primary_key=True) + description = Column(String) + + def __repr__(self): + return "".format( + self.number, self.description) diff --git a/atst/models/task_order.py b/atst/models/task_order.py new file mode 100644 index 00000000..7a7879a7 --- /dev/null +++ b/atst/models/task_order.py @@ -0,0 +1,9 @@ +from sqlalchemy import Column, Integer, String + +from atst.models import Base + +class TaskOrder(Base): + __tablename__ = "task_order" + + id = Column(Integer, primary_key=True) + number = Column(String)