From 8fee5c33395fbe0054289ef105f5b0cf570cf8cd Mon Sep 17 00:00:00 2001
From: Joey LeGrand <jrlegrand@gmail.com>
Date: Mon, 3 Feb 2025 04:52:14 +0000
Subject: [PATCH] Update dbt for ATCPROD

---
 .../atc_codes_to_rxnorm_products.sql               | 12 ++++++------
 .../models/staging/rxclass/_rxclass__sources.yml   | 10 ++++++++--
 ...rxclass__atc_codes_to_rxnorm_product_rxcuis.sql | 11 -----------
 .../staging/rxclass/stg_rxclass__atcprod.sql       | 14 ++++++++++++++
 4 files changed, 28 insertions(+), 19 deletions(-)
 delete mode 100644 dbt/sagerx/models/staging/rxclass/stg_rxclass__atc_codes_to_rxnorm_product_rxcuis.sql
 create mode 100644 dbt/sagerx/models/staging/rxclass/stg_rxclass__atcprod.sql

diff --git a/dbt/sagerx/models/marts/classification/atc_codes_to_rxnorm_products.sql b/dbt/sagerx/models/marts/classification/atc_codes_to_rxnorm_products.sql
index e5fd358b..5649bf7b 100644
--- a/dbt/sagerx/models/marts/classification/atc_codes_to_rxnorm_products.sql
+++ b/dbt/sagerx/models/marts/classification/atc_codes_to_rxnorm_products.sql
@@ -1,8 +1,8 @@
 -- atc_codes_to_rxnorm_products
 
-with atc_codes_to_rxnorm_product_rxcuis as (
+with rxclass_atcprod as (
 
-	select * from {{ ref('stg_rxclass__atc_codes_to_rxnorm_product_rxcuis') }}
+	select * from {{ ref('stg_rxclass__atcprod') }}
 
 )
 
@@ -19,7 +19,7 @@ with atc_codes_to_rxnorm_product_rxcuis as (
 )
 
 select distinct
-	atc_codes_to_rxnorm_product_rxcuis.rxcui
+	rxclass_atcprod.rxcui
 	, rxnorm_product_rxcuis.str as rxnorm_description
 	, atc_codes.atc_1_code
 	, atc_codes.atc_2_code
@@ -29,9 +29,9 @@ select distinct
 	, atc_codes.atc_2_name
 	, atc_codes.atc_3_name
 	, atc_codes.atc_4_name
-from atc_codes_to_rxnorm_product_rxcuis
+from rxclass_atcprod
 left join atc_codes
-	on atc_codes.atc_4_code = atc_codes_to_rxnorm_product_rxcuis.class_id
+	on atc_codes.atc_4_code = rxclass_atcprod.class_id
 left join rxnorm_product_rxcuis
-	on rxnorm_product_rxcuis.rxcui = atc_codes_to_rxnorm_product_rxcuis.rxcui
+	on rxnorm_product_rxcuis.rxcui = rxclass_atcprod.rxcui
 order by rxcui
diff --git a/dbt/sagerx/models/staging/rxclass/_rxclass__sources.yml b/dbt/sagerx/models/staging/rxclass/_rxclass__sources.yml
index 969b8d6d..6a24d69e 100644
--- a/dbt/sagerx/models/staging/rxclass/_rxclass__sources.yml
+++ b/dbt/sagerx/models/staging/rxclass/_rxclass__sources.yml
@@ -3,7 +3,13 @@ version: 2
 sources:
   - name: rxclass
     description: |
-      A product-level mapping of RxNorm to ATC1-4 classes.
+      This data source represents all of RxClass.
+
+      Below are descriptions of a select few of the RxClass sources.
+
+      ## ATCPROD
+
+      > A product-level mapping of RxNorm to ATC1-4 classes.
 
       RxClass has added RxNorm product-level mapping for ATC. The product-level mapping is now the default source mapping for the ATC classes in the RxClass browser.
 
@@ -17,4 +23,4 @@ sources:
       The original ingredient-level mapping to ATC from the WHO Collaborating Centre for Drug Statistics Methodology remains available in RxClass (select ATC under Edit Drug Sources) and through the RxClass API (relaSource=ATC).
     schema: sagerx_lake
     tables:
-      - name: rxclass_atc_to_product
+      - name: rxclass
diff --git a/dbt/sagerx/models/staging/rxclass/stg_rxclass__atc_codes_to_rxnorm_product_rxcuis.sql b/dbt/sagerx/models/staging/rxclass/stg_rxclass__atc_codes_to_rxnorm_product_rxcuis.sql
deleted file mode 100644
index 845b9a5b..00000000
--- a/dbt/sagerx/models/staging/rxclass/stg_rxclass__atc_codes_to_rxnorm_product_rxcuis.sql
+++ /dev/null
@@ -1,11 +0,0 @@
--- stg_rxclass__atc_codes_to_rxnorm_product_rxcuis.sql
-
-with atc_codes_to_rxnorm_product_rxcuis as (
-    
-    select * from {{ source('rxclass', 'rxclass_atc_to_product') }}
-
-)
-
-select
-    *
-from atc_codes_to_rxnorm_product_rxcuis
diff --git a/dbt/sagerx/models/staging/rxclass/stg_rxclass__atcprod.sql b/dbt/sagerx/models/staging/rxclass/stg_rxclass__atcprod.sql
new file mode 100644
index 00000000..1c8f5587
--- /dev/null
+++ b/dbt/sagerx/models/staging/rxclass/stg_rxclass__atcprod.sql
@@ -0,0 +1,14 @@
+-- stg_rxclass__atcprod.sql
+
+with atcprod as (
+    
+    select
+        *
+    from {{ source('rxclass', 'rxclass') }}
+    where rela_source = 'ATCPROD'
+
+)
+
+select
+    *
+from atcprod