kcsrv/admin.py at master · KanColleTool/kcsrv · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from flask.ext.admin import Admin, AdminIndexView, BaseView
from flask.ext.admin.contrib.sqla import ModelView
from flask.ext.security import current_user
from wtforms import fields
from db import db, User, Role, Recipe, RecipeResources, Admiral, Kanmusu, Equipment, Fleet


class AdminAuthMixin(object):
    def is_accessible(self):
        return current_user.has_role('admin')


class MyAdminIndexView(AdminAuthMixin, AdminIndexView):
    pass


class MyBaseView(AdminAuthMixin, BaseView):
    pass


class MyModelView(AdminAuthMixin, ModelView):
    pass


class RoleModelView(MyModelView):
    form_overrides = {'description': fields.TextAreaField}


class UserModelView(MyModelView):
    column_exclude_list = ['password']
    form_excluded_columns = ['password']

    inline_models = (Admiral,)


class AdmiralModelView(MyModelView):
    column_exclude_list = ['tutorial_progress', 'max_furniture', 'furniture', 'furniture_coins', 'max_ships',
                           'max_equips', 'available_fleets', 'available_cdocks', 'available_rdocks', 'sortie_successes',
                           'sortie_total',
                           'expedition_successes', 'expedition_total', 'pvp_successes', 'pvp_total', 'setup']


admin = Admin(index_view=MyAdminIndexView(), template_mode='bootstrap3')
admin.add_view(RoleModelView(Role, db.session, endpoint='role'))
admin.add_view(UserModelView(User, db.session, endpoint='user'))
admin.add_view(AdmiralModelView(Admiral, db.session, endpoint='admiral'))
admin.add_view(MyModelView(Kanmusu, db.session, endpoint='shipgirl'))
admin.add_view(MyModelView(Recipe, db.session, endpoint='recipe'))
admin.add_view(MyModelView(RecipeResources, db.session, endpoint='rr'))
admin.add_view(MyModelView(Equipment, db.session, endpoint='equip'))
admin.add_view(MyModelView(Fleet, db.session, endpoint='fleet'))