CTFd/CTFd/forms/self.py

from flask import session
from wtforms import PasswordField, SelectField, StringField
from wtforms.fields.html5 import DateField, URLField

from CTFd.forms import BaseForm
from CTFd.forms.users import attach_custom_user_fields
from CTFd.forms.fields import SubmitField
from CTFd.models import FieldEntries, UserFields
from CTFd.utils.countries import SELECT_COUNTRIES_LIST


def SettingsForm(*args, **kwargs):
    class _SettingsForm(BaseForm):
        name = StringField("User Name")
        email = StringField("Email")
        password = PasswordField("Password")
        confirm = PasswordField("Current Password")
        affiliation = StringField("Affiliation")
        website = URLField("Website")
        country = SelectField("Country", choices=SELECT_COUNTRIES_LIST)
        submit = SubmitField("Submit")

        @property
        def extra(self):
            fields = []
            new_fields = UserFields.query.filter_by(editable=True).all()
            user_fields = {}

            for f in FieldEntries.query.filter_by(user_id=session["id"]).all():
                user_fields[f.field_id] = f.value

            for field in new_fields:
                form_field = getattr(self, f"fields[{field.id}]")
                initial = user_fields.get(field.id, "")
                form_field.data = initial
                if form_field.render_kw:
                    form_field.render_kw["data-initial"] = initial
                else:
                    form_field.render_kw = {"data-initial": initial}
                entry = (field.name, form_field)
                fields.append(entry)
            return fields

    attach_custom_user_fields(_SettingsForm, editable=True)

    return _SettingsForm(*args, **kwargs)


class TokensForm(BaseForm):
    expiration = DateField("Expiration")
    submit = SubmitField("Generate")