Python学习—数据库篇之pymysql

一、pymysql简介

对于Python操作MySQL主要使用两种方式：

• 原生模块 pymsql
• ORM框架 SQLAchemy

pymsql是Python中操作MySQL的模块，其使用方法和MySQLdb几乎相同。

二、简单使用

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
#执行sql操作
r = cursor.execute("insert into student(age,sex,class_no) values(16,'male',2)")
# r为执行sql语句后受影响的行数
print(r)
# 提交，不然无法保存新建或者修改的数据
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()

 

三、增删改查

增

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()

# 增加一行数据
cursor.execute("insert into tb1(name,part) values('cdc',1)")

# 字符串拼接sql(禁止使用,会引起sql注入)
inp_1 = ('cdcy',2,)
sql = "insert into tb1(name,part) values('%s','%s')" % inp_1
cursor.execute(sql)

# 带参数插入数据(推荐使用)
inp_2 = ('cdcx',2)
cursor.execute("insert into tb1(name,part) values(%s,%s)",inp_2)


# 增加多行数据
lis = [('cdc1',2),
       ('cdc2',2),
       ('cdc3',2),
       ]
cursor.executemany("insert into tb1(name,part) values(%s,%s)",lis)

# 提交，不然无法保存新建或者修改的数据
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()

删

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
#执行sql操作
r = cursor.execute("delete from tb1 where name=%s",('alex',))
# r为执行sql语句后受影响的行数
print(r)
# 提交，不然无法保存新建或者修改的数据
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()

改

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
#执行sql操作
r = cursor.execute("update tb1 set name=%s where part=%s",('ccc',1,))
# r为执行sql语句后受影响的行数
print(r)
# 提交，不然无法保存新建或者修改的数据
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()

 查

查新操作时，不需要执行commit操作

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
# 查询数据
cursor.execute('select * from tb3')

# 展示一行
result = cursor.fetchone()
print(result)
result = cursor.fetchone()
print(result)

# 展示多行
result = cursor.fetchmany(3)
print(result)

# 展示全部
result = cursor.fetchall()
print(result)

注意：fetch操作的机制类似于文件操作中的指针，会在上一次的展示基础位置上继续往下展示n行数据

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
# 查询数据
cursor.execute('select * from tb3')

# 展示一行
result = cursor.fetchone()
print(result)
result = cursor.fetchone()
print(result)

# 展示多行
result = cursor.fetchmany(3)
print(result)


# ******************** 执行结果 ***************
"""
(1, 'cdc', 1)
(2, 'ccc', 1)
((3, 'ctt', 3), (4, 'lj', 4), (5, 'xx', 5))
"""

fetch数据时按照顺序进行，可以使用cursor.scroll(num,mode)来移动游标位置，如：

• cursor.scroll(n,mode='relative')  # 相对当前位置移动，即从当前位置向前或者向后移动n个位置，再进行数据展示
• cursor.scroll(n,mode='absolute') # 相对绝对位置移动，即从当前位置回到n位置后，再进行数据的展示

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
# 查询数据
cursor.execute('select * from tb3')

# 展示一行
result = cursor.fetchone()
print(result)
result = cursor.fetchone()
print(result)

# 向前移动一个位置
"""
(1, 'cdc', 1)
(2, 'ccc', 1)
(2, 'ccc', 1)
"""
cursor.scroll(-1,mode='relative')

# 向后移动一个位置
"""
(1, 'cdc', 1)
(2, 'ccc', 1)
(4, 'lj', 4)

"""
cursor.scroll(1,mode='relative')
result = cursor.fetchone()
print(result)

此外，fetch操作默认获取的数据是元祖类型，如果想要或者字典类型的数据，可以通过改变游标的方式实现：

# -*- coding:utf-8 -*-
# author: cdc
# date: 2019/3/18

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
# 查询数据
cursor.execute('select * from tb3')

# 展示全部
result = cursor.fetchall()
print(result)

# *********  执行结果 ************
"""
[{'part_no': 1, 'name': 'cdc', 'no': 1}, {'part_no': 1, 'name': 'ccc', 'no': 2}, {'part_no': 3, 'name': 'ctt', 'no': 3}, {'part_no': 4, 'name': 'lj', 'no': 4}, {'part_no': 5, 'name': 'xx', 'no': 5}, {'part_no': 9, 'name': 'cc', 'no': 6}, {'part_no': 5, 'name': 'fdc', 'no': 7}]
"""

 其他操作

# 注：必须得有自增列

import pymysql

# 创建连接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='cdc19951216', db='test',charset='utf8')
# 创建游标
cursor = conn.cursor()
# 查询数据
cursor.execute('insert into class(address) values(%s)',('class_3',))

new_id = cursor.lastrowid
print(new_id)