TaskOrders domain responsible for adding officers

atst/domain/task_orders.py

@@ -2,8 +2,10 @@ from sqlalchemy.orm.exc import NoResultFound
 
 from atst.database import db
 from atst.models.task_order import TaskOrder
+from atst.domain.workspaces import Workspaces
 from .exceptions import NotFoundError
 
+
 class TaskOrderError(Exception):
     pass
 
@@ -93,16 +95,25 @@ class TaskOrders(object):
 
         return True
 
-    OFFICERS = ["contracting_officer", "contracting_officer_representative", "security_officer"]
+    OFFICERS = [
+        "contracting_officer",
+        "contracting_officer_representative",
+        "security_officer",
+    ]
 
     @classmethod
-    def add_officer(cls, task_order, user, role):
+    def add_officer(cls, user, task_order, role, officer_data):
         if role in TaskOrders.OFFICERS:
-            setattr(task_order, role, user)
+            member = Workspaces.create_member(
+                user, task_order.workspace, {**officer_data, "workspace_role": role}
+            )
+            setattr(task_order, role, member.user)
 
             db.session.add(task_order)
             db.session.commit()
 
-            return task_order
+            return member.user
         else:
-            raise TaskOrderError("{} is not an officer role on task orders".format(role))
+            raise TaskOrderError(
+                "{} is not an officer role on task orders".format(role)
+            )

atst/routes/task_orders/new.py

@@ -10,6 +10,7 @@ from flask import (
 from . import task_orders_bp
 from atst.domain.task_orders import TaskOrders
 from atst.domain.workspaces import Workspaces
+from atst.domain.workspace_roles import WorkspaceRoles
 import atst.forms.task_order as task_order_form
 from atst.services.invitation import Invitation as InvitationService
 
@@ -167,17 +168,18 @@ class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow):
                     field: getattr(self.task_order, prefix + "_" + field)
                     for field in ["first_name", "last_name", "email", "dod_id"]
                 }
+                officer = TaskOrders.add_officer(
+                    self.user, self.task_order, officer_type["role"], officer_data
+                )
+                ws_officer_member = WorkspaceRoles.get(self.workspace.id, officer.id)
                 invite_service = InvitationService(
                     self.user,
-                    self.workspace,
-                    {**officer_data, "workspace_role": officer_type["role"]},
+                    ws_officer_member,
+                    officer_data["email"],
                     subject=officer_type["subject"],
                     email_template=officer_type["template"],
                 )
-                invite = invite_service.invite()
-                TaskOrders.add_officer(
-                    self.task_order, invite.user, officer_type["role"]
-                )
+                invite_service.invite()
 
     def update(self):
         self._update_task_order()

atst/routes/workspaces/members.py

@@ -70,7 +70,10 @@ def create_member(workspace_id):
 
     if form.validate():
         try:
-            invite_service = InvitationService(g.current_user, workspace, form.data)
+            member = Workspaces.create_member(g.current_user, workspace, form.data)
+            invite_service = InvitationService(
+                g.current_user, member, form.data.get("email")
+            )
             invite_service.invite()
 
             flash("new_workspace_member", new_member=new_member, workspace=workspace)

atst/services/invitation.py

@@ -1,6 +1,5 @@
 from flask import render_template
 
-from atst.domain.workspaces import Workspaces
 from atst.domain.invitations import Invitations
 from atst.queue import queue
 
@@ -9,33 +8,28 @@ class Invitation:
     def __init__(
         self,
         inviter,
-        workspace,
-        user_info,
+        member,
+        email,
         subject="{} has invited you to a JEDI Cloud Workspace",
         email_template="emails/invitation.txt",
     ):
         self.inviter = inviter
-        self.workspace = workspace
-        self.user_info = user_info
+        self.member = member
+        self.email = email
         self.subject = subject
         self.email_template = email_template
 
     def invite(self):
-        member = self._create_member()
-        email = self.user_info.get("email")
-        invite = self._create_invite(member, email)
-        self._send_invite_email(invite.token, email)
+        invite = self._create_invite()
+        self._send_invite_email(invite.token)
 
         return invite
 
-    def _create_member(self):
-        return Workspaces.create_member(self.inviter, self.workspace, self.user_info)
+    def _create_invite(self):
+        return Invitations.create(self.inviter, self.member, self.email)
 
-    def _create_invite(self, member, email):
-        return Invitations.create(self.inviter, member, email)
-
-    def _send_invite_email(self, token, email):
+    def _send_invite_email(self, token):
         body = render_template(
             self.email_template, owner=self.inviter.full_name, token=token
         )
-        queue.send_mail([email], self.subject.format(self.inviter), body)
+        queue.send_mail([self.email], self.subject.format(self.inviter), body)