More generic implementation of fake enum db types

This commit is contained in:
Patrick Jentsch
2022-02-09 16:02:37 +01:00
parent 247ac801b0
commit 86d14f748f
5 changed files with 88 additions and 105 deletions

View File

@ -1,5 +1,5 @@
from app.auth import USERNAME_REGEX
from app.models import JobStatusMailNotificationLevel, User
from app.models import User, UserSettingJobStatusMailNotificationLevel
from flask_wtf import FlaskForm
from wtforms import (
BooleanField,
@ -96,5 +96,5 @@ class EditNotificationSettingsForm(FlaskForm):
super().__init__(*args, **kwargs)
self.job_status_mail_notification_level.choices += [
(enum_member.name, enum_member.name.capitalize())
for enum_member in JobStatusMailNotificationLevel
for enum_member in UserSettingJobStatusMailNotificationLevel
]

View File

@ -8,7 +8,7 @@ from .forms import (
EditNotificationSettingsForm
)
from .. import db
from ..models import JobStatusMailNotificationLevel
from ..models import UserSettingJobStatusMailNotificationLevel
@bp.route('', methods=['GET', 'POST'])
@ -57,7 +57,7 @@ def index():
and edit_notification_settings_form.validate()
):
current_user.setting_job_status_mail_notification_level = \
JobStatusMailNotificationLevel[
UserSettingJobStatusMailNotificationLevel[
edit_notification_settings_form.job_status_mail_notification_level.data # noqa
]
db.session.commit()