import fundz models

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 ###

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

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 "<PENumber(number='{}', description='{}')>".format(
+            self.number, self.description)

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)