0基础跟我学python---进阶篇（2）python对数据库的操作

最新推荐文章于 2024-09-12 17:23:27 发布

智达教育‍

最新推荐文章于 2024-09-12 17:23:27 发布

阅读量919

点赞数 1

文章标签： java jvm servlet

本文链接：https://blog.csdn.net/qq_29235677/article/details/126179838

版权

📢📢📢📣📣📣
哈喽！大家好，我是「奇点」，江湖人称 singularity。刚工作几年，想和大家一同进步🤝🤝
一位上进心十足的【Java ToB端大厂领域博主】！😜😜😜
喜欢java和python，平时比较懒，能用程序解决的坚决不手动解决😜😜😜

✨ 如果有对【java】感兴趣的【小可爱】，欢迎关注我

❤️❤️❤️感谢各位大可爱小可爱！❤️❤️❤️
————————————————

如果觉得本文对你有帮助，欢迎点赞，欢迎关注我，如果有补充欢迎评论交流，我将努力创作更多更好的文章。

MySQL 是最流行的关系型数据库管理系统，如果你不熟悉 MySQL，可以阅读我的mysql文章

专栏，如果熟悉的大佬直接跳过。

学过java的同学知道java中有很多方式实现对mysql数据库操作，这些框架如JDBC、mybatis、hibernate、JPA等都实现了对数据库等很好的支持，同样python也对数据库的支持提供了很多方法。本片文章将介绍python对数据库等操作。

✈️1.mysql-connector的使用

✈️1.mysql-connector的使用

mysql-connector 来连接使用 MySQL， mysql-connector 是 MySQL 官方提供的驱动器。

我们可以使用 pip 命令来安装 mysql-connector：

python -m pip install mysql-connector

使用以下代码测试 mysql-connector 是否安装成功：

import mysql.connector

执行以上代码，如果没有产生错误，表明安装成功。

注意：如果你的 MySQL 是 8.0 版本，密码插件验证方式发生了变化，早期版本为 mysql_native_password，8.0 版本为 caching_sha2_password，所以需要做些改变：

先修改 my.ini 配置：

[mysqld]
default_authentication_plugin=mysql_native_password

然后在 mysql 下执行以下命令来修改密码：

ALTER USER 'root'@'localhost' IDENTIFIED WITH mysql_native_password BY '新密码';

🚗 （1）创建数据库连接

可以使用以下代码来连接数据库：

import mysql.connector
 
mydb = mysql.connector.connect(
  host="localhost",       # 数据库主机地址
  user="yourusername",    # 数据库用户名
  passwd="yourpassword"   # 数据库密码
)
 
print(mydb)

结果

<mysql.connector.connection.MySQLConnection object at 0x10866b9d0>

现在就连接上数据库了，接下来我们创建一个数库 python_db

import mysql.connector

mydb = mysql.connector.connect(
    host="localhost",  # 数据库主机地址
    user="root",  # 数据库用户名
    passwd="root"  # 数据库密码
)

print(mydb)
my_cursor = mydb.cursor()

my_cursor.execute("CREATE DATABASE python_db ")

查询数据库方法

my_cursor.execute("SHOW DATABASES")

for x in my_cursor:
    print(x)

或者我们可以直接连接数据库，如果数据库不存在，会输出错误信息：

import mysql.connector
 
mydb = mysql.connector.connect(
  host="localhost",
  user="root",
  passwd="root",
  database="python_db"
)

🚚(2)创建数据表

创建数据表使用 "CREATE TABLE" 语句，创建数据表前，需要确保数据库已存在，以下创建一个名为 sites 的数据表：

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

表创建完成

查询数据库表

my_cursor.execute("SHOW TABLES")

for x in my_cursor:
    print(x)

('sites',)

主键设置

创建表的时候我们一般都会设置一个主键（PRIMARY KEY），我们可以使用 "INT AUTO_INCREMENT PRIMARY KEY" 语句来创建一个主键，主键起始值为 1，逐步递增。

如果我们的表已经创建，我们需要使用 ALTER TABLE 来给表添加主键：

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

当然在创建表的时候，也能够指明主键

my_cursor.execute("CREATE TABLE IF NOT EXISTS sites (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), url VARCHAR(255))")

🚄（3）插入数据

单一插入

插入数据使用 "INSERT INTO" 语句：

sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = ("baidu", "https://www.baidu.com")
my_cursor.execute(sql, val)

mydb.commit()  # 数据表内容有更新，必须使用到该语句

print(my_cursor.rowcount, "记录插入成功。")

使用元祖来接收了请求的参数

结果

1 记录插入成功。

批量插入

批量插入使用 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/')
]

my_cursor.executemany(sql, val)

mydb.commit()  # 数据表内容有更新，必须使用到该语句

print(my_cursor.rowcount, "记录插入成功。")

4 记录插入成功。

数据库数据

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

"""
如果我们想在数据记录插入后，获取该记录的 ID ，可以使用以下代码：
"""
sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = ("Zhihu", "https://www.zhihu.com")
my_cursor.execute(sql, val)

mydb.commit()

print("1 条记录已插入, ID:", my_cursor.lastrowid)

1 条记录已插入, ID: 6

✈️（4）查询数据

查询数据使用 SELECT 语句

"""
查询数据
"""
my_cursor.execute("SELECT * FROM sites")

my_result = my_cursor.fetchall()  # fetchall() 获取所有记录

for x in my_result:
    print(x)

也可以读取指定的字段数据：

my_cursor.execute("SELECT name, url FROM sites")

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

my_cursor.execute("SELECT * FROM sites")
 
myresult = my_cursor.fetchone()
 
print(myresult)

根据条件查询where 条件语句

如果我们要读取指定条件的数据，可以使用 where 语句：

sql = "SELECT * FROM sites WHERE name ='Google'"
 
mycursor.execute(sql)
 
myresult = mycursor.fetchall()
 
for x in myresult:
  print(x)

('Google', 'https://www.google.com', 2)

也可以使用通配符 %：

sql = "SELECT * FROM sites WHERE url LIKE '%oo%'"
 
my_cursor.execute(sql)
 
myresult = my_cursor.fetchall()
 
for x in myresult:
  print(x)

('Google', 'https://www.google.com', 2)

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

sql = "SELECT * FROM sites WHERE name = %s"
na = ("baidu", )
 
my_cursor.execute(sql, na)
 
my_result = my_cursor.fetchall()
 
for x in my_result:
  print(x)

('baidu', 'https://www.baidu.com', 1)

排序

查询结果排序可以使用 ORDER BY 语句，默认的排序方式为升序，关键字为 ASC，如果要设置降序排序，可以设置关键字 DESC。

"""
排序
"""
sql = "SELECT * FROM sites ORDER BY name"

my_cursor.execute(sql)

myresult = my_cursor.fetchall()

for x in myresult:
    print(x)

('baidu', 'https://www.baidu.com', 1)
('Github', 'https://www.github.com', 3)
('Google', 'https://www.google.com', 2)
('stackoverflow', 'https://www.stackoverflow.com/', 5)
('Taobao', 'https://www.taobao.com', 4)
('Zhihu', 'https://www.zhihu.com', 6)

降序

sql = "SELECT * FROM sites ORDER BY name DESC"

Limit

如果我们要设置查询的数据量，可以通过 "LIMIT" 语句来指定

my_cursor.execute("SELECT * FROM sites LIMIT 3")

myresult = my_cursor.fetchall()

for x in myresult:
    print(x)

('baidu', 'https://www.baidu.com', 1)
('Google', 'https://www.google.com', 2)
('Github', 'https://www.github.com', 3)

也可以指定起始位置，使用的关键字是 OFFSET：

mycursor.execute("SELECT * FROM sites LIMIT 3 OFFSET 1")  # 0 为 第一条，1 为第二条，以此类推

('Google', 'https://www.google.com', 2)
('Github', 'https://www.github.com', 3)
('Taobao', 'https://www.taobao.com', 4)

从0开始，所以1是第二条

🚴‍♀️（5）删除记录

删除记录使用 "DELETE FROM" 语句：

sql = "DELETE FROM sites WHERE name = 'stackoverflow'"
 
my_cursor.execute(sql)
 
mydb.commit()
 
print(my_cursor.rowcount, " 条记录删除")

1 条记录删除

注意：要慎重使用删除语句，删除语句要确保指定了 WHERE 条件语句，否则会导致整表数据被删除。

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

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

🛺(6) 更新表数据

数据表更新使用 "UPDATE" 语句：

sql = "UPDATE sites SET name = 'ZH' WHERE name = 'Zhihu'"
 
my_cursor.execute(sql)
 
mydb.commit()
 
print(my_cursor.rowcount, " 条记录被修改")

1 条记录被修改

注意：UPDATE 语句要确保指定了 WHERE 条件语句，否则会导致整表数据被更新。

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

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

🚢（7）删除表

删除表使用 "DROP TABLE" 语句， IF EXISTS 关键字是用于判断表是否存在，只有在存在的情况才删除：

sql = "DROP TABLE IF EXISTS sites"  # 删除数据表 sites
 
my_cursor.execute(sql)

这里主要就是写一些sql语句来实现数据库的基本操作，如创建数据库、表、增、删、改、查的操作。 mysql-connector对mysql数据库的操作就先讲解到这里吧，一些最基本的操作就将完了。

🚄 2. PyMySQL 驱动

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

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

🚚（1）安装

pip3 install PyMySQL

如果你的系统不支持 pip 命令，可以使用以下方式安装：

1、使用 git 命令下载安装包安装(你也可以手动下载)：

$ git clone https://github.com/PyMySQL/PyMySQL
$ cd PyMySQL/
$ python3 setup.py install

2、如果需要制定版本号，可以使用 curl 命令来安装：

$ # X.X 为 PyMySQL 的版本号
$ curl -L https://github.com/PyMySQL/PyMySQL/tarball/pymysql-X.X | tar xz
$ cd PyMySQL*
$ python3 setup.py install
$ # 现在你可以删除 PyMySQL* 目录

注意：请确保您有root权限来安装上述模块。

安装的过程中可能会出现"ImportError: No module named setuptools"的错误提示，意思是你没有安装setuptools，你可以访问setuptools · PyPI 找到各个系统的安装方法。

Linux 系统安装实例：

$ wget https://bootstrap.pypa.io/ez_setup.py
$ python3 ez_setup.py

连接数据库前，请先确认以下事项：

您已经创建了数据库 TESTDB.
在TESTDB数据库中您已经创建了表 EMPLOYEE
EMPLOYEE表字段为 FIRST_NAME, LAST_NAME, AGE, SEX 和 INCOME。
连接数据库TESTDB使用的用户名为 "root" ，密码为 "root",你可以可以自己设定或者直接使用root用户名及其密码，Mysql数据库用户授权请使用Grant命令。
在你的机子上已经安装了 Python pymysql 模块。

🚓 (2)创建数据库表

如果数据库连接存在我们可以使用execute()方法来为数据库创建表，如下所示创建表EMPLOYEE：

import pymysql
 
# 打开数据库连接
db = pymysql.connect(host='localhost',
                     user='root',
                     password='root',
                     database='python_db')
 
# 使用 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()

🚗 (2)数据库插入操作

以下实例使用执行 SQL INSERT 语句向表 EMPLOYEE 插入记录：

sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
    cursor.execute(sql)
    db.commit()
    print("执行sql了")
except:
    print("发生异常了回滚")
    db.rollback()
finally:
    # 关闭数据库连接
    print("数据库关闭")
    db.close()

方式2

# SQL 插入语句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
       LAST_NAME, AGE, SEX, INCOME) \
       VALUES ('%s', '%s',  %s,  '%s',  %s)" % \
       ('Mac', 'Mohan', 20, 'M', 2000)

🚴‍♀️ (3)数据库查询操作

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

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

# 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()

fname=Mac,lname=Mohan,age=20,sex=M,income=2000.0
fname=win,lname=abama,age=20,sex=M,income=2000.0

🚴（4）数据库更新操作

"""
更新操作
"""

# 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 的所有数据：

sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 提交修改
    db.commit()
except:
    # 发生错误时回滚
    db.rollback()

# 关闭连接
db.close()

数据都删除了

🏍️（5）执行事务

事务机制可以确保数据一致性。

事务应该具有4个属性：原子性、一致性、隔离性、持久性。这四个属性通常称为ACID特性。

原子性（atomicity）。一个事务是一个不可分割的工作单位，事务中包括的诸操作要么都做，要么都不做。
一致性（consistency）。事务必须是使数据库从一个一致性状态变到另一个一致性状态。一致性与原子性是密切相关的。
隔离性（isolation）。一个事务的执行不能被其他事务干扰。即一个事务内部的操作及使用的数据对并发的其他事务是隔离的，并发执行的各个事务之间不能互相干扰。
持久性（durability）。持续性也称永久性（permanence），指一个事务一旦提交，它对数据库中数据的改变就应该是永久性的。接下来的其他操作或故障不应该对其有任何影响。

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

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

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

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

🚗（6）错误处理

DB API中定义了一些数据库操作的错误及异常，下表列出了这些错误和异常:

异常	描述
Warning	当有严重警告时触发，例如插入数据是被截断等等。必须是 StandardError 的子类。
Error	警告以外所有其他错误类。必须是 StandardError 的子类。
InterfaceError	当有数据库接口模块本身的错误（而不是数据库的错误）发生时触发。必须是Error的子类。
DatabaseError	和数据库有关的错误发生时触发。必须是Error的子类。
DataError	当有数据处理时的错误发生时触发，例如：除零错误，数据超范围等等。必须是DatabaseError的子类。
OperationalError	指非用户控制的，而是操作数据库时发生的错误。例如：连接意外断开、数据库名未找到、事务处理失败、内存分配错误等等操作数据库是发生的错误。必须是DatabaseError的子类。
IntegrityError	完整性相关的错误，例如外键检查失败等。必须是DatabaseError子类。
InternalError	数据库的内部错误，例如游标（cursor）失效了、事务同步失败等等。必须是DatabaseError子类。
ProgrammingError	程序错误，例如数据表（table）没找到或已存在、SQL语句语法错误、参数数量错误等等。必须是DatabaseError的子类。
NotSupportedError	不支持错误，指使用了数据库不支持的函数或API等。例如在连接对象上使用.rollback()函数，然而数据库并不支持事务或者事务已关闭。必须是DatabaseError的子类。