Tweaked option name.

Author: sbyrne-sf

src/datacustomcode/cli.py

@@ -145,8 +145,10 @@ def deploy(
 
 @cli.command()
 @click.argument("directory", default=".")
-@click.option("--type", default="script", type=click.Choice(["script", "function"]))
-def init(directory: str, type: str):
+@click.option(
+    "--code-type", default="script", type=click.Choice(["script", "function"])
+)
+def init(directory: str, code_type: str):
     from datacustomcode.scan import (
         dc_config_json_from_file,
         update_config,
@@ -155,18 +157,18 @@ def init(directory: str, type: str):
     from datacustomcode.template import copy_function_template, copy_script_template
 
     click.echo("Copying template to " + click.style(directory, fg="blue", bold=True))
-    if type == "script":
+    if code_type == "script":
         copy_script_template(directory)
-    elif type == "function":
+    elif code_type == "function":
         copy_function_template(directory)
     entrypoint_path = os.path.join(directory, "payload", "entrypoint.py")
     config_location = os.path.join(os.path.dirname(entrypoint_path), "config.json")
 
     # Write package type to SDK-specific config
-    sdk_config = {"type": type}
+    sdk_config = {"type": code_type}
     write_sdk_config(directory, sdk_config)
 
-    config_json = dc_config_json_from_file(entrypoint_path, type)
+    config_json = dc_config_json_from_file(entrypoint_path, code_type)
     with open(config_location, "w") as f:
         json.dump(config_json, f, indent=2)
 

tests/test_cli.py

@@ -41,7 +41,7 @@ def test_init_command(
             # Create test directory structure
             os.makedirs(os.path.join("test_dir", "payload"), exist_ok=True)
 
-            result = runner.invoke(init, ["test_dir", "--type", "script"])
+            result = runner.invoke(init, ["test_dir", "--code-type", "script"])
 
             assert result.exit_code == 0
             mock_copy.assert_called_once_with("test_dir")