Refactor fv form to separate TO from request
This commit is contained in:
@@ -19,14 +19,14 @@ class TaskOrders(object):
|
||||
)
|
||||
except NoResultFound:
|
||||
if TaskOrders._client():
|
||||
task_order = TaskOrders._get_from_eda(order_number)
|
||||
task_order = TaskOrders.get_from_eda(order_number)
|
||||
else:
|
||||
raise NotFoundError("task_order")
|
||||
|
||||
return task_order
|
||||
|
||||
@classmethod
|
||||
def _get_from_eda(cls, order_number):
|
||||
def get_from_eda(cls, order_number):
|
||||
to_data = TaskOrders._client().get_contract(order_number, status="y")
|
||||
if to_data:
|
||||
# TODO: we need to determine exactly what we're getting and storing from the EDA client
|
||||
@@ -38,9 +38,9 @@ class TaskOrders(object):
|
||||
raise NotFoundError("task_order")
|
||||
|
||||
@classmethod
|
||||
def create(cls, **kwargs):
|
||||
to_data = drop(["source"], kwargs)
|
||||
task_order = TaskOrder(source=Source.MANUAL, **to_data)
|
||||
def create(cls, source=Source.MANUAL, **kwargs):
|
||||
to_data = {k: v for k, v in kwargs.items() if v not in ["", None]}
|
||||
task_order = TaskOrder(source=source, **to_data)
|
||||
|
||||
db.session.add(task_order)
|
||||
db.session.commit()
|
||||
@@ -51,33 +51,6 @@ class TaskOrders(object):
|
||||
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:
|
||||
pdf_file = task_order_data.pop("pdf")
|
||||
# should catch the error here
|
||||
attachment = Attachment.attach(pdf_file)
|
||||
return TaskOrders.create(
|
||||
**task_order_data,
|
||||
number=number,
|
||||
source=Source.MANUAL,
|
||||
pdf=attachment,
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def get_or_create(cls, number, attachment=None, data=None):
|
||||
try:
|
||||
return TaskOrders.get(number)
|
||||
except NotFoundError:
|
||||
data = data or {}
|
||||
return TaskOrders.create(
|
||||
**data, number=number, pdf=attachment, source=Source.MANUAL
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def update(cls, task_order, dct):
|
||||
updated = update_obj(task_order, dct)
|
||||
|
Reference in New Issue
Block a user