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

mysql-connector

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

  1. 安装驱动

    python 复制代码
    python -m pip install mysql-connector
  2. 连接数据库获取连接

    python 复制代码
    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()

创建数据库

python 复制代码
cursor.execute("CREATE DATABASE python_test")

输出所有数据库列表:

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

创建数据表

python 复制代码
cursor.execute("CREATE TABLE sites (name VARCHAR(255), url VARCHAR(255))")

输出所有数据库表:

python 复制代码
cursor.execute("SHOW TABLES")
for x in cursor:
    print(x)
#('sites',)

主键设置

python 复制代码
cursor.execute("ALTER TABLE sites ADD COLUMN id INT AUTO_INCREMENT PRIMARY KEY")

插入数据

python 复制代码
#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() 方法,该方法的第二个参数是一个元组列表,包含了我们要插入的数据:
python 复制代码
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 ,可以使用以下代码:

python 复制代码
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)

查询数据

python 复制代码
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() 方法:

python 复制代码
cursor.execute("SELECT * FROM sites")
result = cursor.fetchone()
print(result)

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

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

删除

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

更新

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

关闭连接

python 复制代码
#最后一定要关闭数据库连接
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. 安装

    python 复制代码
    pip3 install PyMySQL
  2. 数据库连接

    python 复制代码
    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()

创建数据库表

python 复制代码
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()

数据库插入操作

python 复制代码
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()

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

python 复制代码
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的所有数据:

python 复制代码
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:
python 复制代码
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 的所有数据:
python 复制代码
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
python 复制代码
# SQL删除记录语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 向数据库提交
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()
  • 对于支持事务的数据库, 在Python数据库编程中,当游标建立之时,就自动开始了一个隐形的数据库事务。

    • commit()方法提交当前游标所有更新操作,rollback()方法回滚当前游标的所有操作。每一个方法都开始了一个新的事务。
相关推荐
不去幼儿园1 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
幽兰的天空2 小时前
Python 中的模式匹配:深入了解 match 语句
开发语言·python
远歌已逝3 小时前
维护在线重做日志(二)
数据库·oracle
qq_433099404 小时前
Ubuntu20.04从零安装IsaacSim/IsaacLab
数据库
Dlwyz4 小时前
redis-击穿、穿透、雪崩
数据库·redis·缓存
网易独家音乐人Mike Zhou5 小时前
【卡尔曼滤波】数据预测Prediction观测器的理论推导及应用 C语言、Python实现(Kalman Filter)
c语言·python·单片机·物联网·算法·嵌入式·iot
安静读书5 小时前
Python解析视频FPS(帧率)、分辨率信息
python·opencv·音视频
工业甲酰苯胺6 小时前
Redis性能优化的18招
数据库·redis·性能优化
没书读了7 小时前
ssm框架-spring-spring声明式事务
java·数据库·spring
小二·7 小时前
java基础面试题笔记(基础篇)
java·笔记·python