domain repos for PE numbers and task orders
This commit is contained in:
15
atst/domain/pe_numbers.py
Normal file
15
atst/domain/pe_numbers.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from atst.models.pe_number import PENumber
|
||||
from .exceptions import NotFoundError
|
||||
|
||||
|
||||
class PENumbers(object):
|
||||
|
||||
def __init__(self, db_session):
|
||||
self.db_session = db_session
|
||||
|
||||
def get(self, number):
|
||||
pe_number = self.db_session.query(PENumber).get(number)
|
||||
if not pe_number:
|
||||
raise NotFoundError("pe_number")
|
||||
|
||||
return pe_number
|
18
atst/domain/task_orders.py
Normal file
18
atst/domain/task_orders.py
Normal file
@@ -0,0 +1,18 @@
|
||||
from sqlalchemy.orm.exc import NoResultFound
|
||||
|
||||
from atst.models.task_order import TaskOrder
|
||||
from .exceptions import NotFoundError
|
||||
|
||||
|
||||
class TaskOrders(object):
|
||||
|
||||
def __init__(self, db_session):
|
||||
self.db_session = db_session
|
||||
|
||||
def get(self, order_number):
|
||||
try:
|
||||
task_order = self.db_session.query(TaskOrder).filter_by(number=order_number).one()
|
||||
except NoResultFound:
|
||||
raise NotFoundError("task_order")
|
||||
|
||||
return task_order
|
Reference in New Issue
Block a user