【Python】从入门到上头—mysql数据库操作模块mysql-connector和PyMySQL应用场景 （15）

mysql-connector

MySQL官方提供了mysql-connector-python驱动

1. 安装驱动

   python -m pip install mysql-connector
   

2. 连接数据库获取连接

   import mysql.connector
   
   db = mysql.connector.connect(
       host="localhost", #ip
       user="root", #用户名
       passwd="root",#密码
       database="python_test",#连接数据库
       auth_plugin='mysql_native_password'
   )
   #获取游标
   cursor = db.cursor()
   

创建数据库

cursor.execute("CREATE DATABASE python_test")

输出所有数据库列表：

cursor.execute("SHOW DATABASES")
for x in cursor:
    print(x)
"""
('basic_project',)
('ceam_mall2',)
('information_schema',)
('jwt-demo',)
('liugh',)
('mysql',)
('niu_b_xx_hou',)
"""

创建数据表

cursor.execute("CREATE TABLE sites (name VARCHAR(255), url VARCHAR(255))")

输出所有数据库表：

cursor.execute("SHOW TABLES")
for x in cursor:
    print(x)
#('sites',)

主键设置

cursor.execute("ALTER TABLE sites ADD COLUMN id INT AUTO_INCREMENT PRIMARY KEY")

插入数据

#sql
sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
#占位符插入数据
val = ("RUNOOB", "https://www.baidu.com")
#执行sql
cursor.execute(sql, val)

db.commit()
print(cursor.rowcount, "记录插入成功。")

批量插入

• 批量插入使用 executemany() 方法，该方法的第二个参数是一个元组列表，包含了我们要插入的数据：

sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = [
    ('Google', 'https://www.google.com'),
    ('Github', 'https://www.github.com'),
    ('Taobao', 'https://www.taobao.com'),
    ('stackoverflow', 'https://www.stackoverflow.com/')
]
#执行sql
cursor.executemany(sql, val)
#提交事务: 数据表内容有更新，必须使用到该语句
db.commit()
print(cursor.rowcount, "记录插入成功。")

如果我们想在数据记录插入后，获取该记录的 ID ，可以使用以下代码：

sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = ("Zhihu", "https://www.zhihu.com")
cursor.execute(sql, val)
db.commit()
print("1 条记录已插入, ID:", cursor.lastrowid)

查询数据

cursor.execute("SELECT * FROM sites")
result = cursor.fetchall()  # fetchall() 获取所有记录
for x in result:
	print(x)
"""
('RUNOOB', 'https://www.runoob.com', 1)
('Google', 'https://www.google.com', 2)
('Github', 'https://www.github.com', 3)
('Taobao', 'https://www.taobao.com', 4)
('stackoverflow', 'https://www.stackoverflow.com/', 5)
('Zhihu', 'https://www.zhihu.com', 6)
"""

如果我们只想读取一条数据，可以使用 fetchone() 方法：

cursor.execute("SELECT * FROM sites")
result = cursor.fetchone()
print(result)

为了防止数据库查询发生 SQL 注入的攻击，我们可以使用 %s 占位符来转义查询的条件：

sql = "SELECT * FROM sites WHERE name = %s"
na = ("RUNOOB",)
cursor.execute(sql, na)
result = cursor.fetchall()
for x in result:
    print(x)

删除

sql = "DELETE FROM sites WHERE name = %s"
na = ("stackoverflow",)
cursor.execute(sql, na)
db.commit()
print(cursor.rowcount, " 条记录删除")

更新

sql = "UPDATE sites SET name = %s WHERE name = %s"
val = ("Zhihu", "ZH")
cursor.execute(sql, val)
db.commit()
print(cursor.rowcount, " 条记录被修改")

关闭连接

#最后一定要关闭数据库连接
db.close()

Python连接MySQL报错：mysql.connector.errors.NotSupportedError: Authentication plugin ‘caching_sha2_password’ is not supported

• 添加一句语句（auth_plugin=‘mysql_native_password’），添加位置如下代码：

小结

• 执行INSERT/UPDATE/DELETE等操作后要调用commit()提交事务；
• MySQL的SQL占位符是%s。

PyMySQL

PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2 中则使用 mysqldb。

• PyMySQL 遵循 Python 数据库 API v2.0 规范，并包含了 pure-Python MySQL 客户端库。

1. 安装

   pip3 install PyMySQL
   

2. 数据库连接

   import pymysql
    
   # 打开数据库连接
   db = pymysql.connect(host='localhost',
                        user='root',
                        password='root',
                        database='python_test')
    
   # 使用 cursor() 方法创建一个游标对象 cursor
   cursor = db.cursor()
    
   # 使用 execute()  方法执行 SQL 查询 
   cursor.execute("SELECT VERSION()")
    
   # 使用 fetchone() 方法获取单条数据.
   data = cursor.fetchone()
    
   print ("Database version : %s " % data)
    
   # 关闭数据库连接
   db.close()
   

创建数据库表

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
 
# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
 
# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""
 
cursor.execute(sql)
 
# 关闭数据库连接
db.close()

数据库插入操作

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用cursor()方法获取操作游标 
cursor = db.cursor()
 
# SQL 插入语句
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
   # 执行sql语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 如果发生错误则回滚
   db.rollback()
 
# 关闭数据库连接
db.close()

以上例子也可以写成如下形式：

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用cursor()方法获取操作游标 
cursor = db.cursor()
 
# SQL 插入语句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
       LAST_NAME, AGE, SEX, INCOME) \
       VALUES ('%s', '%s',  %s,  '%s',  %s)" % \
       ('Mac', 'Mohan', 20, 'M', 2000)
try:
   # 执行sql语句
   cursor.execute(sql)
   # 执行sql语句
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()
 
# 关闭数据库连接
db.close()

数据库查询操作

• Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。

  • fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
  • fetchall(): 接收全部的返回结果行.
  • rowcount: 这是一个只读属性，并返回执行execute()方法后影响的行数。

查询EMPLOYEE表中salary（工资）字段大于1000的所有数据：

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用cursor()方法获取操作游标 
cursor = db.cursor()
 
# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE \
       WHERE INCOME > %s" % (1000)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 获取所有记录列表
   results = cursor.fetchall()
   for row in results:
      fname = row[0]
      lname = row[1]
      age = row[2]
      sex = row[3]
      income = row[4]
       # 打印结果
      print ("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
             (fname, lname, age, sex, income ))
except:
   print ("Error: unable to fetch data")
 
# 关闭数据库连接
db.close()

数据库更新操作

• 将 TESTDB 表中 SEX 为 ‘M’ 的 AGE 字段递增 1：

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用cursor()方法获取操作游标 
cursor = db.cursor()
 
# SQL 更新语句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()
 
# 关闭数据库连接
db.close()

删除操作

• 删除数据表 EMPLOYEE 中 AGE 大于 20 的所有数据：

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')
 
# 使用cursor()方法获取操作游标 
cursor = db.cursor()
 
# SQL 删除语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交修改
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()
 
# 关闭连接
db.close()

执行事务

• Python DB API 2.0 的事务提供了两个方法 commit 或 rollback。

# SQL删除记录语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 向数据库提交
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

• 对于支持事务的数据库， 在Python数据库编程中，当游标建立之时，就自动开始了一个隐形的数据库事务。

  • commit()方法提交当前游标所有更新操作，rollback（）方法回滚当前游标的所有操作。每一个方法都开始了一个新的事务。