python 自定义orm

2026-09-30 07:07:05 点击：586

Python自定(ding)义ORM涉及创建类和函数，自定映射数据库表，自定实现数据增删改查操作。自定

Python自定义ORM（Object-Relational Mapping，自定对象关系映射）是自定一种将

数据库

中的表与Python对象进行映射的技术(shu)，它可以让我们用面向对象的自定方式来操作数据库，而不需要关(guan)心底层的自定SQL语句，本文将详细介绍如何实现一个(ge)简单的自定Python自定义ORM。

python 自定义orm

准备工作

在开始之前，自定我们需要安装一个轻量级的自定数据库，例如SQLite，自(zi)定(ding)可以使用以下命令进行安装：

python 自定义orm

pip install sqlite3

定义模型(xing)类

我(wo)们需要定义一个(ge)模型(xing)类，自定用于描述数(shu)据库中的自定(ding)表结构，这个类需要继承自我们自定义的自定基类，并实现一些特殊方法，自(zi)定例如__init__、__str__等，以下是一个简单的例子：

class Model: def __init__(self, **kwargs): for key, value in kwargs.items(): setattr(self, key, value) def __str__(self): return str(self.__dict__)

接下来，我们可以(yi)定义一个具(ju)体的模型类，例如User：

class User(Model): def __init__(self, name, age, email): super().__init__(name=name, age=age, email=email)

定义数据库操作类

为了方便地操作数据库，我们需要定(ding)义一个数据库操作类(lei)，用于封装(zhuang)一些常(chang)用的数据库操作(zuo)方(fang)法，例如增(zeng)加、删除、修改(gai)和查询，以下是一个简单的例子：

import sqlite3class DB: def __init__(self, db_name): self.conn = sqlite3.connect(db_name) self.cursor = self.conn.cursor() def execute(self, query, params=()): self.cursor.execute(query, params) self.conn.commit() def fetch(self, query, params=()): self.cursor.execute(query, params) return self.cursor.fetchall() def close(self): self.conn.close()

实(shi)现CRUD操作

有了前面的基础，我们现在可以(yi)实现一些基本的CRUD操作了，我们需要在(zai)Model类中添加一些方法，例如save、delete等，在DB类中添加相应的方法，例如add、remove等，以下是一个完(wan)整的例子：

class Model: ...其他代码... def save(self): fields = ', '.join([f'{ k}=?' for k in self.__dict__.keys()]) values = tuple(self.__dict__.values()) query = f'INSERT INTO { self.__class__.__name__.lower()} ({ fields}) VALUES ({ values})' db.execute(query, values) def delete(self): query = f'DELETE FROM { self.__class__.__name__.lower()} WHERE id=?' db.execute(query, (self.id,))class DB: ...其他代码... def add(self, model): model.save() def remove(self, model): model.delete() def find_by_id(self, model_class, id): query = f'SELECT * FROM { model_class.__name__.lower()} WHERE id=?' return model_class(**self.fetch(query, (id,))) def find_all(self, model_class): query = f'SELECT * FROM { model_class.__name__.lower()}' return [model_class(**item) for item in self.fetch(query)]

使用示例

现在我们可以使用自定义的ORM来进行一些简单的数据库操作了，以下是一个简单的例子：

db = DB('test.db')添加用户user1 = User(name='张三', age=25, email='zhangsan@example.com')db.add(user1)user2 = User(name='李四', age=30, email='lisi@example.com')db.add(user2)查询所有用户users = db.find_all(User)print(users)查询单个用(yong)户user = db.find_by_id(User, user1.id)print(user)更新用户信息user.age = 26db.add(user)删除用户db.remove(user1)关闭数(shu)据(ju)库连接db.close()