python操作数据库

在做自动化测试的过程中，有时候需要操作数据库，下面就用pymysql连接数据库，并实现简单的增删改查询功能

一、查询

目标：获取learn_activity表中name="测试"的id值

image.png

import pymysql
# 连接数据库，指定数据库的ip地址，账号、密码、端口号、要操作的数据库
conn = pymysql.connect(host='***.***.80.155', port=3307, user='root', passwd='*****', db='leawod_learn')
# 创建游标
cursor = conn.cursor()

# SQL查询语句
sql = "select * from learn_activity where name = '测试'"

try:
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    results = cursor.fetchall()
    # print(results)
    for i in results:
        print(i)
        id = i[0]
        print("id值是:%s"%id)

except:
    # 如果发生错误则回滚
    conn.rollback()
# 关闭数据库连接
 conn.close()

运行结果：

(237291614169858048, '测试', '/20190517/b93de77c-acb2-4286-8cdd-ab65e3a8874d.jpg', '<p><img src="/20190517/3ea80fc6-a68a-42c4-afb7-9041edaaa0d1.jpg""></p><p>测试</p>', 1, 1, '', '', 0, 105, 1, 16, datetime.datetime(2019, 5, 17, 19, 12, 9), datetime.datetime(2019, 6, 27, 16, 10, 6), 0)
id值是:237291614169858048

二、插入数据

目标：向student表中添加一条数据，id=1，first_name=李， last_name=四

image.png

import pymysql
# 连接数据库，指定数据库的ip地址，账号、密码、端口号、要操作的数据库
conn = pymysql.connect(host='192.168.**.***', port=3306, user='root', passwd='123456', db='employees')
# 创建游标
cursor = conn.cursor()

# SQL插入语句
sql = "insert into student values (1, '李', '四')"
try:
    # 执行SQL语句
    cursor.execute(sql)
    #关闭游标，提交，关闭数据库连接
    # 如果没有这些关闭操作，执行后在数据库中查看不到数据
    cursor.close()
    # 提交到数据库执行
    conn.commit()

except:
    # 如果发生错误则回滚
    conn.rollback()

# 关闭数据库连接
conn.close()

执行结果：

insert.png