Refactor import_to_supabase function to enhance logging, error handling, and parameter support

Author: Domains18

cloudsql_to_supabase/import_.py

@@ -1,14 +1,50 @@
-from . import config, utils
 import os
+import logging
+from pathlib import Path
+from typing import Optional
+from . import config, utils
+
+
+
+logger = logging.getLogger('cloudsql_to_supabase.import')
 
 
 
-def import_to_supabase():
-    print("\n Importing into Supabase")
+def import_to_supabase(input_file: Optional[Path] = None, password: Optional[str] = None) -> None:
+    """
+        import cleaned sql dump into supabase
+        
+        Args:
+            input_file: Path to the sql dump file to import. if None, uses the default.
+            password: supabase database password. if none uses from config
+            
+    """
+    config.validate_config()
+    dump_file = input_file or Path(config.CLEANED_DUMP)
+    if not dump_file.exists():
+        raise FileNotFoundError(f"Dump not found: {dump_file}")
+    
+    logger.info(f'importing data into supabase database: {config.SUPABASE_DB}')
+    
     env = os.environ.copy()
-    env["PGPASSWORD"] = config.SUPABASE_PASSWORD
+    env['PGPASSWORD'] = password or config.SUPABASE_PASSWORD
+    
+    
     cmd = (
-        f"psql -h {config.SUPABASE_HOST} -p {config.SUPABASE_PORT} -U {config.SUPABASE_USER} "
-        f"-d {config.SUPABASE_DB} -f {config.CLEANED_DUMP} \"sslmode=require\""
+        f"psql -h {config.SUPABASE_HOST} "
+        f"-p {config.SUPABASE_PORT} "
+        f"-U {config.SUPABASE_USER} "
+        f"-d {config.SUPABASE_DB} "
+        f"--set ON_ERROR_STOP=on "  # Stop on first error
+        f"--single-transaction "     # Run as a single transaction
+        f"--sslmode={config.SUPABASE_SSL_MODE} "
+        f"-f {dump_file}"
     )
-    utils.run_command(cmd, env=env)
+    
+    
+    try:
+        utils.run_command(cmd, env)
+        logger.info("import completed successfully")
+    except Exception as e:
+        logger.error(f'import failed: {e}')
+        raise