auto add work lead as team lead

epictrack-api/migrations/versions/7a42d4f57279_project_lead_to_team_lead.py

@@ -0,0 +1,24 @@
+"""project lead to team lead
+
+Revision ID: 7a42d4f57279
+Revises: 1e1a2af1605b
+Create Date: 2023-11-24 13:39:01.142953
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '7a42d4f57279'
+down_revision = '1e1a2af1605b'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.execute("Update roles set name='Team Lead' where name='Project Lead'")
+
+
+def downgrade():
+    pass

epictrack-api/src/api/resources/work.py

@@ -170,7 +170,7 @@ def get(work_id):
     @auth.require
     @profiletime
     def post(work_id):
-        """Get all the active staff allocated to the work"""
+        """Add staff member to a work"""
         req.WorkIdPathParameterSchema().load(request.view_args)
         request_json = req.StaffWorkBodyParamSchema().load(API.payload)
         staff = WorkService.create_work_staff(work_id, request_json)

epictrack-api/src/api/services/work.py

@@ -39,6 +39,7 @@
 from api.services.event_template import EventTemplateService
 from api.services.outcome_template import OutcomeTemplateService
 from api.services.phaseservice import PhaseService
+from api.services.code import CodeService
 
 
 class WorkService:  # pylint: disable=too-many-public-methods
@@ -148,6 +149,19 @@ def create_work(cls, payload, commit: bool = True):
             if sort_order == 1:
                 work.current_work_phase_id = work_phase_id
             sort_order = sort_order + 1
+        role_id = CodeService.find_code_values_by_type(
+            "roles",
+            {
+                "name": "Team Lead"
+            }
+        ).get("codes")[0].get("id")
+        WorkService.create_work_staff(
+            work.id,
+            {
+                "staff_id": payload["work_lead_id"],
+                "role_id": role_id, "is_active": True
+            }
+        )
         if commit:
             db.session.commit()
         return work