Skip to content

Commit aed7282

Browse files
author
Arik Kfir
committed
Support custom context for post-processed SQL scripts.
1 parent 7e2cc7c commit aed7282

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

resources/src/gcp_cloud_sql.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -571,6 +571,7 @@ def __init__(self, data: dict, svc: ExternalServices = ExternalServices()) -> No
571571
}
572572
}
573573
},
574+
"scripts_ctx": {"type": "object"},
574575
"scripts": {
575576
"type": "array",
576577
"items": {
@@ -900,7 +901,7 @@ def get_actions_for_discovered_state(self, state: dict) -> Sequence[DAction]:
900901
root_password=cfg['root-password'],
901902
zone=zone,
902903
scripts_data=self.info.config['scripts'],
903-
context=self.info.config)
904+
context=cfg['scripts_ctx'] if 'scripts_ctx' in cfg else {})
904905
with evaluator as evaluator:
905906
for script in evaluator.get_scripts_to_execute():
906907
actions.append(
@@ -985,7 +986,7 @@ def create_sql_instance(self, args) -> None:
985986
root_password=cfg['root-password'],
986987
zone=cfg['zone'],
987988
scripts_data=self.info.config['scripts'],
988-
context=self.info.config)
989+
context=cfg['scripts_ctx'] if 'scripts_ctx' in cfg else {})
989990
with evaluator as evaluator:
990991
evaluator.execute_scripts(scripts=evaluator.get_scripts_to_execute())
991992

@@ -998,7 +999,7 @@ def execute_scripts(self, args) -> None:
998999
root_password=cfg['root-password'],
9991000
zone=cfg['zone'],
10001001
scripts_data=self.info.config['scripts'],
1001-
context=self.info.config)
1002+
context=cfg['scripts_ctx'] if 'scripts_ctx' in cfg else {})
10021003
with evaluator as evaluator:
10031004
scripts: Sequence[Script] = [evaluator.get_script(script_name) for script_name in args.scripts]
10041005
evaluator.execute_scripts(scripts=scripts)

0 commit comments

Comments
 (0)