在 Django 模型中使用 Python 數據類

Question

我的目標是在 Django PostgreSQL 中創建一個類似 struct 的對象，例如：

specs.coordinate.x
specs.coordinate.y
specs.coordinate.z

x , y , z因此應該是坐標的子類。 它們也應該是可變的，因此不能使用命名元組。

我已經嘗試使用新的數據類：

from django.db import models
from dataclasses import dataclass

class Specs(models.Model):
    name = models.CharField(max_length=80)
    age = models.IntegerField()

    @dataclass
    class coordinate:
        x: float
        y: float
        z: float

但是坐標 x,y,z項在 pgAdmin 中不可見：

我究竟做錯了什么？ 有沒有比數據類更好的方法？

Answer 1

如果您的目標只是在數據表中使用語法： specs.coordinate.x ，您可以簡單地使用OneToOne關系。

這也允許特殊情況，即必須設置所有坐標值或不設置。

from django.db import models

class Coordinate(models.Model):
    x = models.FloatField()
    y = models.FloatField()
    z = models.FloatField()


class Specs(models.Model):
    name = models.CharField(max_length=80)
    age = models.IntegerField()
    coordinate = models.OneToOneField(
        Coordinate, 
        null=True, 
        on_delete=models.SET_NULL
    )

現在您可以像使用 Dataclass 一樣使用 Specs。

Answer 2

定義數據類字段

該解決方案使用英安岩庫來實現對嵌套數據類的支持。

from dacite import from_dict
from django.db import models
from dataclasses import dataclass, asdict
import json

"""Field that maps dataclass to django model fields."""
class DataClassField(models.CharField):
    description = "Map python dataclasses to model."

    def __init__(self, dataClass, *args, **kwargs):
        self.dataClass = dataClass
        if not kwargs['max_length'] 
            kwargs['max_length'] = 1024
        super().__init__(*args, **kwargs)

    def deconstruct(self):
        name, path, args, kwargs = super().deconstruct()
        kwargs['dataClass'] = self.dataClass
        return name, path, args, kwargs

    def from_db_value(self, value, expression, connection):
        if value is None:
            return value
        obj = json.loads(value)
        return from_dict(data_class=self.dataClass, data=obj)

    def to_python(self, value):
        if isinstance(value, self.dataClass):
            return value
        if value is None:
            return value
        obj = json.loads(value)
        return from_dict(data_class=self.dataClass, data=obj)

    def get_prep_value(self, value):
        if value is None:
            return value
        return json.dumps(asdict(value)

示例用法

@dataclass
class MyDataclass1:
    foo: str

@dataclass
class MyDataclass2:
    bar: MyDataclass1

class MyModel(models.Model)
    dc1 = DataClassField(dataClass=MyDataclass1, null=True)
    dc2 = DataClassField(dataClass=MyDataclass2, null=True)

instance = MyModel(dc1=MyDataclass1(foo="foo"))
instance.save()
instance.dc1.foo # Should return "foo"
instance.dc1 # Should return <MyDataclass1 .... at ....>

更新

要添加對自定義數據類型（例如日期時間）的支持，您需要為這些類型指定 JSON 解碼和編碼，如下所示：

def decode_datetime(dt: datetime) -> str:
    return dt.strftime('%Y-%m-%dT%H:%M:%S%z')

def encode_datetime(dt: str) -> datetime:
    return datetime.strptime(dt, '%Y-%m-%dT%H:%M:%S%z')
    
def default(o):
    if isinstance(o, (datetime.datetime)):
        return decode_datetime(o)
    else:
        print(o)

class DataclassField(models.CharField):
    ...
    def from_db_value(self, value, expression, connection):
        ...
        obj = json.loads(value)
        return from_dict(data_class=self.data_class, data=obj,
                         config=Config(type_hooks={datetime.datetime: encode_datetime}))

    def to_python(self, value):
        ...
        return from_dict(data_class=self.data_class, data=obj,
                         config=Config(type_hooks={datetime.datetime: encode_datetime}))

    def get_prep_value(self, value):
        ...
        return json.dumps(asdict(value), sort_keys=True, indent=1, default=default)

Answer 3

我會使用 django 的內置PointField 。 它存儲 Point 對象。

所以你會使用

from django.contrib.gis.db import models


class Specs(models.Model):
    coordinate = models.PointField

    

在 django 中，您可以像使用普通 django 模型字段一樣訪問它：

spec.coordinate

由於是Point對象，您還可以執行以下操作：

spec.coordinate.x
spec.coordinate.y
spec.coordinate.z