Enough for today

This commit is contained in:
lvrossem
2023-03-27 23:19:59 +02:00
parent 765f3e9bef
commit fc63176642
9 changed files with 80 additions and 52 deletions

View File

@@ -1,4 +1,24 @@
from fastapi_utils.enums import StrEnum
from sqlalchemy.types import TypeDecorator, Enum
class StrEnumType(TypeDecorator):
impl = Enum
def __init__(self, enum_class, **kw):
self.enum_class = enum_class
super().__init__(enum_class, **kw)
def process_bind_param(self, value, dialect):
if value is None:
return None
return value.value
def process_result_value(self, value, dialect):
if value is None:
return None
return self.enum_class(value)
class MinigameEnum(StrEnum):
SpellingBee = 'SpellingBee'