Infinidat/infi.clickhouse_orm
1
1
Fork 0
mirror of https://github.com/Infinidat/infi.clickhouse_orm.git synced 2024-11-22 00:56:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow assignment of NO_VALUE to fields

Browse Source
This commit is contained in:
Itai Shirav 2020-05-28 23:02:55 +03:00
parent 56cf86a246
commit f0bef7f75d
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/infi/clickhouse_orm/models.py
View File

@ -170,7 +170,7 @@ class Model(metaclass=ModelBase):
This may raise a `ValueError`. This may raise a `ValueError`.
''' '''
field = self.get_field(name) field = self.get_field(name)
if field: if field and (value != NO_VALUE):
try: try:
value = field.to_python(value, pytz.utc) value = field.to_python(value, pytz.utc)
field.validate(value) field.validate(value)

4
tests/test_alias_fields.py
View File

@ -60,6 +60,10 @@ class AliasFieldsTest(unittest.TestCase):
def test_default_value(self): def test_default_value(self):
instance = ModelWithAliasFields() instance = ModelWithAliasFields()
self.assertEqual(instance.alias_str, NO_VALUE) self.assertEqual(instance.alias_str, NO_VALUE)
# Check that NO_VALUE can be assigned to a field
instance.str_field = NO_VALUE
# Check that NO_VALUE can be assigned when creating a new instance
instance2 = ModelWithAliasFields(**instance.to_dict())
class ModelWithAliasFields(Model): class ModelWithAliasFields(Model):