move task order concerns into that domain class

This commit is contained in:
dandds
2018-08-23 10:26:43 -04:00
parent 8aa6185c91
commit f32a3eb90d
2 changed files with 45 additions and 37 deletions

View File

@@ -7,6 +7,7 @@ from .exceptions import NotFoundError
class TaskOrders(object):
TASK_ORDER_DATA = [col.name for col in TaskOrder.__table__.c if col.name != "id"]
@classmethod
def get(cls, order_number):
@@ -28,6 +29,7 @@ class TaskOrders(object):
if to_data:
# TODO: we need to determine exactly what we're getting and storing from the EDA client
return TaskOrders.create(number=to_data["contract_no"], source=Source.EDA)
else:
raise NotFoundError("task_order")
@@ -43,3 +45,14 @@ class TaskOrders(object):
@classmethod
def _client(cls):
return app.eda_client
@classmethod
def get_or_create_task_order(cls, number, task_order_data=None):
try:
return TaskOrders.get(number)
except NotFoundError:
if task_order_data:
return TaskOrders.create(
**task_order_data, number=number, source=Source.MANUAL
)