from flaskbase.admin import ModelView
from flaskbase.extensions import admin, db
from flatpages.models import Page
from flask_pagedown.fields import PageDownField


class PageModelView(ModelView):
    roles = ['editor']

    column_list = ('slug', 'title', 'updated')
    form_columns = ('title', 'slug', 'content')
    form_extra_fields = {
        'content': PageDownField('Content'),
        }

    form_widget_args = {
        'content': {
            'rows': 10,
            }
        }

admin.add_view(PageModelView(Page, db.session))