Python脚本中执行psql.exe并处理I/O重定向

psql.exe postgresql://user:pass@host:port/ < backup.sql

import subprocess
import os

# --- 模拟配置信息，实际使用时请替换为您的配置源 ---
class Config:
    login = "your_user"
    password = "your_password"
    host = "localhost"
    port = "5432"
conf = Config()
# --- 模拟配置信息结束 ---

# 确保 psql.exe 和 SQL 文件路径正确
# 这里的路径是相对当前脚本的父目录，请根据实际情况调整
# 假设 psql.exe 在当前脚本的父目录
psql_commandlet = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "psql.exe"))
# 假设 backup.sql 在当前脚本的父目录
backup_file_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "backup.sql"))

# 构建 PostgreSQL 连接字符串
user = conf.login
password = conf.password
host = conf.host
port = conf.port
connection_string = f"postgresql://{user}:{password}@{host}:{port}/your_database_name" # 记得替换数据库名

def run_psql_restore():
    """
    使用 subprocess 模块执行 psql.exe 命令，并处理文件输入重定向。
    """
    print(f"尝试执行 psql 命令到: {connection_string}")
    print(f"将从文件 '{backup_file_path}' 读取输入。")

    try:
        # 核心解决方案：
        # 1. 将命令、连接字符串、重定向符号和文件路径作为单独的元素放入元组。
        # 2. 设置 shell=True，让系统 shell 解析重定向符号 '<'。
        # 当 shell=True 且 cmd 参数为序列时，subprocess 会将序列元素用空格连接成一个字符串
        # 然后将该字符串传递给系统 shell 执行。
        subprocess.check_call(
            (psql_commandlet, connection_string, "<", backup_file_path),
            shell=True
        )
        print("\npsql.exe 命令执行成功，数据库恢复完成。")
    except subprocess.CalledProcessError as e:
        print(f"\n错误：psql.exe 命令执行失败，返回码: {e.returncode}")
        # 如果需要捕获标准错误输出，可以使用 subprocess.run 并设置 capture_output=True
        # 例如: result = subprocess.run(..., capture_output=True, text=True)
        # print(f"错误输出:\n{result.stderr}")
    except FileNotFoundError:
        print(f"\n错误：无法找到 psql.exe 或 SQL 文件。请检查以下路径：")
        print(f"  psql.exe 路径: {psql_commandlet}")
        print(f"  SQL 文件路径: {backup_file_path}")
    except Exception as e:
        print(f"\n执行过程中发生未知错误: {e}")

if __name__ == "__main__":
    # --- 模拟创建 backup.sql 文件用于测试 ---
    # 在实际应用中，此文件应已存在
    if not os.path.exists(backup_file_path):
        print(f"创建模拟 SQL 文件: {backup_file_path}")
        with open(backup_file_path, "w") as f:
            f.write("CREATE TABLE IF NOT EXISTS test_table (id INT, name VARCHAR(50));\n")
            f.write("INSERT INTO test_table (id, name) VALUES (1, 'Test User');\n")
            f.write("SELECT 'Simulated SQL execution complete.';\n")
    # --- 模拟文件创建结束 ---

    run_psql_restore()

import subprocess
import os

# 假设 psql_commandlet 和 connection_string 已定义
# ... (同上文代码中的定义) ...

def run_psql_restore_safe():
    print(f"尝试执行 psql 命令 (安全模式) 到: {connection_string}")
    print(f"将从文件 '{backup_file_path}' 读取输入。")

    try:
        with open(backup_file_path, 'r') as sql_file:
            # 不使用 shell=True，通过 stdin 参数直接传递文件对象
            subprocess.check_call(
                [psql_commandlet, connection_string], # 命令和参数列表
                stdin=sql_file # 将文件对象作为标准输入
            )
        print("\npsql.exe 命令执行成功（安全模式）。")
    except subprocess.CalledProcessError as e:
        print(f"\n错误：psql.exe 命令执行失败，返回码: {e.returncode}")
    except FileNotFoundError:
        print(f"\n错误：无法找到 psql.exe 或 SQL 文件。")
    except Exception as e:
        print(f"\n执行过程中发生未知错误: {e}")

# if __name__ == "__main__":
#     # ... (模拟文件创建) ...
#     run_psql_restore_safe()