add total clusters initiated

Author: Ada Ball

apps/report-execution/src/libraries/pa_01.py

@@ -12,6 +12,7 @@
 from src.libraries.support.pa_01.queries import (
     case_interview_rows_query,
     cases_with_no_partners_query,
+    clusters_initiated_query,
     filtered_cases_query,
     partner_notification_query,
     period_partners_query,
@@ -87,6 +88,7 @@ def execute(
         testing_index=trx.query(testing_index_query(subset_query)),
         period_partners=trx.query(period_partners_query(subset_query)),
         cases_with_no_partners=trx.query(cases_with_no_partners_query(subset_query)),
+        clusters_initiated=trx.query(clusters_initiated_query(subset_query))
     )
 
     # build output CSV data for each worker

apps/report-execution/src/libraries/support/pa_01/calculations.py

@@ -201,6 +201,7 @@ def build_partners_and_clusters_initiated_output(
     cases_with_no_partners, cases_with_no_partners_percentage = (
         _calc_cases_with_no_partners(tables.cases_with_no_partners, cases_ixd, worker)
     )
+    total_clusters_initiated = _calc_total_clusters_initiated(tables.clusters_initiated, worker)
 
     # output CSV data
     rows: list[Pa01Row] = [
@@ -258,6 +259,15 @@ def build_partners_and_clusters_initiated_output(
             cases_with_no_partners_percentage,
             None,
         ),
+        (
+            _worker_for_csv(worker),
+            PARTNERS_AND_CLUSTERS_INITIATED,
+            'Total Clusters Initiated',
+            None,
+            total_clusters_initiated,
+            None,
+            None,
+        ),
     ]
 
     return rows
@@ -535,6 +545,15 @@ def _calc_cases_with_no_partners(
     return count, _percent_for_csv(count, cases_ixd)
 
 
+def _calc_total_clusters_initiated(
+    clusters_initiated: Table, worker: Pa01Worker | None = None
+) -> int:
+    """Calculate 'Total Clusters Initiated' count. Calculates for all workers if passed
+    in worker is None."""
+    rows = _rows_for_worker(clusters_initiated, worker)
+    return _count_distinct_case_ids(rows)
+
+
 # helpers
 def _rows_for_worker(table: Table, worker: Pa01Worker | None = None) -> list[dict]:
     """Filter a given table's data for the given worker.  If the given worker is None

apps/report-execution/src/libraries/support/pa_01/models.py

@@ -38,6 +38,7 @@ class Pa01Tables:
     testing_index: Table
     period_partners: Table
     cases_with_no_partners: Table
+    clusters_initiated: Table
 
 
 @dataclass(frozen=True)

apps/report-execution/src/libraries/support/pa_01/queries.py

@@ -413,3 +413,56 @@ def cases_with_no_partners_query(subset_query: str) -> str:
           FROM pp
       );
     """
+
+
+def clusters_initiated_query(subset_query: str) -> str:
+    return f"""
+      WITH base AS
+      (
+        {subset_query}
+      ),
+      filtered_cases AS
+      (
+        -- STD_HIV_DATAMART1 in SAS
+        SELECT b.*
+        FROM base b
+          INNER JOIN RDB.dbo.INVESTIGATION i
+                  ON i.INVESTIGATION_KEY = b.INVESTIGATION_KEY
+                 AND i.INV_CASE_STATUS IN ('Probable', 'Confirmed')
+                 AND b.CA_INTERVIEWER_ASSIGN_DT IS NOT NULL
+      )
+      SELECT DISTINCT a.INVESTIGATION_KEY,
+             PROVIDER_QUICK_CODE,
+             a.inv_local_id AS STD_INV_LOCAL_ID,
+             f.FL_FUP_DISPOSITION,
+             e.LOCAL_ID AS INV_LOCAL_ID,
+             a.FL_FUP_DISPO_DT,
+             a.FL_FUP_INIT_ASSGN_DT,
+             a.INVESTIGATOR_INTERVIEW_KEY,
+             a.INVESTIGATOR_INTERVIEW_QC
+      FROM filtered_cases a
+        INNER JOIN RDB.dbo.F_INTERVIEW_CASE b
+                ON a.INVESTIGATION_KEY = b.INVESTIGATION_KEY
+        INNER JOIN RDB.dbo.D_INTERVIEW c
+                ON c.D_INTERVIEW_KEY = b.D_INTERVIEW_KEY
+               AND c.RECORD_STATUS_CD <> 'LOG_DEL'
+        INNER JOIN RDB.dbo.F_CONTACT_RECORD_CASE d
+                ON a.INVESTIGATION_KEY = d.SUBJECT_INVESTIGATION_KEY
+               AND c.d_interview_key = d.CONTACT_INTERVIEW_KEY
+               AND d.CONTACT_INTERVIEW_KEY <> 1
+        INNER JOIN RDB.dbo.D_CONTACT_RECORD e
+                ON e.D_CONTACT_RECORD_KEY = d.D_CONTACT_RECORD_KEY
+               AND e.RECORD_STATUS_CD <> 'LOG_DEL'
+        INNER JOIN RDB.dbo.D_provider
+                ON D_provider.provider_key = a.INVESTIGATOR_INTERVIEW_KEY
+        INNER JOIN RDB.dbo.STD_HIV_DATAMART f
+                ON d.CONTACT_INVESTIGATION_KEY = f.Investigation_key
+               AND e.CTT_REFERRAL_BASIS IN (
+                 'A1 - Associate 1',
+                 'A2 - Associate 2',
+                 'A3 - Associate 3',
+                 'S1 - Social Contact 1',
+                 'S2 - Social Contact 2',
+                 'S3 - Social Contact 3'
+               );
+    """

apps/report-execution/tests/integration/libraries/snapshots/pa_01/test_execute_report_check_data_hiv/snapshot.yml

@@ -158,6 +158,14 @@
   - 180
   - 97.3%
   - null
+- !!python/tuple
+  - ALL
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 8
+  - null
+  - null
 - !!python/tuple
   - adevraj
   - Case Assignments & Outcomes
@@ -318,6 +326,14 @@
   - 32
   - 94.1%
   - null
+- !!python/tuple
+  - adevraj
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 0
+  - null
+  - null
 - !!python/tuple
   - avance
   - Case Assignments & Outcomes
@@ -478,6 +494,14 @@
   - 48
   - 98.0%
   - null
+- !!python/tuple
+  - avance
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 0
+  - null
+  - null
 - !!python/tuple
   - dbowman
   - Case Assignments & Outcomes
@@ -638,6 +662,14 @@
   - 46
   - 95.8%
   - null
+- !!python/tuple
+  - dbowman
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 2
+  - null
+  - null
 - !!python/tuple
   - fpoole
   - Case Assignments & Outcomes
@@ -798,6 +830,14 @@
   - 40
   - 97.6%
   - null
+- !!python/tuple
+  - fpoole
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 3
+  - null
+  - null
 - !!python/tuple
   - gfreeman
   - Case Assignments & Outcomes
@@ -958,3 +998,11 @@
   - 38
   - 95.0%
   - null
+- !!python/tuple
+  - gfreeman
+  - Partners & Clusters Initiated
+  - Total Clusters Initiated
+  - null
+  - 3
+  - null
+  - null