recce-nightly 1.2.0.20250506__py3-none-any.whl → 1.26.0.20251124__py3-none-any.whl

recce/tasks/valuediff.py

@@ -1,12 +1,12 @@
-from typing import TypedDict, Optional, List, Union
+from typing import List, Optional, TypedDict, Union
 
 from pydantic import BaseModel
 
-from .core import Task, TaskResultDiffer, CheckValidator
-from .dataframe import DataFrame
 from ..core import default_context
 from ..exceptions import RecceException
 from ..models import Check
+from .core import CheckValidator, Task, TaskResultDiffer
+from .dataframe import DataFrame
 
 
 class ValueDiffParams(BaseModel):
@@ -26,19 +26,6 @@ class ValueDiffResult(BaseModel):
 
 
 class ValueDiffMixin:
-    def _verify_dbt_packages_deps(self, dbt_adapter):
-        for macro_name, macro in dbt_adapter.manifest.macros.items():
-            if macro.package_name == 'audit_helper':
-                break
-        else:
-            raise RecceException(
-                r"Package 'audit_helper' not found. Please refer to the link to install: https://hub.getdbt.com/dbt-labs/audit_helper/")
-
-        for macro_name, macro in dbt_adapter.manifest.macros.items():
-            if macro.package_name == 'dbt_utils' and macro.name == 'generate_surrogate_key':
-                self.legacy_surrogate_key = False
-                break
-
     def _verify_primary_key(self, dbt_adapter, primary_key: Union[str, List[str]], model: str):
         self.update_progress(message=f"Verify primary key: {primary_key}")
         composite = True if isinstance(primary_key, List) else False
@@ -46,7 +33,21 @@ class ValueDiffMixin:
         if composite:
             if len(primary_key) == 0:
                 raise RecceException("Primary key cannot be empty")
-            sql_template = r"""{{ adapter.dispatch('test_unique_combination_of_columns', 'dbt_utils')(relation, primary_key) }}"""
+            sql_template = r"""
+            {%- set column_list = primary_key %}
+            {%- set columns_csv = column_list | join(', ') %}
+
+            with validation_errors as (
+                select
+                    {{ columns_csv }}
+                from {{ relation }}
+                group by {{ columns_csv }}
+                having count(*) > 1
+            )
+
+            select *
+            from validation_errors
+            """
         else:
             if primary_key is None or len(primary_key) == 0:
                 raise RecceException("Primary key cannot be empty")
@@ -54,7 +55,6 @@ class ValueDiffMixin:
 
         # check primary keys
         for base in [True, False]:
-
             relation = dbt_adapter.create_relation(model, base)
             context = dict(
                 relation=relation,
@@ -69,31 +69,36 @@ class ValueDiffMixin:
                 invalids = row[0]
                 if invalids > 0:
                     raise RecceException(
-                        f"Invalid primary key: \"{primary_key}\". The column should be unique. Please check by this sql: '{sql}'")
+                        f"Invalid primary key: \"{primary_key}\". The column should be unique. Please check by this sql: '{sql}'"
+                    )
                 break
             else:
                 # it will never happen unless we use a wrong check sql
-                raise RecceException('Cannot verify primary key')
+                raise RecceException("Cannot verify primary key")
 
 
 class ValueDiffTask(Task, ValueDiffMixin):
-
     def __init__(self, params):
         super().__init__()
         self.params = ValueDiffParams(**params)
         self.connection = None
         self.legacy_surrogate_key = True
 
-    def _query_value_diff(self, dbt_adpter, primary_key: Union[str, List[str]], model: str,
-                          columns: List[str] = None):
+    def _query_value_diff(
+        self,
+        dbt_adapter,
+        primary_key: Union[str, List[str]],
+        model: str,
+        columns: List[str] = None,
+    ):
         import agate
 
         column_groups = {}
         composite = True if isinstance(primary_key, List) else False
 
         if columns is None or len(columns) == 0:
-            base_columns = [column.column for column in dbt_adpter.get_columns(model, base=True)]
-            curr_columns = [column.column for column in dbt_adpter.get_columns(model, base=False)]
+            base_columns = [column.column for column in dbt_adapter.get_columns(model, base=True)]
+            curr_columns = [column.column for column in dbt_adapter.get_columns(model, base=False)]
             columns = [column for column in base_columns if column in curr_columns]
         completed = 0
 
@@ -106,81 +111,117 @@ class ValueDiffTask(Task, ValueDiffMixin):
                 columns.insert(0, primary_key)
 
         sql_template = r"""
-        {% set a_query %}
-            select {{ __PRIMARY_KEY__ }} as _pk, * from {{ base_relation }}
-        {% endset %}
-
-        {% set b_query %}
-            select {{ __PRIMARY_KEY__ }} as _pk, * from {{ curr_relation }}
-        {% endset %}
-
-        {{ audit_helper.compare_column_values(
-            a_query=a_query,
-            b_query=b_query,
-            primary_key="_pk",
-            column_to_compare=column_to_compare
-        ) }}
-        """
+        {%- set default_null_value = "_recce_surrogate_key_null_" -%}
+        {%- set fields = [] -%}
+
+        {%- for field in primary_keys -%}
+            {%- do fields.append(
+                "coalesce(cast(" ~ field ~ " as " ~ dbt.type_string() ~ "), '" ~ default_null_value  ~"')"
+            ) -%}
+
+            {%- if not loop.last %}
+                {%- do fields.append("'-'") -%}
+            {%- endif -%}
+        {%- endfor -%}
+
+        {%- set _pk = dbt.hash(dbt.concat(fields)) -%}
+
+        with a_query as (
+            select {{ _pk }} as _pk, * from {{ base_relation }}
+        ),
+
+        b_query as (
+            select {{ _pk }} as _pk, * from {{ curr_relation }}
+        ),
+
+        joined as (
+            select
+                coalesce(a_query._pk, b_query._pk) as _pk,
+                a_query.{{ column_to_compare }} as a_query_value,
+                b_query.{{ column_to_compare }} as b_query_value,
+                case
+                    when a_query.{{ column_to_compare }} = b_query.{{ column_to_compare }} then 'perfect match'
+                    when a_query.{{ column_to_compare }} is null and b_query.{{ column_to_compare }} is null then 'both are null'
+                    when a_query._pk is null then 'missing from {{ a_relation_name }}'
+                    when b_query._pk is null then 'missing from {{ b_relation_name }}'
+                    when a_query.{{ column_to_compare }} is null then 'value is null in {{ a_relation_name }} only'
+                    when b_query.{{ column_to_compare }} is null then 'value is null in {{ b_relation_name }} only'
+                    when a_query.{{ column_to_compare }} != b_query.{{ column_to_compare }} then 'values do not match'
+                    else 'unknown' -- this should never happen
+                end as match_status
+            from a_query
+            full outer join b_query on a_query._pk = b_query._pk
+        ),
+
+        aggregated as (
+            select
+                '{{ column_to_compare }}' as column_name,
+                match_status,
+                count(*) as count_records
+            from joined
+            group by 1, 2
+        )
 
-        if composite:
-            if self.legacy_surrogate_key:
-                new_primary_key = 'dbt_utils.surrogate_key(primary_key)'
-            else:
-                new_primary_key = 'dbt_utils.generate_surrogate_key(primary_key)'
-        else:
-            new_primary_key = 'primary_key'
-        sql_template = sql_template.replace('__PRIMARY_KEY__', new_primary_key)
+        select
+            column_name,
+            match_status,
+            count_records,
+            round(100.0 * count_records / sum(count_records) over (), 2) as percent_of_total
+        from aggregated
+        """
 
         for column in columns:
             self.update_progress(message=f"Diff column: {column}", percentage=completed / len(columns))
 
-            sql = dbt_adpter.generate_sql(sql_template, context=dict(
-                base_relation=dbt_adpter.create_relation(model, base=True),
-                curr_relation=dbt_adpter.create_relation(model, base=False),
-                primary_key=primary_key,
-                column_to_compare=column,
-            ))
+            sql = dbt_adapter.generate_sql(
+                sql_template,
+                context=dict(
+                    base_relation=dbt_adapter.create_relation(model, base=True),
+                    curr_relation=dbt_adapter.create_relation(model, base=False),
+                    primary_keys=primary_key if composite else [primary_key],
+                    column_to_compare=column,
+                ),
+            )
 
-            _, table = dbt_adpter.execute(sql, fetch=True)
+            _, table = dbt_adapter.execute(sql, fetch=True)
+            if column not in column_groups:
+                column_groups[column] = dict(added=0, removed=0, mismatched=0, matched=0)
             for row in table.rows:
                 # data example:
                 # ('COLUMN_NAME', 'MATCH_STATUS', 'COUNT_RECORDS', 'PERCENT_OF_TOTAL')
-                # ('EVENT_ID', '✅: perfect match', 158601510, Decimal('100.00'))
+                # ('EVENT_ID', 'perfect match', 158601510, Decimal('100.00'))
                 column_name, column_state, row_count, total_rate = row
-                if 'column_name' == row[0].lower():
+                if "column_name" == row[0].lower():
                     # skip column names
                     return
 
-                #
                 # sample data like this:
                 #     https://github.com/dbt-labs/dbt-audit-helper/blob/main/macros/compare_column_values.sql
                 #
-                #     '✅: perfect match'            -> matched
-                #     '✅: both are null'            -> matched
-                #     '🤷: missing from a'           -> row added
-                #     '🤷: missing from b'           -> row removed
-                #     '🤷: value is null in a only'  -> mismatched
-                #     '🤷: value is null in b only'  -> mismatched
-                #     '🙅: values do not match'      -> mismatched
-                #     'unknown'                      -> this should never happen
+                #     'perfect match'            -> matched
+                #     'both are null'            -> matched
+                #     'missing from a'           -> row added
+                #     'missing from b'           -> row removed
+                #     'value is null in a only'  -> mismatched
+                #     'value is null in b only'  -> mismatched
+                #     'values do not match'      -> mismatched
+                #     'unknown'                  -> this should never happen
                 # end as match_status,
 
-                if column_name not in column_groups:
-                    column_groups[column_name] = dict(added=0, removed=0, mismatched=0, matched=0)
-                if 'perfect match' in column_state:
-                    column_groups[column_name]['matched'] += row_count
-                if 'both are null' in column_state:
-                    column_groups[column_name]['matched'] += row_count
-                if 'missing from a' in column_state:
-                    column_groups[column_name]['added'] += row_count
-                if 'missing from b' in column_state:
-                    column_groups[column_name]['removed'] += row_count
-                if 'value is null in a only' in column_state:
-                    column_groups[column_name]['mismatched'] += row_count
-                if 'value is null in b only' in column_state:
-                    column_groups[column_name]['mismatched'] += row_count
-                if 'values do not match' in column_state:
-                    column_groups[column_name]['mismatched'] += row_count
+                state_mappings = {
+                    "perfect match": "matched",
+                    "both are null": "matched",
+                    "missing from a": "added",
+                    "missing from b": "removed",
+                    "value is null in a only": "mismatched",
+                    "value is null in b only": "mismatched",
+                    "values do not match": "mismatched",
+                }
+
+                # Use the mapping to update counts
+                for state, action in state_mappings.items():
+                    if state in column_state:
+                        column_groups[column_name][action] += row_count
 
             # Cancel as early as possible
             self.check_cancel()
@@ -188,9 +229,9 @@ class ValueDiffTask(Task, ValueDiffMixin):
             completed = completed + 1
 
         first = list(column_groups.values())[0]
-        added = first['added']
-        removed = first['removed']
-        common = first['matched'] + first['mismatched']
+        added = first["added"]
+        removed = first["removed"]
+        common = first["matched"] + first["mismatched"]
         total = common + added + removed
 
         row = []
@@ -200,12 +241,12 @@ class ValueDiffTask(Task, ValueDiffMixin):
             # This is incorrect when there are one side null
             # https://github.com/dbt-labs/dbt-audit-helper/blob/main/macros/compare_column_values.sql#L20-L23
             # matched = v['matched']
-            matched = common - v['mismatched']
+            matched = common - v["mismatched"]
             rate = None if common == 0 else matched / common
             record = [k, matched, rate]
             row.append(record)
 
-        column_names = ['column', 'matched', 'matched_p']
+        column_names = ["column", "matched", "matched_p"]
         column_types = [agate.Text(), agate.Number(), agate.Number()]
         table = agate.Table(row, column_names=column_names, column_types=column_types)
 
@@ -224,9 +265,6 @@ class ValueDiffTask(Task, ValueDiffMixin):
             model: str = self.params.model
             columns: List[str] = self.params.columns
 
-            self._verify_dbt_packages_deps(dbt_adapter)
-            self.check_cancel()
-
             self._verify_primary_key(dbt_adapter, primary_key, model)
             self.check_cancel()
 
@@ -243,35 +281,34 @@ class ValueDiffTask(Task, ValueDiffMixin):
 
 
 class ValueDiffTaskResultDiffer(TaskResultDiffer):
-
     def _check_result_changed_fn(self, result):
         is_changed = False
-        summary = result.get('summary', {})
-        added = summary.get('added', 0)
-        removed = summary.get('removed', 0)
-        changes = {
-            'column_changed': []
-        }
+        summary = result.get("summary", {})
+        added = summary.get("added", 0)
+        removed = summary.get("removed", 0)
+        changes = {"column_changed": []}
 
         if added > 0:
             is_changed = True
-            changes['row_added'] = added
+            changes["row_added"] = added
 
         if removed > 0:
             is_changed = True
-            changes['row_removed'] = removed
+            changes["row_removed"] = removed
 
-        row_data = result.get('data', {}).get('data', [])
+        row_data = result.get("data", {}).get("data", [])
         for row in row_data:
             column, matched, matched_p = row
             if float(matched_p) < 1.0:
                 # if there is any mismatched, we consider it as changed
                 is_changed = True
-                changes['column_changed'].append({
-                    'column': column,
-                    'matched': matched,
-                    'matched_p': matched_p,
-                })
+                changes["column_changed"].append(
+                    {
+                        "column": column,
+                        "matched": matched,
+                        "matched_p": matched_p,
+                    }
+                )
 
         return changes if is_changed else None
 
@@ -287,15 +324,19 @@ class ValueDiffDetailResult(DataFrame):
 
 
 class ValueDiffDetailTask(Task, ValueDiffMixin):
-
     def __init__(self, params):
         super().__init__()
         self.params = ValueDiffParams(**params)
         self.connection = None
         self.legacy_surrogate_key = True
 
-    def _query_value_diff(self, dbt_adapter, primary_key: Union[str, List[str]], model: str, columns: List[str] = None):
-
+    def _query_value_diff(
+        self,
+        dbt_adapter,
+        primary_key: Union[str, List[str]],
+        model: str,
+        columns: List[str] = None,
+    ):
         composite = True if isinstance(primary_key, List) else False
 
         if columns is None or len(columns) == 0:
@@ -312,45 +353,72 @@ class ValueDiffDetailTask(Task, ValueDiffMixin):
                 columns.insert(0, primary_key)
 
         sql_template = r"""
-        {% set col_list %}
-            {%- for col in columns %}
-                {{ col|trim }}
-                {%- if not loop.last %},{{ '\n  ' }}{%- endif -%}
-            {%- endfor -%}
-        {% endset %}
-
-        {% set a_query %}
-            select {{col_list}} from {{ base_relation }}
-        {% endset %}
-
-        {% set b_query %}
-            select {{col_list}} from {{ curr_relation }}
-        {% endset %}
-
-        {{ audit_helper.compare_queries(
-            a_query=a_query,
-            b_query=b_query,
-            primary_key=__PRIMARY_KEY__,
-            summarize=False,
-        ) }} limit {{ limit }}
-        """
+        with a_query as (
+            select {{ columns | join(',\n') }} from {{ base_relation }}
+        ),
+
+        b_query as (
+            select {{ columns | join(',\n') }} from {{ curr_relation }}
+        ),
+
+        a_intersect_b as (
+            select * from a_query
+            {{ dbt.intersect() }}
+            select * from b_query
+        ),
+
+        a_except_b as (
+            select * from a_query
+            {{ dbt.except() }}
+            select * from b_query
+        ),
+
+        b_except_a as (
+            select * from b_query
+            {{ dbt.except() }}
+            select * from a_query
+        ),
+
+        all_records as (
+            select
+                *,
+                true as in_a,
+                true as in_b
+            from a_intersect_b
+
+            union all
+
+            select
+                *,
+                true as in_a,
+                false as in_b
+            from a_except_b
+
+            union all
+
+            select
+                *,
+                false as in_a,
+                true as in_b
+            from b_except_a
+        )
 
-        if composite:
-            if self.legacy_surrogate_key:
-                new_primary_key = 'dbt_utils.surrogate_key(primary_key)'
-            else:
-                new_primary_key = 'dbt_utils.generate_surrogate_key(primary_key)'
-        else:
-            new_primary_key = 'primary_key'
-        sql_template = sql_template.replace('__PRIMARY_KEY__', new_primary_key)
+        select * from all_records
+        where not (in_a and in_b)
+        order by {{ primary_keys | join(',\n') }}, in_a desc, in_b desc
+        limit {{ limit }}
+        """
 
-        sql = dbt_adapter.generate_sql(sql_template, context=dict(
-            base_relation=dbt_adapter.create_relation(model, base=True),
-            curr_relation=dbt_adapter.create_relation(model, base=False),
-            primary_key=primary_key,
-            columns=columns,
-            limit=1000,
-        ))
+        sql = dbt_adapter.generate_sql(
+            sql_template,
+            context=dict(
+                base_relation=dbt_adapter.create_relation(model, base=True),
+                curr_relation=dbt_adapter.create_relation(model, base=False),
+                primary_keys=primary_key if composite else [primary_key],
+                columns=columns,
+                limit=1000,
+            ),
+        )
 
         _, table = dbt_adapter.execute(sql, fetch=True)
         self.check_cancel()
@@ -358,8 +426,8 @@ class ValueDiffDetailTask(Task, ValueDiffMixin):
         return DataFrame.from_agate(table)
 
     def execute(self):
-
         from recce.adapter.dbt_adapter import DbtAdapter
+
         dbt_adapter: DbtAdapter = default_context().adapter
 
         with dbt_adapter.connection_named("value diff"):
@@ -369,9 +437,6 @@ class ValueDiffDetailTask(Task, ValueDiffMixin):
             model: str = self.params.model
             columns: List[str] = self.params.columns
 
-            self._verify_dbt_packages_deps(dbt_adapter)
-            self.check_cancel()
-
             self._verify_primary_key(dbt_adapter, primary_key, model)
             self.check_cancel()
 
@@ -379,6 +444,7 @@ class ValueDiffDetailTask(Task, ValueDiffMixin):
 
     def cancel(self):
         from recce.adapter.dbt_adapter import DbtAdapter
+
         if self.connection:
             adapter: DbtAdapter = default_context().adapter
             with adapter.connection_named("cancel"):
@@ -386,9 +452,8 @@ class ValueDiffDetailTask(Task, ValueDiffMixin):
 
 
 class ValueDiffDetailTaskResultDiffer(TaskResultDiffer):
-
     def _check_result_changed_fn(self, result):
-        diff_data = result.get('data')
+        diff_data = result.get("data")
         if diff_data is None or len(diff_data) == 0:
             return None
 
@@ -397,7 +462,6 @@ class ValueDiffDetailTaskResultDiffer(TaskResultDiffer):
 
 
 class ValueDiffCheckValidator(CheckValidator):
-
     def validate_check(self, check: Check):
         try:
             ValueDiffParams(**check.params)

recce/util/__init__.py

@@ -1 +1,4 @@
 from .singleton import SingletonMeta
+
+# Explicitly declare exports
+__all__ = ["SingletonMeta"]

recce/util/api_token.py

@@ -0,0 +1,80 @@
+import click
+from rich.console import Console
+
+from recce import event
+from recce.event import get_recce_api_token, update_recce_api_token
+from recce.exceptions import RecceConfigException
+from recce.util.recce_cloud import (
+    RECCE_CLOUD_BASE_URL,
+    RecceCloud,
+)
+
+console = Console()
+
+
+def show_invalid_api_token_message():
+    """
+    Show the message when the API token is invalid.
+    """
+    console.print("[[red]Error[/red]] Invalid Recce Cloud API token.")
+    console.print("Please associate with your Recce Cloud account by the following command 'recce connect-to-cloud'.")
+    console.print(
+        "For more information, please visit: https://docs.reccehq.com/recce-cloud/share-recce-session-securely/#configure-recce-cloud-association-manually"
+    )
+
+
+def prepare_api_token(
+    interaction=False,
+    **kwargs,
+):
+    """
+    Prepare the API token for the request.
+    """
+    # Verify the API token for Recce Cloud Share Link
+    api_token = get_recce_api_token()
+    new_api_token = kwargs.get("api_token")
+    if new_api_token is not None and new_api_token.startswith("rct-"):
+        # Task Token
+        valid = RecceCloud(new_api_token).verify_token()
+        if not valid:
+            raise RecceConfigException("Invalid Recce Cloud Task token")
+        api_token = new_api_token
+    elif api_token != new_api_token and new_api_token is not None:
+        # Handle the API token provided by option `--api-token`
+        valid = RecceCloud(new_api_token).verify_token()
+        if not valid:
+            raise RecceConfigException("Invalid Recce Cloud API token")
+        event.log_connected_to_cloud()
+        api_token = new_api_token
+        update_recce_api_token(api_token)
+        console.print(
+            "[[green]Success[/green]] User profile has been updated to include the Recce Cloud API Token. "
+            "You no longer need to append --api-token to the recce command"
+        )
+    elif api_token:
+        # Verify the API token from the user profile
+        valid = RecceCloud(api_token).verify_token()
+        if not valid:
+            console.print("[[yellow]Warning[/yellow]] Invalid Recce Cloud API token. Skipping the share link.")
+            api_token = None
+        if valid:
+            event.log_connected_to_cloud()
+    else:
+        # No api_token provided
+        if interaction:
+            console.print(
+                "An API token is required for this feature. This can be obtained in your user account settings.\n"
+                f"{RECCE_CLOUD_BASE_URL}/settings#tokens\n"
+                "Your API token can be added to '~/.recce/profile.yml' for more convenient sharing."
+            )
+            api_token = click.prompt("Your Recce API token", type=str, hide_input=True, show_default=False)
+            valid = RecceCloud(api_token).verify_token()
+            if not valid:
+                raise RecceConfigException("Invalid Recce Cloud API token")
+            update_recce_api_token(api_token)
+            console.print(
+                "[[green]Success[/green]] User profile has been updated to include the Recce Cloud API Token. "
+                "You no longer need to append --api-token to the recce command"
+            )
+
+    return api_token