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

@@ -25,6 +25,7 @@
 from api.utils import auth, profiletime
 from api.utils.util import cors_preflight
 from api.utils.datetime_helper import get_start_of_day
+from api.services.code import CodeService
 
 
 API = Namespace("works", description="Works")
@@ -74,6 +75,8 @@ def post():
         request_json = req.WorkBodyParameterSchema().load(API.payload)
         request_json["start_date"] = get_start_of_day(request_json["start_date"])
         work = WorkService.create_work(request_json)
+        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": request_json["work_lead_id"], "role_id": role_id, "is_active": True })
         return res.WorkResponseSchema().dump(work), HTTPStatus.CREATED
 
 
@@ -170,7 +173,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)