Coverage for src/chuck_data/commands/__init__.py: 0%
37 statements
« prev ^ index » next coverage.py v7.8.0, created at 2025-06-05 22:56 -0700
« prev ^ index » next coverage.py v7.8.0, created at 2025-06-05 22:56 -0700
1"""
2Command registration for Chuck.
4This module registers all commands with the unified command registry,
5making them available for both CLI and agent usage.
6"""
8from ..command_registry import register_command
10# Import command definitions
11from .models import DEFINITION as models_definition
12from .list_models import DEFINITION as list_models_definition
13from .model_selection import DEFINITION as model_selection_definition
14from .catalog_selection import DEFINITION as catalog_selection_definition
15from .schema_selection import DEFINITION as schema_selection_definition
16from .tag_pii import DEFINITION as tag_pii_definition
17from .scan_pii import DEFINITION as scan_pii_definition
18from .setup_stitch import DEFINITION as setup_stitch_definition
19from .workspace_selection import DEFINITION as workspace_selection_definition
20from .help import DEFINITION as help_definition
21from .status import DEFINITION as status_definition
22from .jobs import DEFINITION as jobs_definition
23from .setup_wizard import DEFINITION as setup_wizard_definition
24from .agent import DEFINITION as agent_definition
25from .auth import DEFINITION as auth_definition
27# Import new Databricks commands
28from .list_warehouses import DEFINITION as list_warehouses_definition
29from .warehouse import DEFINITION as warehouse_definition
30from .warehouse_selection import DEFINITION as warehouse_selection_definition
31from .create_warehouse import DEFINITION as create_warehouse_definition
32from .run_sql import DEFINITION as run_sql_definition
33from .list_volumes import DEFINITION as list_volumes_definition
34from .create_volume import DEFINITION as create_volume_definition
35from .upload_file import DEFINITION as upload_file_definition
36from .job_status import DEFINITION as job_status_definition
37from .list_tables import DEFINITION as list_tables_definition
38from .add_stitch_report import DEFINITION as add_stitch_report_definition
40# Import catalog/schema/table commands
41from .list_catalogs import DEFINITION as list_catalogs_definition
42from .catalog import DEFINITION as catalog_definition
43from .list_schemas import DEFINITION as list_schemas_definition
44from .schema import DEFINITION as schema_definition
45from .table import DEFINITION as table_definition
47# Import bug report command
48from .bug import DEFINITION as bug_definition
50# List of all command definitions to register
51ALL_COMMAND_DEFINITIONS = [
52 # Authentication & Workspace commands
53 *auth_definition,
54 workspace_selection_definition,
55 setup_wizard_definition,
56 # Model related commands
57 models_definition,
58 list_models_definition,
59 model_selection_definition,
60 # Catalog & Schema commands
61 catalog_selection_definition,
62 schema_selection_definition,
63 list_catalogs_definition,
64 catalog_definition,
65 list_schemas_definition,
66 schema_definition,
67 list_tables_definition,
68 table_definition,
69 # PII and Stitch related commands
70 tag_pii_definition,
71 scan_pii_definition,
72 setup_stitch_definition,
73 add_stitch_report_definition,
74 # Job commands
75 *jobs_definition,
76 job_status_definition,
77 # Warehouse commands
78 list_warehouses_definition,
79 warehouse_definition,
80 warehouse_selection_definition,
81 create_warehouse_definition,
82 run_sql_definition,
83 # Volume commands
84 list_volumes_definition,
85 create_volume_definition,
86 upload_file_definition,
87 # Utility commands
88 help_definition,
89 status_definition,
90 bug_definition,
91 # Agent command
92 agent_definition,
93]
96def register_all_commands():
97 """Register all commands with the unified registry."""
98 for definition in ALL_COMMAND_DEFINITIONS:
99 register_command(definition)