m4sk1n/calendar-social
1
0
Fork 0
forked from gergely/calendar-social
Code Issues Pull Requests Releases Wiki Activity

Make sure the EnumField gets an Enum subclass as its parameter

Browse Source
This commit is contained in:
Gergely Polonkai 2018-07-16 13:33:00 +02:00
parent 41b4b9d7ea
commit a763662cd6
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
calsocial/forms.py
View File

@ -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