From 3b913b2ca6dde7e3451ebb6bfb65bf92457babef Mon Sep 17 00:00:00 2001
From: michael-corey <mcorey@americafirstpolicy.com>
Date: Sat, 18 Apr 2026 12:37:22 -0500
Subject: [PATCH] adding user prompt for db creds

---
 generic_loader/.gitignore     |  3 ++-
 generic_loader/load_folder.py | 16 +++++++++++++++-
 generic_loader/load_sas.py    | 29 +++++++++++++++++++++++++----
 3 files changed, 42 insertions(+), 6 deletions(-)

diff --git a/generic_loader/.gitignore b/generic_loader/.gitignore
index 00d5b98..a54db63 100644
--- a/generic_loader/.gitignore
+++ b/generic_loader/.gitignore
@@ -1,4 +1,5 @@
 /.venv
 /samples
 /.env
-/__pycache__
\ No newline at end of file
+/__pycache__
+/venv
\ No newline at end of file
diff --git a/generic_loader/load_folder.py b/generic_loader/load_folder.py
index baecba9..099c419 100644
--- a/generic_loader/load_folder.py
+++ b/generic_loader/load_folder.py
@@ -95,6 +95,7 @@ Exit codes:
 from __future__ import annotations
 
 import argparse
+import getpass
 import re
 import sys
 from dataclasses import dataclass, field
@@ -463,6 +464,14 @@ def _build_argparser() -> argparse.ArgumentParser:
             "cluster back and continue with the next one."
         ),
     )
+    p.add_argument(
+        "--dbcreds",
+        action="store_true",
+        help=(
+            "Prompt for database username and password instead of reading "
+            "PGUSER / PGPASSWORD from the environment or .env file."
+        ),
+    )
     return p
 
 
@@ -512,7 +521,12 @@ def main(argv: Optional[List[str]] = None) -> int:
             print()
         return 0
 
-    conn = connect()
+    db_user = db_password = None
+    if args.dbcreds:
+        db_user = input("Database username: ")
+        db_password = getpass.getpass("Database password: ")
+
+    conn = connect(user=db_user, password=db_password)
     conn.autocommit = False
     failures: List[Tuple[str, Exception]] = []
     totals: List[Tuple[str, int, int]] = []  # (tablename, files, rows)
diff --git a/generic_loader/load_sas.py b/generic_loader/load_sas.py
index b514917..0be9602 100644
--- a/generic_loader/load_sas.py
+++ b/generic_loader/load_sas.py
@@ -207,6 +207,7 @@ from __future__ import annotations
 
 import argparse
 import datetime as dt
+import getpass
 import io
 import json
 import os
@@ -308,18 +309,25 @@ class ValidationError(RuntimeError):
 # ---------------------------------------------------------------------------
 
 
-def connect() -> psycopg2.extensions.connection:
+def connect(
+    *,
+    user: Optional[str] = None,
+    password: Optional[str] = None,
+) -> psycopg2.extensions.connection:
     """Open a psycopg2 connection using standard libpq env vars.
 
     Assumes `.env` has already been loaded (the CLI does this before calling).
     Orchestrators that wrap this module should either call ``load_dotenv()``
     themselves or ensure the env vars are set.
+
+    ``user`` and ``password`` override the corresponding env vars when supplied
+    (used by the ``--dbcreds`` CLI flag to accept interactive input).
     """
     conn = psycopg2.connect(
         host=os.environ.get("PGHOST"),
         port=os.environ.get("PGPORT"),
-        user=os.environ.get("PGUSER"),
-        password=os.environ.get("PGPASSWORD"),
+        user=user or os.environ.get("PGUSER"),
+        password=password or os.environ.get("PGPASSWORD"),
         dbname=os.environ.get("PGDATABASE"),
     )
     return conn
@@ -1150,6 +1158,14 @@ def _build_argparser() -> argparse.ArgumentParser:
         action="store_true",
         help="Print inferred CREATE TABLE and stop; don't touch Postgres.",
     )
+    p.add_argument(
+        "--dbcreds",
+        action="store_true",
+        help=(
+            "Prompt for database username and password instead of reading "
+            "PGUSER / PGPASSWORD from the environment or .env file."
+        ),
+    )
     return p
 
 
@@ -1208,7 +1224,12 @@ def main(argv: Optional[List[str]] = None) -> int:
             print(f"  streaming... {seen:,} rows", file=sys.stderr)
             yield chunk_df
 
-    conn = connect()
+    db_user = db_password = None
+    if args.dbcreds:
+        db_user = input("Database username: ")
+        db_password = getpass.getpass("Database password: ")
+
+    conn = connect(user=db_user, password=db_password)
     conn.autocommit = False
     try:
         create_table(conn, cfg.schemaname, cfg.tablename, columns, cfg.if_exists)