Make sure the EnumField gets an Enum subclass as its parameter
This commit is contained in:
parent
41b4b9d7ea
commit
a763662cd6
@ -17,6 +17,8 @@
|
|||||||
"""Forms for Calendar.social
|
"""Forms for Calendar.social
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
from flask_babelex import lazy_gettext as _
|
from flask_babelex import lazy_gettext as _
|
||||||
from flask_security.forms import LoginForm as BaseLoginForm
|
from flask_security.forms import LoginForm as BaseLoginForm
|
||||||
from flask_wtf import FlaskForm
|
from flask_wtf import FlaskForm
|
||||||
@ -173,6 +175,9 @@ class TimezoneField(SelectField):
|
|||||||
|
|
||||||
class EnumField(SelectField):
|
class EnumField(SelectField):
|
||||||
def __init__(self, enum_type, translations, *args, **kwargs):
|
def __init__(self, enum_type, translations, *args, **kwargs):
|
||||||
|
if not issubclass(enum_type, Enum):
|
||||||
|
raise TypeError('enum_type must be a subclass of Enum')
|
||||||
|
|
||||||
kwargs.update({'choices': [(value, None) for value in enum_type]})
|
kwargs.update({'choices': [(value, None) for value in enum_type]})
|
||||||
self.data = None
|
self.data = None
|
||||||
self.enum_type = enum_type
|
self.enum_type = enum_type
|
||||||
|
Loading…
Reference in New Issue
Block a user