【Python数据库操作】使用SQLite和MySQL进行数据存储和查询！

在现代应用程序中，数据存储与管理是至关重要的。Python为开发者提供了多种与数据库进行交互的方式，其中SQLite和MySQL是最常用的两种数据库。本文将深入探讨如何使用Python进行SQLite和MySQL数据库的操作，包括创建数据库、表、插入数据、查询数据等基本操作。以下是本博客的结构：
在这里插入图片描述

1. 数据库基础知识

1.1 什么是数据库？

数据库是一个有组织的集合，用于存储、管理和检索数据。数据库系统能够确保数据的一致性、安全性和完整性。

1.2 SQLite与MySQL的区别

SQLite：
- 嵌入式数据库，适合小型应用。
- 数据库存储在单个文件中。
- 支持ACID事务。
MySQL：
- 关系数据库管理系统，适合大型应用。
- 需要服务器支持，支持多用户并发。
- 提供更复杂的功能，如存储过程、触发器等。

2. 使用SQLite

SQLite是Python内置的数据库库，非常适合快速原型开发和小型应用程序。

2.1 安装与导入SQLite

SQLite通常是Python标准库的一部分，无需额外安装。你只需导入sqlite3模块即可使用。

import sqlite3

2.2 创建SQLite数据库与连接

你可以使用sqlite3.connect()函数创建或连接到一个SQLite数据库。

# 创建或连接到SQLite数据库
connection = sqlite3.connect('example.db')

2.3 创建表

使用SQL语句创建表。以下是一个创建用户表的示例。

# 创建表
def create_table():with connection:connection.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY AUTOINCREMENT,name TEXT NOT NULL,age INTEGER NOT NULL);''')
create_table()

2.4 插入数据

使用INSERT INTO语句将数据插入表中。

def insert_user(name, age):with connection:connection.execute('''INSERT INTO users (name, age) VALUES (?, ?);''', (name, age))# 插入数据
insert_user('Alice', 30)
insert_user('Bob', 25)

2.5 查询数据

使用SELECT语句查询数据。你可以使用游标来获取查询结果。

def fetch_users():cursor = connection.cursor()cursor.execute('SELECT * FROM users;')users = cursor.fetchall()return users# 查询并打印数据
for user in fetch_users():print(user)

2.6 更新数据

使用UPDATE语句更新现有数据。

def update_user(user_id, name, age):with connection:connection.execute('''UPDATE users SET name = ?, age = ? WHERE id = ?;''', (name, age, user_id))# 更新数据
update_user(1, 'Alice Smith', 31)

2.7 删除数据

使用DELETE语句删除数据。

def delete_user(user_id):with connection:connection.execute('''DELETE FROM users WHERE id = ?;''', (user_id,))# 删除数据
delete_user(2)

在这里插入图片描述

3. 使用MySQL

MySQL是一个功能丰富的关系数据库，适用于大规模应用。我们将使用mysql-connector-python库与MySQL数据库交互。

3.1 安装MySQL Connector

使用pip安装MySQL连接器：

pip install mysql-connector-python

3.2 连接到MySQL数据库

使用mysql.connector.connect()方法连接到MySQL数据库。

import mysql.connector# 连接到MySQL数据库
connection = mysql.connector.connect(host='localhost',user='your_username',password='your_password',database='your_database'
)

3.3 创建表

创建表的SQL语句与SQLite类似。

def create_mysql_table():cursor = connection.cursor()cursor.execute('''CREATE TABLE IF NOT EXISTS employees (id INT AUTO_INCREMENT PRIMARY KEY,name VARCHAR(100) NOT NULL,salary DECIMAL(10, 2) NOT NULL);''')connection.commit()create_mysql_table()

3.4 插入数据

使用INSERT INTO语句插入数据。

def insert_employee(name, salary):cursor = connection.cursor()cursor.execute('''INSERT INTO employees (name, salary) VALUES (%s, %s);''', (name, salary))connection.commit()# 插入数据
insert_employee('John Doe', 50000.00)
insert_employee('Jane Doe', 60000.00)

3.5 查询数据

查询MySQL中的数据。

def fetch_employees():cursor = connection.cursor()cursor.execute('SELECT * FROM employees;')employees = cursor.fetchall()return employees# 查询并打印数据
for employee in fetch_employees():print(employee)

3.6 更新数据

使用UPDATE语句更新现有数据。

def update_employee(employee_id, name, salary):cursor = connection.cursor()cursor.execute('''UPDATE employees SET name = %s, salary = %s WHERE id = %s;''', (name, salary, employee_id))connection.commit()# 更新数据
update_employee(1, 'John Smith', 55000.00)

3.7 删除数据

使用DELETE语句删除数据。

def delete_employee(employee_id):cursor = connection.cursor()cursor.execute('''DELETE FROM employees WHERE id = %s;''', (employee_id,))connection.commit()# 删除数据
delete_employee(2)

在这里插入图片描述

4. 数据库连接管理

在实际应用中，数据库连接管理至关重要。我们应该确保在使用后关闭连接，以避免资源泄露。

# 关闭连接
def close_connection():connection.close()close_connection()

在这里插入图片描述

5. 使用ORM进行数据库操作

ORM（对象关系映射）是一种通过对象来操作数据库的方式，简化了数据库操作。Python中常用的ORM库有SQLAlchemy和Django ORM。

5.1 使用SQLAlchemy

安装SQLAlchemy：

pip install SQLAlchemy

使用SQLAlchemy进行数据操作的基本步骤包括创建模型、创建数据库和执行查询。

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker# 创建引擎和基础类
engine = create_engine('sqlite:///example.db')
Base = declarative_base()# 定义模型
class User(Base):__tablename__ = 'users'id = Column(Integer, primary_key=True)name = Column(String)age = Column(Integer)# 创建表
Base.metadata.create_all(engine)# 创建会话
Session = sessionmaker(bind=engine)
session = Session()# 插入数据
new_user = User(name='Alice', age=30)
session.add(new_user)
session.commit()# 查询数据
for user in session.query(User).all():print(user.name)

在这里插入图片描述

6. 数据库事务管理

事务确保一系列数据库操作要么全部成功，要么全部失败。使用Python的数据库库时，应确保使用事务以保持数据一致性。

# 使用事务
def insert_multiple_users(users):try:with connection:for user in users:insert_user(user['name'], user['age'])except Exception as e:print(f"Error: {e}")# 插入多条数据
insert_multiple_users([{'name': 'Tom', 'age': 28}, {'name': 'Jerry', 'age': 26}])

在这里插入图片描述

7. 数据库性能优化

性能优化是数据库管理的重要组成部分，以下是一些基本的优化策略：

使用索引：为常用查询字段创建索引，以提高查询速度。
避免使用SELECT *：明确列出需要的字段，减少数据传输量。
使用连接池：在高并发情况下使用连接池来复用连接，提高性能。

8. 数据库安全性

确保数据库安全性是开发中的一项重要任务，以下是一些基本的安全措施：

使用参数化查询：防止SQL注入攻击。
定期备份：确保数据的安全和恢复。
限制数据库权限：仅授予必要的权限，避免过度权限。

9. 处理异常

在与数据库交互时，处理异常至关重要。使用try-except语句捕获可能的错误，以便采取适当的措施。

try:# 进行数据库操作insert_user('Error Test', 20)
except sqlite3.Error as e:print(f"An error occurred: {e}")

在这里插入图片描述

10. 结束与总结

在这篇博客中，我们深入探讨了如何在Python中使用SQLite和MySQL进行数据库操作。我们学习了如何创建数据库和表、插入、查询、更新和删除数据、使用ORM以及如何处理事务和异常。希望这些实用的技巧能帮助你在开发中高效地进行数据库操作，并确保数据的安全和完整性。无论你是新手还是经验丰富的开发者，掌握这些技能都将极大地提升你的开发能力。
在这里插入图片描述