Source code for invenio_admin.forms

# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2015-2018 CERN.
#
# Invenio is free software; you can redistribute it and/or modify it
# under the terms of the MIT License; see LICENSE file for more details.

"""Flask-Admin form utilities."""

from __future__ import absolute_import, print_function


class LazyChoices(object):
    """Lazy form choices."""

    def __init__(self, func):
        """Initialize lazy choices.

        :param func: Function returning an iterable of choices.
        """
        self._func = func

    def __iter__(self):
        """Iterate over lazy choices."""
        return iter(self._func())