Added restore_backup.py restore_postgres_databases.py

restore_postgres_databases.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+"""
+Restore multiple PostgreSQL databases from .backup.sql files via a Docker container.
+
+Usage:
+  ./restore_databases.py /path/to/backup_dir [--container central-postgres]
+"""
+import argparse
+import subprocess
+import sys
+import os
+import glob
+
+def run_command(cmd, input_data=None):
+    """
+    Run a subprocess command and exit on failure.
+    :param cmd: list of command parts
+    :param input_data: bytes to send to process stdin
+    """
+    try:
+        subprocess.run(cmd, input=input_data, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error running command: {' '.join(cmd)}", file=sys.stderr)
+        sys.exit(e.returncode)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Restore Postgres databases from backup SQL files via Docker container."
+    )
+    parser.add_argument(
+        "backup_dir",
+        help="Path to directory containing .backup.sql files"
+    )
+    parser.add_argument(
+        "container",
+        help="Name of the Postgres Docker container"
+    )
+    args = parser.parse_args()
+
+    backup_dir = args.backup_dir
+    container = args.container
+
+    pattern = os.path.join(backup_dir, "*.backup.sql")
+    sql_files = sorted(glob.glob(pattern))
+    if not sql_files:
+        print(f"No .backup.sql files found in {backup_dir}", file=sys.stderr)
+        sys.exit(1)
+
+    for sqlfile in sql_files:
+        dbname = os.path.splitext(os.path.basename(sqlfile))[0]
+        print(f"=== Processing {sqlfile} → database: {dbname} ===")
+
+        # Drop the database if it already exists
+        run_command([
+            "docker", "exec", "-i", container,
+            "psql", "-U", "postgres", "-c",
+            f"DROP DATABASE IF EXISTS \"{dbname}\";"
+        ])
+
+        # Create a fresh database
+        run_command([
+            "docker", "exec", "-i", container,
+            "psql", "-U", "postgres", "-c",
+            f"CREATE DATABASE \"{dbname}\";"
+        ])
+
+        # Restore the dump into the newly created database
+        print(f"Restoring dump into {dbname}…")
+        with open(sqlfile, "rb") as f:
+            sql_data = f.read()
+        run_command([
+            "docker", "exec", "-i", container,
+            "psql", "-U", "postgres", "-d", dbname
+        ], input_data=sql_data)
+
+        print(f"✔ {dbname} restored.")
+
+    print("All databases have been restored.")
+
+
+if __name__ == "__main__":
+    main()