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

Uppercase and lowercase string enums behaviour updated.

Browse Source
This commit is contained in:
Evgeni (Gene) Makarov 2020-10-26 19:46:43 +03:00
parent ebdadb3aee
commit c4ab81ebe1
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -468,7 +468,10 @@ class BaseEnumField(Field):
return value return value
try: try:
if isinstance(value, str): if isinstance(value, str):
try:
return self.enum_cls[value] return self.enum_cls[value]
except Exception:
return self.enum_cls(value)
if isinstance(value, bytes): if isinstance(value, bytes):
return self.enum_cls[value.decode('UTF-8')] return self.enum_cls[value.decode('UTF-8')]
if isinstance(value, int): if isinstance(value, int):
@ -665,4 +668,3 @@ class LowCardinalityField(Field):
# Expose only relevant classes in import * # Expose only relevant classes in import *
__all__ = get_subclass_names(locals(), Field) __all__ = get_subclass_names(locals(), Field)