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)

pymysql查询操作

注意：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)

获取最新自增ID