python操作mysql（增、删、改、查）

用python操作数据库，特别是做性能测试造存量数据时特别简单方便，比存储过程方便多了。

连接数据库

前提：安装mysql、python，参考：https://www.cnblogs.com/UncleYong/p/10530261.html

数据库qzcsjb的test表中初始化的数据：

安装pymysql模块，pip install pymysql

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor()

# 执行sql语句

sql = 'select * from test where name = "%s" and id="%s"' %('qzcsbj1','1')

rows=cursor.execute(sql)  # 返回结果是受影响的行数

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

# 判断是否连接成功

if rows >= 0:

    print('连接数据库成功')

else:

    print('连接数据库失败')

增加数据

单条

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor()

# 执行sql语句

sql='insert into test(id,name) values(%s,%s)'

rows=cursor.execute(sql,('4','qzcsbj4'))

# 提交

conn.commit()

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

多条

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor()

# 执行sql语句

sql='insert into test(id,name) values(%s,%s)'

rows=cursor.executemany(sql,[('5','qzcsbj5'),('6','qzcsbj6'),('7','qzcsbj7')])

# 提交

conn.commit()

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

大批量新增

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

values=[]

for i in range(100, 201):

    values.append((i, 'qzcsbj'+str(i)))

sql='insert into test(id,name) values(%s,%s)'

rows=cursor.executemany(sql,values)

# 提交

conn.commit()

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

修改数据

把上面大批量新增的数据删除，delete from test where id>=100;

单条

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor()

# 执行sql语句

sql='update test set name = %s where id = %s'

rows=cursor.execute(sql,('qzcsbj','7'))

# 提交

conn.commit()

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

多条

import pymysql

# 建立数据库连接

conn=pymysql.connect(

    host='192.168.168.168',

    port=3306,

    user='root',

    password='mysql',

    db='qzcsbj',

    charset='utf8'

)

# 获取游标

cursor=conn.cursor()

# 执行sql语句

sql='update test set name = %s where id = %s'

rows=cursor.executemany(sql,[('全栈测试笔记5','5'),('全栈测试笔记6','6')])

# 提交

conn.commit()

# 关闭游标

cursor.close()

# 关闭连接

conn.close()

删除数据

单条

下面脚本和上面增加数据，除了执行sql语句部分不一样，其余都一样

# 执行sql语句

sql='delete from test where id = %s'

rows=cursor.execute(sql,('1',))

多条

下面脚本和上面增加数据，除了执行sql语句部分不一样，其余都一样

# 执行sql语句

sql='delete from test where id = %s'

rows=cursor.executemany(sql,[('2'),('3')])

查询数据

fetchone

有点像从管道中取一个，如果再来一个fetchone，会又取下一个，如果取完了再取，就返回None

每条记录为元组格式

下面脚本和上面增加数据，除了执行sql语句部分不一样，其余都一样

# 执行sql语句

rows=cursor.execute('select * from test;')

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

运行结果：

(4, 'qzcsbj4')
(5, '全栈测试笔记5')
(6, '全栈测试笔记6')
(7, 'qzcsbj')
None

每条记录为字典格式

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

rows=cursor.execute('select * from test;')

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

print(cursor.fetchone())

运行结果：

{'id': 4, 'name': 'qzcsbj4'}
{'id': 5, 'name': '全栈测试笔记5'}
{'id': 6, 'name': '全栈测试笔记6'}
{'id': 7, 'name': 'qzcsbj'}
None

fetchmany

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

rows=cursor.execute('select * from test;')

print(cursor.fetchmany(2))

运行结果：

[{'id': 4, 'name': 'qzcsbj4'}, {'id': 5, 'name': '全栈测试笔记5'}]

fetchall

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

rows=cursor.execute('select * from test;')

print(cursor.fetchall())

print(cursor.fetchall())

运行结果：

[{'id': 4, 'name': 'qzcsbj4'}, {'id': 5, 'name': '全栈测试笔记5'}, {'id': 6, 'name': '全栈测试笔记6'}, {'id': 7, 'name': 'qzcsbj'}]
[]

相对绝对位置移动

从头开始跳过n个

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

rows=cursor.execute('select * from test;')

cursor.scroll(3,mode='absolute')

print(cursor.fetchone())

运行结果：

{'id': 7, 'name': 'qzcsbj'}

相对当前位置移动

# 获取游标

cursor=conn.cursor(pymysql.cursors.DictCursor)

# 执行sql语句

rows=cursor.execute('select * from test;')

print(cursor.fetchone())

cursor.scroll(2,mode='relative')

print(cursor.fetchone())

运行结果：

{'id': 4, 'name': 'qzcsbj4'}
{'id': 7, 'name': 'qzcsbj'}