from flask_admin import AdminIndexView
from flask_admin.contrib.sqla import ModelView
from flask_security import current_user


class SecureAdminIndexView(AdminIndexView):
    def is_accessible(self):
        return current_user.has_role("admin")


class SecureModelView(ModelView):
    can_export = True
    can_set_page_size = True

    def is_accessible(self):
        return current_user.has_role("admin")