通过Python脚本执行psql命令，包含连接字符串和输入重定向

import subprocess
import os

# 假设 conf 模块已定义数据库连接信息
# 示例配置类，实际项目中应从安全配置中加载
class Config:
    login = "your_user"
    password = "your_password"
    host = "localhost"
    port = "5432"
conf = Config()

# 定义 psql.exe 和备份文件的路径
# 实际项目中，这些路径应更具鲁棒性，例如使用 os.path.join
commandlet = os.path.abspath(r"psql.exe") # 假设 psql.exe 在当前或可访问路径
backup_file = os.path.abspath(r"backup.sql") # 假设 backup.sql 在当前或可访问路径

# 构建数据库连接字符串
user = conf.login
password = conf.password
host = conf.host
port = conf.port
con_str = f"postgresql://{user}:{password}@{host}:{port}/"

# 尝试执行命令（不正确的方式）
def main_incorrect():
    # 方式一：将所有参数作为元组元素传递
    # subprocess.check_call((commandlet, con_str, "<", backup_file)) # 这里的 "<" 会被当做普通参数
    # 方式二：将整个命令作为字符串传递，但默认 shell=False
    # subprocess.check_call(f"{commandlet} {con_str} < {backup_file}") # 同样会失败

    print("尝试使用不正确的方式执行 psql 命令，这将无法正确解析重定向符。")
    # 为了避免实际执行错误，这里不运行上述代码，仅作说明。
    # 实际运行时，psql.exe 会启动，但不会从 backup.sql 读取输入。

if __name__ == "__main__":
    main_incorrect()

import subprocess
import os

# 假设 conf 模块已定义数据库连接信息
class Config:
    login = "your_user"
    password = "your_password"
    host = "localhost"
    port = "5432"
conf = Config()

# 定义 psql.exe 和备份文件的路径
# 实际项目中，这些路径应更具鲁棒性，例如使用 os.path.join
commandlet = os.path.abspath(r"psql.exe") # 假设 psql.exe 在当前或可访问路径
backup_file = os.path.abspath(r"backup.sql") # 假设 backup.sql 在当前或可访问路径

# 构建数据库连接字符串
user = conf.login
password = conf.password
host = conf.host
port = conf.port
con_str = f"postgresql://{user}:{password}@{host}:{port}/"

def restore_database_correct():
    """
    使用 psql.exe 恢复数据库，通过 Python 脚本执行。
    正确处理输入重定向。
    """
    try:
        # 为了示例可运行，创建一个虚拟的 backup.sql 文件
        if not os.path.exists(backup_file):
            print(f"创建虚拟备份文件：{backup_file}")
            with open(backup_file, "w") as f:
                f.write("-- This is a dummy SQL backup file for testing\n")
                f.write("SELECT 1;\n")
                f.write("CREATE TABLE IF NOT EXISTS test_table (id INT);\n")
                f.write("INSERT INTO test_table (id) VALUES (100);\n")

        print(f"尝试使用正确的方式执行 psql 命令：从 {backup_file} 恢复到 {con_str}")
        # 正确的命令结构：将命令、连接字符串、重定向符和文件路径作为单独的元素传递
        # 并设置 shell=True 以便系统shell解析重定向符
        subprocess.check_call((commandlet, con_str, "<", backup_file), shell=True)
        print(f"数据库恢复成功：从 {backup_file} 到 {con_str}")

    except subprocess.CalledProcessError as e:
        print(f"执行 psql 命令失败：{e}")
        # 如果命令执行失败，e.stderr 会包含错误信息
        if e.stderr:
            print(f"错误输出：{e.stderr.decode('utf-8', errors='ignore')}")
        if e.stdout:
            print(f"标准输出：{e.stdout.decode('utf-8', errors='ignore')}")
    except FileNotFoundError:
        print(f"错误：psql.exe 或 {backup_file} 未找到。请检查路径是否正确。")
    except Exception as e:
        print(f"发生未知错误：{e}")

if __name__ == "__main__":
    restore_database_correct()