change task orders domain class name

This commit is contained in:
dandds
2018-12-12 16:48:18 -05:00
parent e509c34cdd
commit 15d0db01b6
4 changed files with 20 additions and 17 deletions

View File

@@ -7,7 +7,7 @@ from .exceptions import NotFoundError
from atst.utils import update_obj
class TaskOrders(object):
class LegacyTaskOrders(object):
TASK_ORDER_DATA = [
col.name for col in LegacyTaskOrder.__table__.c if col.name != "id"
]
@@ -19,8 +19,8 @@ class TaskOrders(object):
db.session.query(LegacyTaskOrder).filter_by(number=order_number).one()
)
except NoResultFound:
if TaskOrders._client():
legacy_task_order = TaskOrders.get_from_eda(order_number)
if LegacyTaskOrders._client():
legacy_task_order = LegacyTaskOrders.get_from_eda(order_number)
else:
raise NotFoundError("legacy_task_order")
@@ -28,10 +28,10 @@ class TaskOrders(object):
@classmethod
def get_from_eda(cls, order_number):
to_data = TaskOrders._client().get_contract(order_number, status="y")
to_data = LegacyTaskOrders._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
return TaskOrders.create(
return LegacyTaskOrders.create(
source=Source.EDA, funding_type=FundingType.PROC, **to_data
)

View File

@@ -1,6 +1,6 @@
import re
from atst.domain.task_orders import TaskOrders
from atst.domain.legacy_task_orders import LegacyTaskOrders
from atst.domain.pe_numbers import PENumbers
from atst.domain.exceptions import NotFoundError
@@ -63,7 +63,7 @@ class PENumberValidator(object):
class TaskOrderNumberValidator(object):
def validate(self, field):
try:
TaskOrders.get(field.data)
LegacyTaskOrders.get(field.data)
except NotFoundError:
self._apply_error(field)
return False