云服务导入数据库

import psycopg2
import subprocess

# 导出数据
def export_data(source_db_config, output_file):
    conn = psycopg2.connect(**source_db_config)
    with open(output_file, 'w') as f:
        subprocess.run(['pg_dump', '-U', source_db_config['user'], '-h', source_db_config['host'], '-p', str(source_db_config['port']), '-f', output_file, source_db_config['dbname']])
    conn.close()

# 导入数据
def import_data(target_db_config, input_file):
    conn = psycopg2.connect(**target_db_config)
    with open(input_file, 'r') as f:
        subprocess.run(['psql', '-U', target_db_config['user'], '-h', target_db_config['host'], '-p', str(target_db_config['port']), '-d', target_db_config['dbname']], stdin=f)
    conn.close()

# 示例配置
source_db_config = {
    'user': 'source_user',
    'password': 'source_password',
    'host': 'source_host',
    'port': 5432,
    'dbname': 'source_db'
}

target_db_config = {
    'user': 'target_user',
    'password': 'target_password',
    'host': 'target_host',
    'port': 5432,
    'dbname': 'target_db'
}

output_file = 'exported_data.sql'

# 执行导出和导入
export_data(source_db_config, output_file)
import_data(target_db_config, output_file)