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

@@ -101,7 +101,7 @@ class Requests(object):
@classmethod
def approve_and_create_workspace(cls, request):
approved_request = Requests.set_status(request, RequestStatus.APPROVED)
workspace = Workspaces.create(approved_request)
workspace = Workspaces.create_from_request(approved_request)
RequestsQuery.add_and_commit(approved_request)

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

View File

@@ -16,7 +16,16 @@ class WorkspaceError(Exception):
class Workspaces(object):
@classmethod
def create(cls, request, name=None):
def create(cls, user, name):
workspace = WorkspacesQuery.create(name=name)
Workspaces._create_workspace_role(
user, workspace, "owner", status=WorkspaceRoleStatus.ACTIVE
)
WorkspacesQuery.add_and_commit(workspace)
return workspace
@classmethod
def create_from_request(cls, request, name=None):
name = name or request.displayname
workspace = WorkspacesQuery.create(request=request, name=name)
Workspaces._create_workspace_role(