from flask_login import current_user
from wtforms import BooleanField, SelectField
from ..models import Role
from ..settings.forms import EditGeneralSettingsForm


class EditGeneralSettingsAdminForm(EditGeneralSettingsForm):
    confirmed = BooleanField('Confirmed')
    role = SelectField('Role', coerce=int)

    def __init__(self, user=current_user, *args, **kwargs):
        super().__init__(*args, user=user, **kwargs)
        self.role.choices = [(role.id, role.name)
                             for role in Role.query.order_by(Role.name).all()]
        self.user = user