【Postgres_Python】使用python脚本批量创建和导入多个PG数据库

之前批量创建和导入数据库分为2个python脚本进行,现整合优化代码合并为一个python脚本,可同步实现数据库的创建和数据导入。之前的文章链接:

【Postgres_Python】使用python脚本批量创建PG数据库
【Postgres_Python】使用python脚本将多个.SQL文件批量导入不同的PG数据库

示例代码说明:

有多个.sql格式的数据库需要导入,数据库名与文件名一致。 先判断服务器中数据库是否存在,若存在则删除,创建数据库,导入SQL文件内容

复制代码
import psycopg2
import subprocess
import os
folder_path = 'E:/Dataset/chongqinghq/'
# 获取文件夹下所有文件和文件夹的名称
filenames = [f for f in os.listdir(folder_path) if os.path.isfile(os.path.join(folder_path, f))]
# 遍历文件列表并以.提取名称
for filename in filenames:
    name_parts = filename.split('.')
    if name_parts:
        # 提取第一部分作为提取的名称,创建的数据库名称
        database_name = name_parts[0]
        username = 'postgres'
        password = 'postgres'
        host = 'localhost'
        port = '5432'
        output_file = folder_path + filename
        os.environ['PGPASSWORD'] = password
        try:
            #数据库存在则删除数据库
            commanddelete = f'D:/Program Files/PostgreSQL/10/bin/psql -h {host} -p {port} -U {username}  -c "DROP DATABASE IF EXISTS {database_name}"'
            subprocess.run(commanddelete, check=True)
            print(f"Database {database_name} dropped successfully.")
            # 创建数据库
            commandcreate = f'D:/Program Files/PostgreSQL/10/bin/psql -h {host} -p {port} -U {username}  -c "create database {database_name}"'
            subprocess.run(commandcreate, check=True)
            print(f"Database {database_name} successfully create")
            # 导入数据库
            commandimport = f'D:/Program Files/PostgreSQL/10/bin/psql -h {host} -p {port} -U {username} -d {database_name} -f {output_file}'
            subprocess.run(commandimport, check=True)
            print(f"Database {database_name} on host {host} and port {port} successfully created and imported")
        finally:
            # 清除环境变量以防止密码泄露
            del os.environ['PGPASSWORD']

结果示例:

相关推荐
看天走路吃雪糕28 分钟前
墨者:SQL注入漏洞测试(布尔盲注)
数据库·sql·sqlmap·墨者学院·布尔盲注
*愿风载尘*28 分钟前
ksql连接数据库免输入密码交互
数据库·后端
追风少年浪子彦30 分钟前
mybatis-plus实体类主键生成策略
java·数据库·spring·mybatis·mybatis-plus
溟洵34 分钟前
Qt 窗口 工具栏QToolBar、状态栏StatusBar
开发语言·前端·数据库·c++·后端·qt
problc1 小时前
大模型API和秘钥获取地址
数据库·redis·缓存
Antonio9151 小时前
【Redis】Linux 配置Redis
linux·数据库·redis
qq_463944862 小时前
如何把Excel文件导入Navicat?
数据库·excel
不太厉害的程序员2 小时前
Excel 将数据导入到SQLServer数据库
数据库·sqlserver·excel
betazhou4 小时前
MySQL ROUTER安装部署
android·数据库·mysql·adb·mgr·mysql router
中东大鹅4 小时前
Mybatis Plus 多数据源
java·数据库·spring boot·后端·mybatis