一、文章介绍
本文主要介绍在pycharm集成开发坏境中，使用python语言，利用psycopg2第三方库，向postgre数据库中插入数据，为测试环境构造模拟数据的过程。包括psycopg2介绍、random函数介绍，以及根据案例一个具体的数据库表里插入20条记录进行代码拆分。
为了测试以上内容，需要在web页面构造测试数据，接下来介绍psycopg2插件及random函数。
二、psycopg2介绍
什么是psycopg2
psycopg2是一个用于访问PostgreSQL数据库的第三方库。它提供了一个简单易用的API，使得在Python程序中进行数据库操作变得容易，包括查询、插入、更新和删除等操作。psycopg2基于C语言实现，具有高效性，能够提供高效的数据库操作。
特别注意psycopg2是特别针对PostgreSQL的数据库，如果你的数据库是其他类型还需要替换其他插件如sqlalchemy，SQLAlchemy适用于多种数据库库，包括MySQL、PostgreSQL、SQLite、Oracle等。
psycopg2使用过程：
总共分为6个过程
·安装psycopg2库。可以通过pip安装，使用命令：pip install psycopg2
· 建立与数据库的连接。
import psycopg2
# 创建连接对象
conn = psycopg2.connect(
host="localhost", #指定数据库服务器的地址
database="mydatabase", #指定要连接的数据库的名称
user="myuser", #指定用于连接到数据库的用户名，这里是myuser
password="mypassword" #指定用于连接到数据库的密码，这里是mypassword
)
· 创建游标对象cur = conn.cursor()
· 执行SQL语句，插入数据。可以使用execute()方法执行SQL语句。
cur.execute("INSERT INTO mytable ("INSERT INTO device_water (id, device_water_id, data_group,) VALUES ({id}, {device_water_id}, '{data_group}') #向device_water表里id, device_water_id, data_group字段插入数据
· 提交事务，conn.commit()
· 关闭游标和连接cur.close() conn.close()
三 random函数
接下来就案例中用到random函数常用的语法进行介绍
·random.randint(a, b)：返回一个[a, b]范围内的随机整数，包括a和b。
Eg: del_flag = random.randint(0, 1)
· random.choice(a,b)从给定的集合中返回随机元素。
Eg:workspace_code=''.join(random.choices('abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ', k=5))
workspace_code变量将包含一个随机的5位字符串，由小写字母、大写字母和数字组成。
本文节选自第七十七期《51测试天地》
《python 脚本-利用psycopg2 插件向postgre 构造测试数据》一文
想继续阅读全文或查看更多精彩内容，请点击下载：
