如何在Tortoise-ORM

问题描述

标题差不多，我正在尝试设置Tortoise-ORM模型，其中将包括与Postgresql Array列相对应的字段。

似乎要正确执行此操作，我需要从asyncpg（因为它具有完整的数组支持）构建，然后扩展Tortoise Field。但是，我只是从Tortoise开始，也许有一些更好/更容易的前进方式/有人已经做了类似的事情。

解决方法

您需要实现自己的字段类型。这是我的实现：

from typing import List,Union,Type,Optional,Any
import json

from tortoise.fields.base import Field
from tortoise.models import Model


class IntArrayField(Field,list):
    """
    Int Array field specifically for PostgreSQL.

    This field can store list of int values.
    """

    SQL_TYPE = "int[]"

    def __init__(self,**kwargs):
        super().__init__(**kwargs)

    def to_db_value(
        self,value: List[int],instance: "Union[Type[Model],Model]"
    ) -> Optional[List[int]]:
        return value

    def to_python_value(self,value: Any) -> Optional[List[int]]:
        if isinstance(value,str):
            array = json.loads(value.replace("'",'"'))
            return [int(x) for x in array]
        return value

我没有动态类型字段的实现，这并不难。

在某些情况下，您需要在 to_db_value 和 to_python_value 转换中加入更多逻辑。例如，如果您使用 UUID[]。