enpaul/peewee-plus
1
0
Fork 0
mirror of https://github.com/enpaul/peewee-plus.git synced 2024-11-14 18:46:47 +00:00
Code Issues Projects Releases Wiki Activity

Ensure NULL-able Enums don't raise peewee.IntegrityError

Browse Source
This commit is contained in:
Evgeny Chernyavskiy 2022-07-20 14:42:44 -04:00 committed by Ethan Paul
parent 44ad5da339
commit dac554e8d4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
peewee_plus.py
View File

@ -367,7 +367,11 @@ class EnumField(peewee.CharField): # pylint: disable=abstract-method
def python_value(self, value: str) -> enum.Enum: def python_value(self, value: str) -> enum.Enum:
try: try:
return self.enumeration[super().python_value(value)] return (
None
if value is None and self.null
else self.enumeration[super().python_value(value)]
)
except KeyError: except KeyError:
raise peewee.IntegrityError( raise peewee.IntegrityError(
f"Enum {self.enumeration.__name__} has no value with name '{value}'" f"Enum {self.enumeration.__name__} has no value with name '{value}'"