new workspace and task order routes

This commit is contained in:
dandds
2018-12-13 16:05:44 -05:00
parent 3ca9d51b04
commit 6d92755a7f
24 changed files with 427 additions and 30 deletions

View File

@@ -0,0 +1,35 @@
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):
@classmethod
def get(cls, task_order_id):
try:
task_order = db.session.query(TaskOrder).filter_by(id=task_order_id).one()
return task_order
except NoResultFound:
raise NotFoundError("task_order")
@classmethod
def create(cls, workspace, creator):
task_order = TaskOrder(workspace=workspace, creator=creator)
db.session.add(task_order)
db.session.commit()
return task_order
@classmethod
def update(cls, task_order, **kwargs):
for key, value in kwargs.items():
setattr(task_order, key, value)
db.session.add(task_order)
db.session.commit()
return task_order