python访问pg数据库，并将查询结果保存到文件（txt，word，Excel）_python pg

（1）Python所有方向的学习路线（新版）

这是我花了几天的时间去把Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

最近我才对这些路线做了一下新的更新，知识体系更全面了。

（2）Python学习视频

包含了Python入门、爬虫、数据分析和web开发的学习视频，总共100多个，虽然没有那么全面，但是对于入门来说是没问题的，学完这些之后，你可以按照我上面的学习路线去网上找其他的知识资源进行进阶。

（3）100多个练手项目

我们在看视频学习的时候，不能光动眼动脑不动手，比较科学的学习方法是在理解之后运用它们，这时候练手项目就很适合了，只是里面的项目比较多，水平也是参差不齐，大家可以挑自己能做的项目去练练。

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

• 前言
• 一、安装pg数据库驱动程序
• 二、使用步骤
• • 1.引入库
    • 2.连接pg数据库，执行操作语句
    • 3.将查询结果保存到文件中
    • • • curPath获取当前文件的绝对路径
          • rootPath为获取当前文件所在目录的父级目录路径
        • 保存为txt文件
        • 保存为Excel 文件
        • 保存为Word 文件
• 总结

---

前言

例如：学习locust时，需要读取数据库的用户信息，涉及多个环境，手动一个环境一个环境数据库查询后复制到文件里太麻烦。

---

一、安装pg数据库驱动程序

首先，你需要确定你要连接的数据库类型，并安装相应的数据库驱动程序。常见的数据库驱动程序包括psycopg2（PostgreSQL），mysql-connector-python（MySQL），pyodbc（Microsoft SQL Server），cx_Oracle（Oracle）等。你可以使用Python的包管理器（如pip）来安装这些驱动程序。

pip install psycopg2

二、使用步骤

1.引入库

代码如下（示例）：

import psycopg2  # 导入psycopg2驱动程序

2.连接pg数据库，执行操作语句

使用数据库驱动程序提供的函数来建立与数据库的连接。你需要提供数据库的连接参数，例如主机名、端口号、用户名、密码等。
具体的连接参数取决于你要连接的数据库类型
这里我读取的是单独建的一个字典文件config中的配置
连接pg数据库,执行查询代码如下：

def pg\_sql(sql):
    result = []
    conn = None
    cur = None
    try:
        # 连接到数据库
        conn = psycopg2.connect(
            host=config.DB["host"],
            port=config.DB["port"],
            database=config.DB["database"],
            user=config.DB["user"],
            password=config.DB["password"]
        )

执行SQL查询：一旦连接成功，你可以使用conn对象来执行SQL查询并获取结果。

        cur = conn.cursor()  # 创建游标
        cur.execute(sql)  # 执行语句
        rows = cur.fetchall()  # 处理查询结果
        for row in rows:
            result.append(row[0])  # 这里的[0]表示只保存了查询出来的第一列数据
    except psycopg2.Error as err:
        print('Error msg: ' + str(err))
    # finally 是一个在 try-except 语句块中的可选部分用于定义无论是否发生异常都会被执行的代码块
    finally:  
        if cur is not None:
            cur.close()  # 关闭游标
        if conn is not None:
            conn.close()  # 关闭数据库连接
    return result

3.将查询结果保存到文件中

results的值为sql的查询结果
name的值为config.HT[“name”]

curPath获取当前文件的绝对路径

file 是一个内置变量，表示当前文件的路径。os.path.dirname(file) 可以获取当前文件所在的目录路径，而 os.path.abspath() 则将相对路径转换为绝对路径。

rootPath为获取当前文件所在目录的父级目录路径

最后

不知道你们用的什么环境，我一般都是用的Python3.6环境和pycharm解释器，没有软件，或者没有资料，没人解答问题，都可以免费领取（包括今天的代码），过几天我还会做个视频教程出来，有需要也可以领取~

给大家准备的学习资料包括但不限于：

Python 环境、pycharm编辑器/永久激活/翻译插件

python 零基础视频教程

Python 界面开发实战教程

Python 爬虫实战教程

Python 数据分析实战教程

python 游戏开发实战教程

Python 电子书100本

Python 学习路线规划

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！