Infinidat/infi.clickhouse_orm
1
1
Fork 0
mirror of https://github.com/Infinidat/infi.clickhouse_orm.git synced 2025-07-25 23:39:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

Minor refactoring of extra_null_values

Browse Source
This commit is contained in:
Itai Shirav 2017-06-23 11:10:49 +03:00
parent 7d3b3129c5
commit 2c0ef08ee1
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/infi/clickhouse_orm/fields.py
View File

@ -370,10 +370,9 @@ class NullableField(Field):
def __init__(self, inner_field, default=None, alias=None, materialized=None,
extra_null_values=None):
self.inner_field = inner_field
if extra_null_values is None:
self._extra_null_values = list()
else:
self._extra_null_values = extra_null_values
self._null_values = [None]
if extra_null_values:
self._null_values.extend(extra_null_values)
super(NullableField, self).__init__(default, alias, materialized)
def to_python(self, value, timezone_in_use):
@ -385,7 +384,7 @@ class NullableField(Field):
value is None or self.inner_field.validate(value)
def to_db_string(self, value, quote=True):
if value is None or value in self._extra_null_values:
if value in self._null_values:
return '\\N'
return self.inner_field.to_db_string(value, quote=quote)