command line option for profile

Author: joroscoSF

src/datacustomcode/cli.py

@@ -193,7 +193,8 @@ def scan(filename: str, config: str, dry_run: bool, no_requirements: bool):
 @click.argument("entrypoint")
 @click.option("--config-file", default=None)
 @click.option("--dependencies", default=[], multiple=True)
-def run(entrypoint: str, config_file: Union[str, None], dependencies: List[str]):
+@click.option("--profile", default="default")
+def run(entrypoint: str, config_file: Union[str, None], dependencies: List[str], profile: str):
     from datacustomcode.run import run_entrypoint
 
-    run_entrypoint(entrypoint, config_file, dependencies)
+    run_entrypoint(entrypoint, config_file, dependencies, profile)

src/datacustomcode/run.py

@@ -20,15 +20,22 @@
 
 
 def run_entrypoint(
-    entrypoint: str, config_file: Union[str, None], dependencies: List[str]
+    entrypoint: str, config_file: Union[str, None], dependencies: List[str], profile: str
 ) -> None:
     """Run the entrypoint script with the given config and dependencies.
 
     Args:
         entrypoint: The entrypoint script to run.
         config_file: The config file to use.
         dependencies: The dependencies to import.
+        profile: The profile to use.
     """
+    if profile != "default":
+        if config.reader_config and hasattr(config.reader_config, 'options'):
+            config.reader_config.options["credentials_profile"] = profile
+        if config.writer_config and hasattr(config.writer_config, 'options'):
+            config.writer_config.options["credentials_profile"] = profile
+    
     if config_file:
         config.load(config_file)
     for dependency in dependencies: