macro-out the staging model unioning

models/staging/stripe/stg_stripe__balance_transactions.sql

@@ -1,10 +1,20 @@
 with
     raw_balance_transactions as (
-        select *, 'stripe_uk_account' as stripe_source_account
-        from {{ source("stripe_uk", "balance_transactions") }}
+        /*
+        This macro runs for every country code and:
+          - Reads from the right source (stripe_<country-code>)
+          - Adds a column identifying the source account with a string like "stripe_<country>_account"
+        */
+        {% set countries = ["us", "uk"] %}
+
+        {% for country in countries %}
+            select *, 'stripe_{{ country }}_account' as stripe_source_account
+            from {{ source("stripe_" ~ country, "balance_transactions") }}
+            {% if not loop.last -%}
                 union all
-        select *, 'stripe_us_account' as stripe_source_account
-        from {{ source("stripe_us", "balance_transactions") }}
+            {%- endif %}
+        {% endfor %}
+
     ),
     stg_seed__currencies as (select * from {{ ref("stg_seed__currencies") }}),
     stg_stripe__balance_transactions as (