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

1""" 

2Command registration for Chuck. 

3 

4This module registers all commands with the unified command registry, 

5making them available for both CLI and agent usage. 

6""" 

7 

8from ..command_registry import register_command 

9 

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 

26 

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 

39 

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 

46 

47# Import bug report command 

48from .bug import DEFINITION as bug_definition 

49 

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] 

94 

95 

96def register_all_commands(): 

97 """Register all commands with the unified registry.""" 

98 for definition in ALL_COMMAND_DEFINITIONS: 

99 register_command(definition)