mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 00:07:22 +00:00
17 lines
459 B
Python
17 lines
459 B
Python
from sqlalchemy.types import TypeDecorator
|
|
from sqlalchemy import Integer
|
|
|
|
class IntegerEnum(TypeDecorator):
|
|
impl = Integer
|
|
def __init__(self, enumtype, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self._enumtype = enumtype
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
return value
|
|
|
|
def process_result_value(self, value, dialect):
|
|
if value:
|
|
return self._enumtype(value)
|
|
return None
|