sqlbuild 0.57.0__tar.gz → 0.58.0__tar.gz

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/Cargo.lock

@@ -572,7 +572,7 @@ checksum = "f8fadd59c855ef2080decdef8ff161eb6661b86933c9d82e5ba29dc602a55aba"
 
 [[package]]
 name = "sqlbuild-kata-native"
-version = "0.57.0"
+version = "0.58.0"
 dependencies = [
  "fensu-policy",
  "globset",

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlbuild
-Version: 0.57.0
+Version: 0.58.0
 Requires-Dist: duckdb>=1.3.2
 Requires-Dist: filelock>=3.20.3
 Requires-Dist: polyglot-sql>=0.9.2

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/crates/sqlbuild-kata-native/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "sqlbuild-kata-native"
-version = "0.57.0"
+version = "0.58.0"
 edition.workspace = true
 license.workspace = true
 publish.workspace = true

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "maturin"
 
 [project]
 name = "sqlbuild"
-version = "0.57.0"
+version = "0.58.0"
 description = "Typed, test-first SQL pipelines with local E2E testing"
 readme = "README.md"
 requires-python = ">=3.12"

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/cli/commands/_helpers/playground/copy.py

@@ -271,8 +271,8 @@ SELECT 1
 """
 
 _VIRTUAL_SCENARIO_CUSTOMER_REVENUE_SQL: str = """SCENARIO (
-  description: "Customer revenue includes seed-backed waffle categories",
-  tags: ["virtual", "example"]
+  description "Customer revenue includes seed-backed waffle categories",
+  tags ["virtual", "example"]
 );
 
 WITH

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/cli/commands/_helpers/playground/templates/waffle_shop/tests/scenarios/daily_revenue_minimal.sql

@@ -1,6 +1,6 @@
 SCENARIO (
-  description: "Daily revenue includes only successful payments",
-  tags: ["revenue", "example"]
+  description "Daily revenue includes only successful payments",
+  tags ["revenue", "example"]
 );
 
 WITH

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/cli/commands/_helpers/playground/templates/waffle_shop/tests/scenarios/daily_revenue_multi_order.sql

@@ -1,6 +1,6 @@
 SCENARIO (
-  description: "Daily revenue aggregates multiple successful orders",
-  tags: ["revenue", "example"]
+  description "Daily revenue aggregates multiple successful orders",
+  tags ["revenue", "example"]
 );
 
 WITH

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/cli/commands/_helpers/playground/templates/waffle_shop/tests/unit/test_line_total_cents_macro.sql

@@ -1,4 +1,4 @@
-TEST (mode: macro, name: "calculates line total cents");
+TEST (mode macro, name "calculates line total cents");
 
 WITH
 input_values AS (

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/compile/_helpers/sql_tests/core.py

@@ -323,17 +323,17 @@ def _classify_model_sql_test_ctes(
         if cte.name in {MACRO_ACTUAL_TEST_CTE_NAME, MACRO_EXPECTED_TEST_CTE_NAME}:
             raise CompileInputError(
                 f"SQL test '{file_label}' is mode '{DEFAULT_SQL_TEST_MODE.value}' but defines "
-                f"macro-test CTE '{cte.name}'; use TEST (mode: {SqlTestMode.MACRO.value})"
+                f"macro-test CTE '{cte.name}'; use TEST (mode {SqlTestMode.MACRO.value})"
             )
         if cte.name in {UDF_ACTUAL_TEST_CTE_NAME, UDF_EXPECTED_TEST_CTE_NAME}:
             raise CompileInputError(
                 f"SQL test '{file_label}' is mode '{DEFAULT_SQL_TEST_MODE.value}' but defines "
-                f"UDF-test CTE '{cte.name}'; use TEST (mode: {SqlTestMode.UDF.value})"
+                f"UDF-test CTE '{cte.name}'; use TEST (mode {SqlTestMode.UDF.value})"
             )
         if cte.name in {TABLE_FN_ACTUAL_TEST_CTE_NAME, TABLE_FN_EXPECTED_TEST_CTE_NAME}:
             raise CompileInputError(
                 f"SQL test '{file_label}' is mode '{DEFAULT_SQL_TEST_MODE.value}' but defines "
-                f"table_fn-test CTE '{cte.name}'; use TEST (mode: {SqlTestMode.TABLE_FN.value})"
+                f"table_fn-test CTE '{cte.name}'; use TEST (mode {SqlTestMode.TABLE_FN.value})"
             )
         if cte.name.startswith(MACRO_TEST_CTE_PREFIX):
             macro_name: str = _require_prefixed_name(

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/discovery/_helpers/sql/audits.py

@@ -7,9 +7,7 @@ from inspect import cleandoc
 from pathlib import Path
 from typing import cast
 
-import yaml
-from yaml import YAMLError
-
+from sqlbuild.compiler.discovery._helpers.sql.model_files import parse_header_values
 from sqlbuild.compiler.discovery.exceptions import SqlAuditParseError
 from sqlbuild.compiler.discovery.models import DiscoveredAuditBlock
 
@@ -21,8 +19,17 @@ _AUDIT_HEADER_ONLY_PATTERN: re.Pattern[str] = re.compile(
     r"^\s*AUDIT\s*\((?P<header>.*?)\)\s*;\s*",
     re.DOTALL | re.MULTILINE,
 )
+_AUDIT_NAME_HEADER_KEY: str = "name"
+_AUDIT_SEVERITY_HEADER_KEY: str = "severity"
+_AUDIT_RUN_SCOPE_HEADER_KEY: str = "run_scope"
+_AUDIT_ALWAYS_RUN_HEADER_KEY: str = "always_run"
 _SUPPORTED_AUDIT_HEADER_KEYS: frozenset[str] = frozenset(
-    {"name", "severity", "run_scope", "always_run"}
+    {
+        _AUDIT_NAME_HEADER_KEY,
+        _AUDIT_SEVERITY_HEADER_KEY,
+        _AUDIT_RUN_SCOPE_HEADER_KEY,
+        _AUDIT_ALWAYS_RUN_HEADER_KEY,
+    }
 )
 
 
@@ -96,7 +103,7 @@ def _parse_single_sql_audit_block(
     if not sql_body:
         raise SqlAuditParseError(f"SQL audit '{file_path}' must define SQL after AUDIT(...)")
 
-    name_value: object | None = header_values.get("name")
+    name_value: object | None = header_values.get(_AUDIT_NAME_HEADER_KEY)
     audit_name: str | None = cast(str | None, name_value)
     return DiscoveredAuditBlock(
         audit_index=audit_index,
@@ -107,22 +114,12 @@ def _parse_single_sql_audit_block(
 
 
 def _parse_audit_header(*, header: str, file_path: Path) -> dict[str, object]:
-    stripped_header: str = header.strip()
-    if not stripped_header:
-        return {}
-
-    try:
-        parsed_header: object = yaml.safe_load(f"{{{stripped_header}}}")
-    except YAMLError as error:
-        raise SqlAuditParseError(
-            f"AUDIT() header in '{file_path}' could not be parsed: {error}"
-        ) from error
-    if not isinstance(parsed_header, dict) or not all(
-        isinstance(key, str) for key in parsed_header
-    ):
-        raise SqlAuditParseError(
-            f"AUDIT() header in '{file_path}' must be a mapping like `AUDIT (name: \"...\");`"
-        )
+    parsed_header: dict[str, object] = parse_header_values(
+        header=header,
+        file_path=file_path,
+        statement_name="AUDIT",
+        error_class=SqlAuditParseError,
+    )
 
     unsupported_keys: tuple[str, ...] = tuple(
         str(key) for key in parsed_header if key not in _SUPPORTED_AUDIT_HEADER_KEYS
@@ -132,11 +129,20 @@ def _parse_audit_header(*, header: str, file_path: Path) -> dict[str, object]:
             f"AUDIT() in '{file_path}' has unsupported keys: {', '.join(unsupported_keys)}"
         )
 
-    name_value: object | None = parsed_header.get("name")
-    if name_value is not None and (not isinstance(name_value, str) or not name_value.strip()):
+    name_value: object | None = parsed_header.get(_AUDIT_NAME_HEADER_KEY)
+    if _AUDIT_NAME_HEADER_KEY in parsed_header and (
+        not isinstance(name_value, str) or not name_value.strip()
+    ):
         raise SqlAuditParseError(f"AUDIT() name in '{file_path}' must be a non-empty string")
-
-    return cast(dict[str, object], parsed_header)
+    for key in (_AUDIT_SEVERITY_HEADER_KEY, _AUDIT_RUN_SCOPE_HEADER_KEY):
+        value: object | None = parsed_header.get(key)
+        if key in parsed_header and (not isinstance(value, str) or not value.strip()):
+            raise SqlAuditParseError(f"AUDIT() {key} in '{file_path}' must be a non-empty string")
+    always_run: object | None = parsed_header.get(_AUDIT_ALWAYS_RUN_HEADER_KEY)
+    if _AUDIT_ALWAYS_RUN_HEADER_KEY in parsed_header and not isinstance(always_run, bool):
+        raise SqlAuditParseError(f"AUDIT() always_run in '{file_path}' must be a boolean")
+
+    return parsed_header
 
 
 def _validate_audit_names(*, file_path: Path, blocks: tuple[DiscoveredAuditBlock, ...]) -> None:

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/discovery/_helpers/sql/hooks.py

@@ -6,13 +6,12 @@ from inspect import cleandoc
 from pathlib import Path
 from typing import cast
 
-import yaml
-from yaml import YAMLError
-
+from sqlbuild.compiler.discovery._helpers.sql.model_files import parse_header_values
 from sqlbuild.compiler.discovery.exceptions import SqlHookParseError
 from sqlbuild.compiler.discovery.models import DiscoveredSqlHookFile
 
-_SUPPORTED_HOOK_HEADER_KEYS: frozenset[str] = frozenset({"description"})
+_HOOK_DESCRIPTION_HEADER_KEY: str = "description"
+_SUPPORTED_HOOK_HEADER_KEYS: frozenset[str] = frozenset({_HOOK_DESCRIPTION_HEADER_KEY})
 _SQL_ESCAPE_CHARACTER: str = "\\"
 _SQL_QUOTE_CHARACTERS: frozenset[str] = frozenset({"'", '"', "`"})
 _SQL_STATEMENT_TERMINATOR: str = ";"
@@ -98,21 +97,12 @@ def _split_hook_file(*, contents: str, file_path: Path) -> tuple[str, str]:
 
 
 def _parse_hook_header(*, header: str, file_path: Path) -> dict[str, object]:
-    stripped_header: str = header.strip()
-    if not stripped_header:
-        return {}
-    try:
-        parsed_header: object = yaml.safe_load(f"{{{stripped_header}}}")
-    except YAMLError as error:
-        raise SqlHookParseError(
-            f"HOOK() header in '{file_path}' could not be parsed: {error}"
-        ) from error
-    if not isinstance(parsed_header, dict) or not all(
-        isinstance(key, str) for key in parsed_header
-    ):
-        raise SqlHookParseError(
-            f"HOOK() header in '{file_path}' must be a mapping like `HOOK (description: \"...\");`"
-        )
+    parsed_header: dict[str, object] = parse_header_values(
+        header=header,
+        file_path=file_path,
+        statement_name="HOOK",
+        error_class=SqlHookParseError,
+    )
     unsupported_keys: tuple[str, ...] = tuple(
         str(key) for key in parsed_header if key not in _SUPPORTED_HOOK_HEADER_KEYS
     )
@@ -120,7 +110,9 @@ def _parse_hook_header(*, header: str, file_path: Path) -> dict[str, object]:
         raise SqlHookParseError(
             f"HOOK() in '{file_path}' has unsupported keys: {', '.join(unsupported_keys)}"
         )
-    description: object | None = parsed_header.get("description")
-    if description is not None and (not isinstance(description, str) or not description.strip()):
+    description: object | None = parsed_header.get(_HOOK_DESCRIPTION_HEADER_KEY)
+    if _HOOK_DESCRIPTION_HEADER_KEY in parsed_header and (
+        not isinstance(description, str) or not description.strip()
+    ):
         raise SqlHookParseError(f"HOOK() description in '{file_path}' must be a non-empty string")
-    return cast(dict[str, object], parsed_header)
+    return parsed_header

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/discovery/_helpers/sql/model_files.py

@@ -7,7 +7,11 @@ from bisect import bisect_right
 from dataclasses import dataclass
 from pathlib import Path
 
-from sqlbuild.compiler.discovery.exceptions import ModelHeaderSyntaxError, ModelSqlParseError
+from sqlbuild.compiler.discovery.exceptions import (
+    DiscoveryError,
+    ModelHeaderSyntaxError,
+    ModelSqlParseError,
+)
 from sqlbuild.compiler.discovery.models import NamedSqlHookEntry, PythonHookEntry, SqlHookEntry
 from sqlbuild.compiler.references.types import SqlReferenceKind
 from sqlbuild.spec.contracts.models import SourceLocation
@@ -461,7 +465,13 @@ def _keyword_at(*, sql: str, keyword: str, index: int) -> bool:
     )
 
 
-def parse_header_values(*, header: str, file_path: Path, statement_name: str) -> dict[str, object]:
+def parse_header_values(
+    *,
+    header: str,
+    file_path: Path,
+    statement_name: str,
+    error_class: type[DiscoveryError] = ModelSqlParseError,
+) -> dict[str, object]:
     """Parse one SQLBuild parenthesized header into nested Python values."""
 
     try:
@@ -470,7 +480,7 @@ def parse_header_values(*, header: str, file_path: Path, statement_name: str) ->
     except ModelSqlParseError:
         raise
     except ModelHeaderSyntaxError as error:
-        raise ModelSqlParseError(
+        raise error_class(
             f"{statement_name}(...) in '{file_path}' contains invalid SQLBuild header syntax: "
             f"{error}"
         ) from error

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/discovery/_helpers/sql/scenarios.py

@@ -5,11 +5,8 @@ from __future__ import annotations
 import re
 from inspect import cleandoc
 from pathlib import Path
-from typing import cast
-
-import yaml
-from yaml import YAMLError
 
+from sqlbuild.compiler.discovery._helpers.sql.model_files import parse_header_values
 from sqlbuild.compiler.discovery.exceptions import SqlScenarioParseError
 from sqlbuild.compiler.discovery.models import DiscoveredSqlScenarioFile
 
@@ -17,6 +14,8 @@ _SCENARIO_HEADER_PATTERN: re.Pattern[str] = re.compile(
     r"^\s*SCENARIO\s*\((?P<header>.*?)\)\s*;\s*(?P<sql>.*)\Z",
     re.DOTALL,
 )
+_SCENARIO_DESCRIPTION_HEADER_KEY: str = "description"
+_SCENARIO_TAGS_HEADER_KEY: str = "tags"
 
 
 def parse_sql_scenario_file(
@@ -52,25 +51,17 @@ def parse_sql_scenario_file(
 
 
 def _parse_scenario_header(*, header: str, file_path: Path) -> dict[str, object]:
-    stripped_header: str = header.strip()
-    if not stripped_header:
-        return {}
-
-    try:
-        parsed_header: object = yaml.safe_load(f"{{{stripped_header}}}")
-    except YAMLError as error:
-        raise SqlScenarioParseError(
-            f"SCENARIO() header in '{file_path}' could not be parsed: {error}"
-        ) from error
-    if not isinstance(parsed_header, dict) or not all(
-        isinstance(key, str) for key in parsed_header
-    ):
-        raise SqlScenarioParseError(
-            f"SCENARIO() header in '{file_path}' must be a mapping like "
-            '`SCENARIO (description: "...");`'
-        )
+    parsed_header: dict[str, object] = parse_header_values(
+        header=header,
+        file_path=file_path,
+        statement_name="SCENARIO",
+        error_class=SqlScenarioParseError,
+    )
 
-    supported_keys: set[str] = {"description", "tags"}
+    supported_keys: set[str] = {
+        _SCENARIO_DESCRIPTION_HEADER_KEY,
+        _SCENARIO_TAGS_HEADER_KEY,
+    }
     unsupported_keys: tuple[str, ...] = tuple(
         str(key) for key in parsed_header if key not in supported_keys
     )
@@ -80,13 +71,13 @@ def _parse_scenario_header(*, header: str, file_path: Path) -> dict[str, object]
             f"unsupported keys: {', '.join(unsupported_keys)}"
         )
 
-    description_value: object | None = parsed_header.get("description")
-    if description_value is not None and not isinstance(description_value, str):
+    description_value: object | None = parsed_header.get(_SCENARIO_DESCRIPTION_HEADER_KEY)
+    if _SCENARIO_DESCRIPTION_HEADER_KEY in parsed_header and not isinstance(description_value, str):
         raise SqlScenarioParseError(f"SCENARIO() description in '{file_path}' must be a string")
-    tags_value: object | None = parsed_header.get("tags")
-    if tags_value is not None and (
+    tags_value: object | None = parsed_header.get(_SCENARIO_TAGS_HEADER_KEY)
+    if _SCENARIO_TAGS_HEADER_KEY in parsed_header and (
         not isinstance(tags_value, list) or not all(isinstance(tag, str) for tag in tags_value)
     ):
         raise SqlScenarioParseError(f"SCENARIO() tags in '{file_path}' must be a list of strings")
 
-    return cast(dict[str, object], parsed_header)
+    return parsed_header

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/compiler/discovery/_helpers/sql/tests.py

@@ -7,11 +7,9 @@ from inspect import cleandoc
 from pathlib import Path
 from typing import cast
 
-import yaml
-from yaml import YAMLError
-
 from sqlbuild.compiler.compile.constants import DEFAULT_SQL_TEST_MODE
 from sqlbuild.compiler.compile.types import SqlTestMode
+from sqlbuild.compiler.discovery._helpers.sql.model_files import parse_header_values
 from sqlbuild.compiler.discovery.exceptions import SqlTestParseError
 from sqlbuild.compiler.discovery.models import DiscoveredSqlTestBlock
 
@@ -23,6 +21,8 @@ _TEST_HEADER_ONLY_PATTERN: re.Pattern[str] = re.compile(
     r"^\s*TEST\s*\((?P<header>.*?)\)\s*;\s*",
     re.DOTALL | re.MULTILINE,
 )
+_TEST_NAME_HEADER_KEY: str = "name"
+_TEST_MODE_HEADER_KEY: str = "mode"
 
 
 def parse_sql_test_file(*, contents: str, file_path: Path) -> tuple[DiscoveredSqlTestBlock, ...]:
@@ -109,24 +109,14 @@ def _parse_single_sql_test_block(
 
 
 def _parse_test_header(*, header: str, file_path: Path) -> dict[str, object]:
-    stripped_header: str = header.strip()
-    if not stripped_header:
-        return {}
-
-    try:
-        parsed_header: object = yaml.safe_load(f"{{{stripped_header}}}")
-    except YAMLError as error:
-        raise SqlTestParseError(
-            f"TEST() header in '{file_path}' could not be parsed: {error}"
-        ) from error
-    if not isinstance(parsed_header, dict) or not all(
-        isinstance(key, str) for key in parsed_header
-    ):
-        raise SqlTestParseError(
-            f"TEST() header in '{file_path}' must be a mapping like `TEST (name: \"...\");`"
-        )
+    parsed_header: dict[str, object] = parse_header_values(
+        header=header,
+        file_path=file_path,
+        statement_name="TEST",
+        error_class=SqlTestParseError,
+    )
 
-    supported_keys: frozenset[str] = frozenset({"name", "mode"})
+    supported_keys: frozenset[str] = frozenset({_TEST_NAME_HEADER_KEY, _TEST_MODE_HEADER_KEY})
     unsupported_keys: tuple[str, ...] = tuple(
         str(key) for key in parsed_header if key not in supported_keys
     )
@@ -136,22 +126,20 @@ def _parse_test_header(*, header: str, file_path: Path) -> dict[str, object]:
             f"{', '.join(unsupported_keys)}"
         )
 
-    _validate_test_name(name_value=parsed_header.get("name"), file_path=file_path)
-    _validate_test_mode(mode_value=parsed_header.get("mode"), file_path=file_path)
+    if _TEST_NAME_HEADER_KEY in parsed_header:
+        _validate_test_name(name_value=parsed_header[_TEST_NAME_HEADER_KEY], file_path=file_path)
+    if _TEST_MODE_HEADER_KEY in parsed_header:
+        _validate_test_mode(mode_value=parsed_header[_TEST_MODE_HEADER_KEY], file_path=file_path)
 
-    return cast(dict[str, object], parsed_header)
+    return parsed_header
 
 
-def _validate_test_name(*, name_value: object | None, file_path: Path) -> None:
-    if name_value is None:
-        return
+def _validate_test_name(*, name_value: object, file_path: Path) -> None:
     if not isinstance(name_value, str) or not name_value.strip():
         raise SqlTestParseError(f"TEST() name in '{file_path}' must be a non-empty string")
 
 
-def _validate_test_mode(*, mode_value: object | None, file_path: Path) -> None:
-    if mode_value is None:
-        return
+def _validate_test_mode(*, mode_value: object, file_path: Path) -> None:
     if not isinstance(mode_value, str):
         raise SqlTestParseError(f"TEST() mode in '{file_path}' must be a string")
 

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/lint/_helpers/native.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from pathlib import Path
 
 from sqlbuild.compiler.discovery._helpers.sql.model_files import (
-    parse_header_values,  # noqa: FFL102 - lint owns header rules and reuses the compiler's header parser
+    parse_header_values,  # noqa: FFL102 - the compiler owns the canonical native header grammar
 )
 from sqlbuild.lint.constants import (
     DESCRIPTION_HEADER_KINDS,
@@ -20,7 +20,6 @@ from sqlbuild.lint.constants import (
     RULE_LEADING_COMMENT_DESCRIPTION,
     VIOLATION_SEVERITY_FAULT,
 )
-from sqlbuild.lint.exceptions import LintError
 from sqlbuild.lint.models import HeaderSpan, LintConfig, LintViolation
 
 _QUOTE_CHARACTERS: frozenset[str] = frozenset({"'", '"'})
@@ -32,7 +31,6 @@ _LINE_COMMENT_PREFIX: str = "--"
 _DESCRIPTION_KEY: str = "description"
 _DESCRIPTION_SENTINEL_PATH: str = "<lint>"
 _HEADER_INDENT: str = "  "
-_FUNCTION_KIND: str = "FUNCTION"
 
 
 @dataclass(frozen=True)
@@ -154,21 +152,11 @@ def _lint_header_values(
 
 def _parse_header_values(*, kind: str, header_text: str) -> dict[str, object]:
     inner: str = _inner_header_text(header_text=header_text)
-    if kind in (HEADER_KIND_MODEL, _FUNCTION_KIND):
-        return parse_header_values(
-            header=inner,
-            file_path=Path(_DESCRIPTION_SENTINEL_PATH),
-            statement_name=kind,
-        )
-    import yaml
-
-    stripped: str = inner.strip()
-    if not stripped:
-        return {}
-    parsed: object = yaml.safe_load(f"{{{stripped}}}")
-    if not isinstance(parsed, dict):
-        raise LintError("header must be a mapping")
-    return dict(parsed)
+    return parse_header_values(
+        header=inner,
+        file_path=Path(_DESCRIPTION_SENTINEL_PATH),
+        statement_name=kind,
+    )
 
 
 def _lint_header_whitespace(

{sqlbuild-0.57.0 → sqlbuild-0.58.0}/src/sqlbuild/lint/constants.py

@@ -16,6 +16,7 @@ HEADER_KIND_FUNCTION: str = "FUNCTION"
 HEADER_KIND_ENUM: str = "ENUM"
 HEADER_KIND_CONSTANT: str = "CONSTANT"
 HEADER_KIND_HOOK: str = "HOOK"
+HEADER_KIND_SCHEMA: str = "SCHEMA"
 
 DSL_HEADER_KINDS: frozenset[str] = frozenset(
     {
@@ -27,6 +28,7 @@ DSL_HEADER_KINDS: frozenset[str] = frozenset(
         HEADER_KIND_ENUM,
         HEADER_KIND_CONSTANT,
         HEADER_KIND_HOOK,
+        HEADER_KIND_SCHEMA,
     }
 )
 
@@ -46,6 +48,7 @@ LINT_DIRECTORY_NAMES: tuple[str, ...] = (
     "enums",
     "constants",
     "hooks",
+    "schemas",
 )
 
 DEFAULT_SQRUFF_CONFIG_CONTENT: str = """\