sqlmesh.core.config.categorizer

 1from __future__ import annotations
 2
 3from enum import Enum
 4
 5from sqlmesh.core.config.base import BaseConfig
 6
 7
 8class AutoCategorizationMode(Enum):
 9    FULL = "full"
10    """Full-auto mode in which the categorizer falls back to the most conservative choice (breaking)."""
11
12    SEMI = "semi"
13    """Semi-auto mode in which a user is promted to provide a category in case when the categorizer
14    failed to determine it automatically.
15    """
16
17    OFF = "off"
18    """Disables automatic categorization."""
19
20
21class CategorizerConfig(BaseConfig):
22    """Configuration for the automatic categorizer of snapshot changes.
23
24    Args:
25        external: the auto categorization mode for External models.
26        python: the auto categorization mode for Python models.
27        sql: the auto categorization mode for SQL models.
28        seed: the auto categorization mode for Seed models.
29    """
30
31    external: AutoCategorizationMode = AutoCategorizationMode.FULL
32    python: AutoCategorizationMode = AutoCategorizationMode.OFF
33    sql: AutoCategorizationMode = AutoCategorizationMode.FULL
34    seed: AutoCategorizationMode = AutoCategorizationMode.FULL
35
36    @classmethod
37    def all_off(cls) -> CategorizerConfig:
38        return cls(
39            external=AutoCategorizationMode.OFF,
40            python=AutoCategorizationMode.OFF,
41            sql=AutoCategorizationMode.OFF,
42            seed=AutoCategorizationMode.OFF,
43        )
44
45    @classmethod
46    def all_semi(cls) -> CategorizerConfig:
47        return cls(
48            external=AutoCategorizationMode.SEMI,
49            python=AutoCategorizationMode.SEMI,
50            sql=AutoCategorizationMode.SEMI,
51            seed=AutoCategorizationMode.SEMI,
52        )
53
54    @classmethod
55    def all_full(cls) -> CategorizerConfig:
56        return cls(
57            external=AutoCategorizationMode.FULL,
58            python=AutoCategorizationMode.FULL,
59            sql=AutoCategorizationMode.FULL,
60            seed=AutoCategorizationMode.FULL,
61        )