single workspace role for task order officers

atst/domain/roles.py

@@ -141,22 +141,10 @@ WORKSPACE_ROLES = [
         ],
     },
     {
-        "name": "contracting_officer",
-        "description": "Can approve funding for a task order. Has view and edit permissions for task orders in a workspace.",
-        "display_name": "Contracting Officer",
-        "permissions": [Permissions.VIEW_WORKSPACE, Permissions.KO_SIGN_TASK_ORDER],
-    },
-    {
-        "name": "contracting_officer_representative",
-        "description": "Assists in submitting task order documents. Can update Task Order.",
-        "display_name": "Contracting Officer Representative",
-        "permissions": [Permissions.VIEW_WORKSPACE],
-    },
-    {
-        "name": "security_officer",
-        "description": "Can edit security questions for a task order.",
-        "display_name": "Security Officer",
-        "permissions": [Permissions.VIEW_WORKSPACE, Permissions.SO_SIGN_TASK_ORDER],
+        "name": "officer",
+        "description": "Officer involved with setting up a Task Order",
+        "display_name": "Task Order Officer",
+        "permissions": [],
     },
 ]
 

atst/domain/task_orders.py

@@ -102,18 +102,34 @@ class TaskOrders(object):
     ]
 
     @classmethod
-    def add_officer(cls, user, task_order, role, officer_data):
-        if role in TaskOrders.OFFICERS:
-            member = Workspaces.create_member(
-                user, task_order.workspace, {**officer_data, "workspace_role": role}
+    def add_officer(cls, user, task_order, officer_type, officer_data):
+        if officer_type in TaskOrders.OFFICERS:
+            workspace = task_order.workspace
+
+            existing_member = next(
+                (
+                    member
+                    for member in workspace.members
+                    if member.user.dod_id is officer_data["dod_id"]
+                ),
+                None,
             )
-            setattr(task_order, role, member.user)
+
+            if existing_member:
+                user = existing_member.user
+            else:
+                member = Workspaces.create_member(
+                    user, workspace, {**officer_data, "workspace_role": "officer"}
+                )
+                user = member.user
+
+            setattr(task_order, officer_type, user)
 
             db.session.add(task_order)
             db.session.commit()
 
-            return member.user
+            return user
         else:
             raise TaskOrderError(
-                "{} is not an officer role on task orders".format(role)
+                "{} is not an officer role on task orders".format(officer_type)
             )