From a26947da9de9bc3f143079485c244aba488a889b Mon Sep 17 00:00:00 2001
From: Ketan <73937490+devketanpro@users.noreply.github.com>
Date: Thu, 12 Dec 2024 09:13:10 +0530
Subject: [PATCH] Add Coverage Provider to Embedded Planning and Synchronize
 Coverage Assignee Updates to Assignments [SDCP-875] (#2151)

* Fix : Coverages assignment: Reassigned / unassigned user is not retained on Save / Update

* update schema and implement changes to populate coverage_provider through  event form

* refactore code

* remove unwanted code

* add tests

* refactore logix
---
 client/api/events.ts                               |  2 ++
 server/features/event_embedded_planning.feature    | 14 +++++++++++++-
 server/features/planning.feature                   |  2 +-
 server/planning/events/events_schema.py            |  9 +++++++++
 .../events/events_sync/embedded_planning.py        | 14 +++++++++++++-
 server/planning/planning/planning.py               |  7 +++++++
 server/planning/types/__init__.py                  |  1 +
 7 files changed, 46 insertions(+), 3 deletions(-)

diff --git a/client/api/events.ts b/client/api/events.ts
index 703a6cc3b..178c58ddb 100644
--- a/client/api/events.ts
+++ b/client/api/events.ts
@@ -146,6 +146,7 @@ function create(updates: Partial<IEventItem>): Promise<Array<IEventItem>> {
                 ednote: coverage.planning.ednote,
                 internal_note: coverage.planning.internal_note,
                 headline: coverage.planning.headline,
+                coverage_provider: coverage.assigned_to.coverage_provider
             })),
         })),
         update_method: updates.update_method?.value ?? updates.update_method
@@ -193,6 +194,7 @@ function update(original: IEventItem, updates: Partial<IEventItem>): Promise<Arr
                 ednote: coverage.planning.ednote,
                 internal_note: coverage.planning.internal_note,
                 headline: coverage.planning.headline,
+                coverage_provider: coverage.assigned_to.coverage_provider
             })),
         })),
         update_method: updates.update_method?.value ?? updates.update_method ?? original.update_method
diff --git a/server/features/event_embedded_planning.feature b/server/features/event_embedded_planning.feature
index 0b79ec35d..72dbc8342 100644
--- a/server/features/event_embedded_planning.feature
+++ b/server/features/event_embedded_planning.feature
@@ -18,7 +18,12 @@ Feature: Event Embedded Planning
                 "coverages": [{
                     "g2_content_type": "text",
                     "news_coverage_status": "ncostat:int",
-                    "scheduled": "2029-11-21T15:00:00.000Z"
+                    "scheduled": "2029-11-21T15:00:00.000Z",
+                    "coverage_provider": {
+                        "name":"Stringer",
+                        "qcode":"stringer",
+                        "contact_type":"stringer"
+                    }
                 }]
             }]
         }]
@@ -53,6 +58,13 @@ Feature: Event Embedded Planning
                 "original_creator": "#CONTEXT_USER_ID#",
                 "version_creator": "#CONTEXT_USER_ID#",
                 "workflow_status": "draft",
+                "assigned_to": {
+                    "coverage_provider": {
+                        "name":"Stringer",
+                        "qcode":"stringer",
+                        "contact_type":"stringer"
+                    }
+                },
                 "news_coverage_status": {"qcode": "ncostat:int", "name": "coverage intended", "label": "Planned"},
                 "planning": {
                     "g2_content_type": "text",
diff --git a/server/features/planning.feature b/server/features/planning.feature
index bdd30c513..744a331e8 100644
--- a/server/features/planning.feature
+++ b/server/features/planning.feature
@@ -1976,7 +1976,7 @@ Feature: Planning
                         "slugline": "test slugline"
                     },
                     "assigned_to": {
-                        "desk": "Politic Desk",
+                        "desk": "Sports Desk",
                         "user": "507f191e810c19729de870eb",
                         "assignment_id": "#firstassignment#",
                         "state": "assigned"
diff --git a/server/planning/events/events_schema.py b/server/planning/events/events_schema.py
index 88f547840..15b028232 100644
--- a/server/planning/events/events_schema.py
+++ b/server/planning/events/events_schema.py
@@ -374,6 +374,15 @@
                             "ednote": {"type": "string", "nullable": True},
                             "internal_note": {"type": "string", "nullable": True},
                             "priority": {"type": "integer", "nullable": True},
+                            "coverage_provider": {
+                                "type": "dict",
+                                "nullable": True,
+                                "schema": {
+                                    "qcode": {"type": "string"},
+                                    "name": {"type": "string"},
+                                    "contact_type": {"type": "string"},
+                                },
+                            },
                         },
                     },
                 },
diff --git a/server/planning/events/events_sync/embedded_planning.py b/server/planning/events/events_sync/embedded_planning.py
index 571add070..58fc73846 100644
--- a/server/planning/events/events_sync/embedded_planning.py
+++ b/server/planning/events/events_sync/embedded_planning.py
@@ -150,12 +150,14 @@ def create_new_coverage_from_event_and_planning(
         "planning": {},
     }
 
-    if coverage.get("desk") or coverage.get("user"):
+    if coverage.get("desk") or coverage.get("user") or coverage.get("coverage_provider"):
         new_coverage["assigned_to"] = {}
         if coverage.get("desk"):
             new_coverage["assigned_to"]["desk"] = coverage["desk"]
         if coverage.get("user"):
             new_coverage["assigned_to"]["user"] = coverage["user"]
+        if coverage.get("coverage_provider"):
+            new_coverage["assigned_to"]["coverage_provider"] = coverage["coverage_provider"]
 
     if "language" in profiles.coverages.enabled_fields:
         # If ``language`` is enabled for Coverages but not defined in ``embedded_planning``
@@ -357,6 +359,16 @@ def get_existing_plannings_from_embedded_planning(
             except KeyError:
                 pass
 
+            try:
+                if (
+                    existing_coverage.get("assigned_to", {}).get("coverage_provider")
+                    != embedded_coverage["coverage_provider"]
+                ):
+                    existing_coverage["assigned_to"]["coverage_provider"] = embedded_coverage["coverage_provider"]
+                    update_required = True
+            except KeyError:
+                pass
+
         # Create new Coverages from the ``embedded_planning`` Event field
         for coverage_id, embedded_coverage in embedded_plan["coverages"].items():
             if coverage_id in updated_coverage_ids:
diff --git a/server/planning/planning/planning.py b/server/planning/planning/planning.py
index fac3e8ad2..14de66af6 100644
--- a/server/planning/planning/planning.py
+++ b/server/planning/planning/planning.py
@@ -1011,6 +1011,13 @@ def _create_update_assignment(
             if planning_original.get("name") != planning_updates.get("name"):
                 assignment["name"] = planning["name"] if not translated_value and translated_name else translated_name
 
+            # If the coverage assignee has been changed and workflow status is active
+            if original.get("workflow_status") != WORKFLOW_STATE.DRAFT and self.is_coverage_assignment_modified(
+                updates, original_assignment
+            ):
+                assigned_to["state"] = ASSIGNMENT_WORKFLOW_STATE.ASSIGNED
+                assignment["assigned_to"] = assigned_to
+
             # If there has been a change in the planning internal note then notify the assigned users/desk
             if planning_updates.get("internal_note") and planning_original.get("internal_note") != planning_updates.get(
                 "internal_note"
diff --git a/server/planning/types/__init__.py b/server/planning/types/__init__.py
index 627d16e76..b757c468b 100644
--- a/server/planning/types/__init__.py
+++ b/server/planning/types/__init__.py
@@ -37,6 +37,7 @@ class EmbeddedCoverageItem(TypedDict, total=False):
     ednote: str
     internal_note: str
     priority: int
+    coverage_provider: Dict[str, Any]
 
 
 class EmbeddedPlanning(TypedDict, total=False):