request can update financial verification data
This commit is contained in:
@@ -2,7 +2,7 @@ from sqlalchemy.orm.exc import NoResultFound
|
||||
from flask import current_app as app
|
||||
|
||||
from atst.database import db
|
||||
from atst.models.task_order import TaskOrder
|
||||
from atst.models.task_order import TaskOrder, Source
|
||||
from .exceptions import NotFoundError
|
||||
|
||||
|
||||
@@ -26,13 +26,14 @@ class TaskOrders(object):
|
||||
def _get_from_eda(cls, order_number):
|
||||
to_data = TaskOrders._client().get_contract(order_number, status="y")
|
||||
if to_data:
|
||||
return TaskOrders.create(to_data["contract_no"])
|
||||
# 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")
|
||||
|
||||
@classmethod
|
||||
def create(cls, order_number):
|
||||
task_order = TaskOrder(number=order_number)
|
||||
def create(cls, **kwargs):
|
||||
task_order = TaskOrder(**kwargs)
|
||||
|
||||
db.session.add(task_order)
|
||||
db.session.commit()
|
||||
|
Reference in New Issue
Block a user