domain repos for PE numbers and task orders

This commit is contained in:
dandds
2018-07-30 15:50:33 -04:00
parent 56690ed631
commit fc535ea715
5 changed files with 112 additions and 1 deletions

15
atst/domain/pe_numbers.py Normal file
View 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