python dao框架

原创

mob64ca12e9cad4 2024-09-08 04:05:22 ©著作权

文章标签 数据访问 Python 删除用户 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e9cad4的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何实现一个 Python DAO 框架

作为一名刚入行的小白，了解和实现一个 DAO（数据访问对象）框架是一个很好的学习经历。DAO 模式可以帮助我们将数据访问逻辑与业务逻辑分离，使得代码更清晰，易于维护。在这篇文章中，我将带你逐步实现一个简单的 Python DAO 框架。

实现流程

在开始之前，让我们先来看一下实现 DAO 框架的步骤。下面是一个简单的步骤表：

步骤	描述
1	设计数据库模型
2	创建数据访问对象（DAO）
3	实现数据访问逻辑
4	测试 DAO 实现
5	文档化和维护

步骤详解

1. 设计数据库模型

我们以一个简单的用户管理系统为例，构建“用户”表。这个表将包含以下字段：

id: 主键
username: 用户名
email: 电子邮件

对应的 ER 图如下所示（使用 mermaid 语法）：

erDiagram
    USERS {
        int id PK
        string username
        string email
    }

2. 创建数据访问对象（DAO）

我们将定义一个 UserDAO 类，用于与数据库交互。

import sqlite3

class UserDAO:
    def __init__(self, db_name):
        # 初始化数据库连接
        self.connection = sqlite3.connect(db_name)
        self.create_table()

    def create_table(self):
        # 创建用户表
        query = '''CREATE TABLE IF NOT EXISTS users (
                    id INTEGER PRIMARY KEY AUTOINCREMENT,
                    username TEXT NOT NULL,
                    email TEXT NOT NULL)'''
        self.connection.execute(query)
        self.connection.commit()

__init__ 方法用于初始化数据库连接。
create_table 方法用于创建用户表。

3. 实现数据访问逻辑

在 UserDAO 类中添加基础的 CRUD（创建、读取、更新、删除）操作。

    def add_user(self, username, email):
        # 添加新用户
        query = 'INSERT INTO users (username, email) VALUES (?, ?)'
        self.connection.execute(query, (username, email))
        self.connection.commit()

    def get_user(self, user_id):
        # 根据 ID 获取用户
        query = 'SELECT * FROM users WHERE id = ?'
        cursor = self.connection.cursor()
        cursor.execute(query, (user_id,))
        return cursor.fetchone()

    def update_user(self, user_id, username, email):
        # 更新用户信息
        query = 'UPDATE users SET username = ?, email = ? WHERE id = ?'
        self.connection.execute(query, (username, email, user_id))
        self.connection.commit()

    def delete_user(self, user_id):
        # 删除用户
        query = 'DELETE FROM users WHERE id = ?'
        self.connection.execute(query, (user_id,))
        self.connection.commit()

add_user 方法将添加一个新用户。
get_user 方法将根据用户 ID 获取用户记录。
update_user 方法将更新用户信息。
delete_user 方法将删除用户记录。

4. 测试 DAO 实现

最后，我们需要测试我们的 UserDAO 实现。可以创建一个简单的测试脚本：

if __name__ == "__main__":
    dao = UserDAO('test.db')

    # 添加用户
    dao.add_user('Alice', 'alice@example.com')
    dao.add_user('Bob', 'bob@example.com')

    # 获取用户
    print(dao.get_user(1))  # 打印 Alice 的信息

    # 更新用户
    dao.update_user(1, 'Alice', 'alice_new@example.com')

    # 获取更新后的用户
    print(dao.get_user(1))  # 打印更新后的 Alice 的信息

    # 删除用户
    dao.delete_user(2)  # 删除 Bob