pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

migration fix

Browse Source
This commit is contained in:
leigh-mil 2019-12-12 11:58:59 -05:00
parent 6446b4fbd0
commit 9caef2883a
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
alembic/versions/3bd8552f1c57_add_unique_constraint_to_task_order_.py
View File

@ -18,11 +18,11 @@ depends_on = None
def upgrade(): def upgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
op.create_unique_constraint(None, 'task_orders', ['number']) op.create_unique_constraint('task_orders_number_key', 'task_orders', ['number'])
# ### end Alembic commands ### # ### end Alembic commands ###
def downgrade(): def downgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
op.drop_constraint(None, 'task_orders', type_='unique') op.drop_constraint('task_orders_number_key', 'task_orders', type_='unique')
# ### end Alembic commands ### # ### end Alembic commands ###

5
atst/domain/task_orders.py
View File

@ -14,11 +14,13 @@ class TaskOrders(BaseDomainClass):
@classmethod @classmethod
def create(cls, portfolio_id, number, clins, pdf): def create(cls, portfolio_id, number, clins, pdf):
try:
task_order = TaskOrder(portfolio_id=portfolio_id, number=number, pdf=pdf) task_order = TaskOrder(portfolio_id=portfolio_id, number=number, pdf=pdf)
db.session.add(task_order) db.session.add(task_order)
try:
db.session.commit() db.session.commit()
except IntegrityError: except IntegrityError:
db.session.rollback()
raise AlreadyExistsError("task_order") raise AlreadyExistsError("task_order")
TaskOrders.create_clins(task_order.id, clins) TaskOrders.create_clins(task_order.id, clins)
@ -43,6 +45,7 @@ class TaskOrders(BaseDomainClass):
try: try:
db.session.commit() db.session.commit()
except IntegrityError: except IntegrityError:
db.session.rollback()
raise AlreadyExistsError("task_order") raise AlreadyExistsError("task_order")
return task_order return task_order