from flask_admin import AdminIndexView as OldAdminIndexView
from flask_admin.contrib.sqla import ModelView as OldModelView
from flask_security import current_user


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


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