对peewee字段类型进行子类化以添加行为

时间:2018-05-14 10:33:00

标签: python-3.x peewee

我正在尝试将所需行为添加到CharFiledTextField,以便我可以存储列表列表并再次将其作为列表列表检索。我不是要求解决方案,而是希望看到一个示例,其中已经完成了已经支持的字段类型的子类化,因为我在文档或Internet中找不到任何内容。

我是否必须按照创建自定义类型的文档中的说明进行操作?

例如:

class mylistoflists(TextField):

如果是,那么我必须分配给field_type

1 个答案:

答案 0 :(得分:0)

示例代码(有关完整示例,请参阅tests / fields.py):

class ListField(TextField):
    def db_value(self, value):
        return ','.join(value) if value else ''

    def python_value(self, value):
        return value.split(',') if value else []


class Todo(TestModel):
    content = TextField()
    tags = ListField()


class TestCustomField(ModelTestCase):
    requires = [Todo]

    def test_custom_field(self):
        t1 = Todo.create(content='t1', tags=['t1-a', 't1-b'])
        t2 = Todo.create(content='t2', tags=[])

        t1_db = Todo.get(Todo.id == t1.id)
        self.assertEqual(t1_db.tags, ['t1-a', 't1-b'])

        t2_db = Todo.get(Todo.id == t2.id)
        self.assertEqual(t2_db.tags, [])

        t1_db = Todo.get(Todo.tags == Value(['t1-a', 't1-b'], unpack=False))
        self.assertEqual(t1_db.id, t1.id)