python3入门教程（二）操作数据库（一）

概述

　　最近在准备写一个爬虫的练手项目，基本想法是把某新闻网站的内容分类爬取下来，保存至数据库，再通过接口对外输出（提供后台查询接口）。那么问题就来了，python到底是怎么去操作数据库的呢？我们今天就来研究下。

准备

我这边数据库使用的是mysql5.7，python去操作mysql数据库需要使用到pymysql模块。

安装pymysql很简单，可以使用pip，

pip install pymysql

也可以通过pycharm直接添加安装。

使用操作

我们新建一个数据库，名字就叫做news，我们再创建一张news表（取名字这种事情真的非常纠结），语句如下：

CREATE TABLE `news` (
  `id` bigint(100) NOT NULL AUTO_INCREMENT,
  `title` varchar(1000) DEFAULT NULL,
  `url` varchar(1000) DEFAULT NULL,
  `hash` varchar(1000) DEFAULT NULL,
  `publish_time` varchar(15) DEFAULT NULL,
  `catagary` varchar(1) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8

这张表设计的有些随意，不过这不是重点，重点是我们如何利用python去操作这张表，下面我们就进入主题。

首先，我们新建python file，然后导入我们要用到的模块

import pymysql

1.获取数据库连接及游标

之前有用过数据库的同学肯定知道，用代码去操作数据库之前，肯定先要获取连接，后续的增删改查都是基于这个连接，当然，相应的操作完成之后，一定要记得把连接关闭，不然会造成一些不好的后果。

python获取数据库连接代码如下：

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')

print(conn)

运行之后输出：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000001D359180B00>

Process finished with exit code 0

显然，我们已经成功获取到了一个连接，当然要执行sql语句，我们好需要获取一个游标对象，

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')

print(conn) 
# 获取操作sql的游标对象（这样写默认返回的数据类型是元组，我们下面会有介绍）
cursor = conn.cursor()

print(cursor)

运行之后输出：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x0000025F14470B70>
<pymysql.cursors.Cursor object at 0x0000025F14AC8E10>

Process finished with exit code 0

游标对象已经成功拿到了

2.新增数据

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor()
print(cursor)
# 新增一条数据
insert_sql = 'insert into news(title, url, hash, publish_time, catagary) values(%s, %s, %s, %s, %s)'
row = cursor.execute(insert_sql, ('百度', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0')) # 参数为元组类型
print(row)
print(cursor.lastrowid) # 获取最后一次自增的ID

执行上面的代码，输出结果如下：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x0000023455416358>
<pymysql.cursors.Cursor object at 0x00000234559F8E10>
1
2

Process finished with exit code 0

从输出结果可以看出，似乎已经插入成功了，我们取数据库看一下，结果是令人失望的，数据库没有，什么情况呢？

原因其实很简单，我们没有做commit操作，我们把commit加上，再试下

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor()
print(cursor)

# 新增一条数据
insert_sql = 'insert into news(title, url, hash, publish_time, catagary) values(%s, %s, %s, %s, %s)'  # 使用pymysql的参数化语句防止SQL注入
# 返回受影响的行数
row = cursor.execute(insert_sql, ('百度', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0'))  # 参数为元组类型
print(row)
print(cursor.lastrowid)  # 获取最后一次自增的ID

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接

下面我们再运行下，输出如下：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000001D60F605630>
<pymysql.cursors.Cursor object at 0x000001D60FD88E80>
1
3

Process finished with exit code 0　

我们查一下数据库，这下就有了，插入成功了。

当然也可以批量插入，使用executemany

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor()
print(cursor)

# 新增一条数据
insert_sql = 'insert into news(title, url, hash, publish_time, catagary) values(%s, %s, %s, %s, %s)'  # 使用pymysql的参数化语句防止SQL注入
# 返回受影响的行数
#row = cursor.execute(insert_sql, ('百度', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0'))  # 参数为元组类型
#print(row)
#print(cursor.lastrowid)  # 获取最后一次自增的ID

# 新增多条数据
row = cursor.executemany(insert_sql, [('百度1', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0'),
                                      ('百度2', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0'),
                                      ('百度3', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0')])

print(row)

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接

执行上面的脚本，输出结果如下：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000001A454EF5630>
<pymysql.cursors.Cursor object at 0x000001A455678E48>
3
Process finished with exit code 0

查询数据库，如下：

批量插入成功。

3.查询数据

前面插入数据成功了，我们研究下怎么查询

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor()
print(cursor)

query_sql = 'select * from news where id = %s'
row = cursor.execute(query_sql, (3,))
print(row)  # 查询到的行数
print(cursor.fetchall())  # fetchall:获取所有行的信息 fetchone:获取结果集中第一行的数据 fetchmany:获取结果集中的多行

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接

执行上面的查询，输出：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000001C5D9107320>
<pymysql.cursors.Cursor object at 0x000001C5D9748EF0>
1
((3, '百度', 'http://www.baidu.com', '1qaz2wsx', '2018-11-02', '0'),)

Process finished with exit code 0

从输出结果我们可以看到，最后拿到的结果集是一个元组类型，那假如我们想要的是字典类型呢？

我们只需要在初始化游标的时候，稍微设置一下

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

我们加上再试下，

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)
print(cursor)

query_sql = 'select * from news where id = %s'
row = cursor.execute(query_sql, (3,))
print(row)  # 查询到的行数
print(cursor.fetchall())  # fetchall:获取所有行的信息 fetchone:获取结果集中第一行的数据 fetchmany:获取结果集中的多行

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接

运行之后输出：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000002A433136358>
<pymysql.cursors.DictCursor object at 0x000002A433778E48>
1
[{'id': 3, 'title': '百度', 'url': 'http://www.baidu.com', 'hash': '1qaz2wsx', 'publish_time': '2018-11-02', 'catagary': '0'}]

Process finished with exit code 0

可以看到，结果集的数据类型变了，这样比较方面去处理具体的数据内容。

4.更新数据

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)
print(cursor)

update_sql = 'update news set title = %s where id = %s'
row_upd = cursor.execute(update_sql, ('百度Updata', 3)) # 也可使用executemany实现批量更新
print(row_upd)

query_sql = 'select * from news where id = %s'
row = cursor.execute(query_sql, (3,))
print(row)  # 查询到的行数
print(cursor.fetchall())  # fetchall:获取所有行的信息 fetchone:获取结果集中第一行的数据 fetchmany:获取结果集中的多行

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接

我们更新id=3的那条数据的标题，运行上面的代码，输出：

E:\pythonsource\venv\Scripts\python.exe E:/pythonsource/study/pymysqldemo.py
<pymysql.connections.Connection object at 0x000001FCDBB36390>
<pymysql.cursors.DictCursor object at 0x000001FCDC158EB8>
1
1
[{'id': 3, 'title': '百度Updata', 'url': 'http://www.baidu.com', 'hash': '1qaz2wsx', 'publish_time': '2018-11-02', 'catagary': '0'}]

Process finished with exit code 0

更新成功。

5.删除数据

删除数据其实更上面的更新差不多，只是sql不一样。

import pymysql

# 获取连接 参数（数据库IP,端口，用户名,密码,数据库名称，编码格式）
conn = pymysql.connect(host='127.0.0.1', port=3306,user='root', passwd='123456', db='news', charset='utf8')
print(conn)
# 获取操作sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)
print(cursor)

delete_sql = 'delete from news where id=%s'
row_delete = cursor.execute(delete_sql, (3,)) # 也可以使用executemany实现批量删除

query_sql = 'select * from news where id = %s'
row = cursor.execute(query_sql, (3,))
print(row)  # 查询到的行数
print(cursor.fetchall())  # fetchall:获取所有行的信息 fetchone:获取结果集中第一行的数据 fetchmany:获取结果集中的多行

conn.commit()  # 提交数据
cursor.close()  # 关闭游标
conn.close()  # 关闭连接