时间戳

首页 > 解决方案 > 尝试为 peewee 使用自定义 UUID mixin 不会覆盖基本 AutoField ID

python - 尝试为 peewee 使用自定义 UUID mixin 不会覆盖基本 AutoField ID

问题描述

我想创建一个仅包含 UUIDField 的字段 mixin，作为包含该 mixin 的任何模型的主键，用于某些模型，但不是全部（某些模型的主键由外部程序定义）。我希望我的大多数模型不使用 AutoField 作为主键。具体来说，我正在设计一个简单的货币系统，并且我有一个 Wallet 模型，它将继承自 peewee 建议您创建用于初始化数据库的 BaseModel 类，以及这个 UUID mixin。我的问题是，当我尝试时database.create_tables([Wallet])，我收到以下错误：

peewee.ImproperlyConfigured: database attribute does not appear to be set on the model: <Model: Wallet>

我的文件夹结构如下（我将省略__init__.py文件）：

models
|_ base_model.py
|_ currencies
   |_ wallet.py
|_ mixins
   |_ uuid_mixin.py

base_model.py：

db = SqliteDatabase('database.db',
                    pragmas={
                        'foreign_keys': 1,
                        'journal_mode': 'wal',
                    })


class BaseModel(Model):

    class Meta:
        database = db

uuid_mixin.py：

import uuid

from peewee import UUIDField, Model


class UUIDMixin(Model):
    id = UUIDField(primary_key=True, default=uuid.uuid4)

钱包.py：

from models import BaseModel, UUIDMixin


class Wallet(UUIDMixin, BaseModel):
    pass

我尝试将 Wallet 类中的继承顺序翻转为BaseModel, UUIDMixin，错误消失了，但它导致了 wallet 表的字段：{'id': <AutoField: Wallet.id>}，而不是我期望的 UUID。我的代码有什么问题？

标签： pythonmodeluuidmixinspeewee

解决方案

为了避免导致这种情况的多重继承问题，我将构建一个单独的基本模型，例如：

class BaseModel(Model):

    class Meta:
        database = db


class BaseModelWithUUID(BaseModel):
    id = UUIDField(primary_key=True, default=uuid.uuid4)


class Wallet(BaseModelWithUUID):
    pass

另外，我不知道您的实现细节，但在处理货币时，自定义字段而不是模型可能更有意义。

推荐阅读