Как динамически загрузить выбор для одного из полей form_extra_fields в ModelView во время обслуживания данных для рендеринга?

Question

У меня есть User admin ModelView с form_extra_field constituency, и я хотел, чтобы он динамически загружался выборы (это SelectField). Модель для этого представления - User, в которой группа будет храниться как id элемента.

Я пытался загрузить выборы из удаленного API, но, конечно, было неподходящее время для загрузки этих вещейиз-за RuntimeError: Работа вне контекста приложения. Это произошло при команде миграции ...venv/bin/flask db init -d ctiweb/migrations. Я прочитал некоторый API_KEY из объекта конфигурации, который должен иметь контекст приложения ...

Таким образом, решением будет выбор загрузки при обслуживании или рендеринге времени. Я не хочу иметь решение в настроенном шаблоне, а хочу в настроенном администраторе ModelView для User модели.

Есть ли такое решение? Могу ли я перегрузить какой-то метод ModelView, который уже имеет контекст приложения (как, например, во время обслуживания данных или шаблона рендеринга)?

часть исходного кода из models.py

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    password_hash = db.Column(db.String(128))
    email = db.Column(db.String(255), nullable=False)
    is_superuser = db.Column(db.Boolean())
    constituency_id = db.Column(db.Integer())

    def __str__(self):
        return self.username

    @property
    def password(self):
        raise AttributeError("not readable attribute")

    @password.setter
    def password(self, password):
        self.password_hash = generate_password_hash(password)

    def verify_password(self, password):
        return check_password_hash(self.password_hash, password)

часть исходного кодаот admin.py

from flask_admin.contrib.sqla import ModelView

class UserAdminModelView(ModelView):

    def serve_constituency_choices():
        consituencies = (
            get_constituency(constituency_id) for constituency_id in get_constituencies()
        )
        return ((constituency['id'], constituency['name']) for constituency in consituencies)

    column_list = ['username', 'email', 'is_superuser']
    form_columns = ['username', 'email', 'password', 'is_superuser', 'constituency']
    form_extra_fields = {
        'password': PasswordField('Password'),
        'constituency': SelectField('Constituency', choices=serve_constituency_choices())
    }

Answer 1

На самом деле, я наконец-то нашел более простое решение - только с итератором класса по выбору избирателей. Я думаю, что лучше, чем первый: -)

class ConstituencyChoices(Iterator):
    def __init__(self, first_item=None):
        self.first_item = first_item
        self.constituencies = None

    def __next__(self):
        if self.first_item is not None:
            res = self.first_item
            self.first_item = None
            return res
        if self.constituencies is None:
            self.constituencies = (
                get_constituency(constituency_id) for constituency_id in get_constituencies()
            )
        constituency = self.constituencies.__next__()
        res = (constituency['id'], constituency['name'])
        return res

# pylint: disable=too-many-ancestors
class UserAdminModelView(ModelView):

    column_list = ['username', 'email', 'is_superuser']
    form_columns = ['username', 'email', 'password', 'is_superuser', 'constituency']
    form_extra_fields = {
        'password': PasswordField('Password'),
        'constituency': SelectField(
            'Constituency',
            coerce=int,
            choices=ConstituencyChoices(
                (0, "--- {} ---".format(_('no constituency')))
            )
        )
    }

Answer 2

В ModelView есть методы - create_form и edit_form, которые могут быть перегружены и возвращать форму с правильным выбором, загруженным из удаленного API. Посмотрите документацию https://flask -admin.readthedocs.io / en / latest / api / mod_model / # flask_admin.model.BaseModelView.create_form и https://flask -admin.readthedocs.io /ru / latest / api / mod_model / # flask_admin.model.BaseModelView.edit_form .

class UserAdminModelView(ModelView):

    column_list = ['username', 'email', 'is_superuser']
    form_columns = ['username', 'email', 'password', 'is_superuser', 'constituency']
    form_extra_fields = {
        'password': PasswordField('Password'),
        'constituency': SelectField('Constituency', coerce=int)
    }

    class ConstituencyChoices(Iterator):
        def __init__(self, constituencies, first_item=None):
            self.first_item = first_item
            self.constituencies = constituencies

        def __next__(self):
            if self.first_item is not None:
                res = self.first_item
                self.first_item = None
                return res
            constituency = self.constituencies.__next__()
            res = (constituency['id'], constituency['name'])
            return res

    def serve_constituency_choices(self):
        constituencies = (
            get_constituency(constituency_id) for constituency_id in get_constituencies()
        )
        return self.ConstituencyChoices(constituencies, (0, "-----"))

    def create_form(self, obj=None):
        form = super().create_form(obj=obj)
        form.constituency.choices = self.serve_constituency_choices()
        return form

    def edit_form(self, obj=None):
        form = super().edit_form(obj=obj)
        form.constituency.choices = self.serve_constituency_choices()
        return form