⚙️ fixed: change way to get dq.

Author: korawica

README.md

@@ -69,7 +69,7 @@ print(statement.strip().strip('\n'))
 
 The result SQL statement:
 
-```text
+```sql
 MERGE INTO catalog-name.schema-name.table-name AS target
 USING (
     WITH change_query AS (
@@ -117,20 +117,26 @@ WHEN NOT MATCHED THEN INSERT
 
 ```python
 from sqlplate import SQLPlate
-from sqlplate.rules import Unique, NotNull, Count
 
-report: str = (
+statement: str = (
     SQLPlate.format('databricks')
-    .quality(mode="pushdown")
+    .template('quality.check')
     .option('catalog', 'catalog-name')
     .option('schema', 'schema-name')
     .option('table', 'table-name')
-    .check('unique', Unique(cols=['pk_col']))
-    .check('not-null', NotNull(cols=['col01', 'col02']))
-    .check('row-count', Count())
-    .validate(output='html')
+    .option('filter', "load_date >= to_timestamp('20250201', 'yyyyMMdd')")
+    .option('unique', ['pk_col'])
+    .option('notnull', ['col01', 'col02'])
+    .option('row_count', True)
+    .load()
 )
-print(report.strip().strip('\n'))
+print(statement.strip().strip('\n'))
+```
+
+The result SQL statement:
+
+```sql
+
 ```
 
 ## :chains: Support Systems

src/sqlplate/rules.py

@@ -147,21 +147,3 @@ def stream(
             )
             if trim(s) != ''
         )
-
-    def check(
-        self,
-        name: str,
-        rule: Any,
-    ) -> 'SQLPlate':
-        return self
-
-    def validate(
-        self,
-        output: Literal["json", "html"],
-    ):
-        return self
-
-    def filter(
-        self,
-    ):
-        return self

src/sqlplate/sqlplate.py

@@ -0,0 +1,19 @@
+{% extends "base.jinja" %}
+
+{% block statement %}
+WITH source AS (
+    SELECT
+        *
+    FROM {{ catalog }}.{{ schema }}.{{ table }}
+    {%+ if filter %}WHERE {{ filter }}{% endif +%}
+)
+SELECT
+    *
+    {%+ if row_count %}, (SELECT COUNT(1) FROM source) AS table_records{% endif +%}
+    {%+ if unique -%}
+        {%- for col in unique -%}
+    , (SELECT COUNT {{ col }} FROM (SELECT DISTINCT {{ col}} FROM source)) AS unique_{{ col }}
+        {%- endfor -%}
+    {%- endif +%}
+FROM source
+{% endblock statement %}

templates/databricks/quality.check.sql

@@ -277,3 +277,19 @@ def test_sql_full_dump(template_path):
         FROM ( SELECT * FROM catalog-name.schema-name.source-name ) AS sub_query
         ;
         """).strip('\n')
+
+
+def test_quality_check(template_path):
+    statement: SQLPlate = (
+        SQLPlate.format('databricks', path=template_path)
+        .template('quality.check')
+        .option('catalog', 'catalog-name')
+        .option('schema', 'schema-name')
+        .option('table', 'table-name')
+        .option('filter', "load_date >= to_timestamp('20250201', 'yyyyMMdd')")
+        .option('unique', ['pk_col'])
+        .option('notnull', ['col01', 'col02'])
+        .option("row_count", True)
+        .load()
+    )
+    print(statement)