Convert Users, Roles, WorkspaceUsers, TaskOrders to use classmethods

This commit is contained in:
richard-dds
2018-08-03 13:12:54 -04:00
committed by dandds
parent 7b5d76e260
commit b72a16569f
8 changed files with 86 additions and 107 deletions

View File

@@ -1,17 +1,17 @@
from sqlalchemy.orm.exc import NoResultFound
from atst.database import db
from atst.models.task_order import TaskOrder
from .exceptions import NotFoundError
class TaskOrders(object):
def __init__(self, db_session):
self.db_session = db_session
@classmethod
def get(self, order_number):
try:
task_order = (
self.db_session.query(TaskOrder).filter_by(number=order_number).one()
db.session.query(TaskOrder).filter_by(number=order_number).one()
)
except NoResultFound:
raise NotFoundError("task_order")