python自定义orm

Python自定义ORM涉及创建类和函数，映射数据库表，实现数据增删改查操作。
剑阁ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景，ssl证书未来市场广阔！成为创新互联建站的ssl证书销售渠道，可以享受市场价格4-6折优惠！如果有意向欢迎电话联系或者加微信：18980820575（备注：SSL证书合作）期待与您的合作！

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

准备工作

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

pip install sqlite3

定义模型类

我们需要定义一个模型类，用于描述数据库中的表结构，这个类需要继承自我们自定义的基类，并实现一些特殊方法，例如__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__)

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

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

定义数据库操作类

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

import sqlite3
class 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()

实现CRUD操作

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

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)
查询单个用户
user = db.find_by_id(User, user1.id)
print(user)
更新用户信息
user.age = 26
db.add(user)
删除用户
db.remove(user1)
关闭数据库连接
db.close()