chore: update sample script to accept positional argument for connector name (#658)

Co-authored-by: octavia-squidington-iii <[email protected]>

Author: aaronsteers

examples/run_gsm_connector_secret_fetch.py

@@ -10,18 +10,21 @@
 exists and is excluded from git before running the script.
 
 Usage:
-    poetry run python examples/run_gsm_secret_fetch.py
+    poetry run python examples/run_gsm_connector_secret_fetch.py
+    poetry run python examples/run_gsm_connector_secret_fetch.py <connector_name>
+    poetry run python examples/run_gsm_connector_secret_fetch.py source-github
 """
 
 from __future__ import annotations
 
+import sys
 from pathlib import Path
 
 import airbyte as ab
 from airbyte.secrets import GoogleGSMSecretManager, SecretHandle
 
 AIRBYTE_INTERNAL_GCP_PROJECT = "dataline-integration-testing"
-CONNECTOR_NAME = "source-s3"
+CONNECTOR_NAME = sys.argv[1] if len(sys.argv) > 1 else "source-klaviyo"
 
 AIRBYTE_REPO_ROOT = Path(__file__).parent.parent.parent / "airbyte"
 
@@ -33,6 +36,10 @@
     / CONNECTOR_NAME
     / "secrets"
 )
+if not AIRBYTE_REPO_ROOT.exists():
+    raise FileNotFoundError(f"Airbyte repo root does not exist: {AIRBYTE_REPO_ROOT}")
+if not CONNECTOR_SECRETS_DIR.exists():
+    CONNECTOR_SECRETS_DIR.mkdir(parents=True, exist_ok=True)
 
 
 def main() -> None:
@@ -42,12 +49,14 @@ def main() -> None:
     )
 
     secret: SecretHandle
-    for secret in secret_mgr.fetch_connector_secrets("source-s3"):
+    for secret in secret_mgr.fetch_connector_secrets(
+        connector_name=CONNECTOR_NAME,
+    ):
+        filename_base = "config"  # Default filename if not overridden
         if "filename" in secret.labels:
-            secret_file_path = (
-                CONNECTOR_SECRETS_DIR / f"{secret.labels['filename']}.json"
-            )
-            secret.write_to_file(secret_file_path)
+            filename_base = secret.labels["filename"]
+        secret_file_path = CONNECTOR_SECRETS_DIR / f"{filename_base}.json"
+        secret.write_to_file(secret_file_path)
 
 
 if __name__ == "__main__":