mirror of
https://github.com/Llloooggg/TextSouls.git
synced 2026-03-06 04:26:23 +03:00
34 lines
1007 B
Python
34 lines
1007 B
Python
from flask import Blueprint
|
|
from flask_admin.contrib.sqla import ModelView
|
|
|
|
from . import admin
|
|
from textsouls.models import db
|
|
|
|
from textsouls.models import User
|
|
|
|
from textsouls.models import CharacterRace
|
|
from textsouls.models import CharacterClass
|
|
from textsouls.models import Character
|
|
|
|
ts_admin = Blueprint("ts_admin", __name__)
|
|
|
|
|
|
class AdminView(ModelView):
|
|
def __init__(self, model, *args, **kwargs):
|
|
self.column_list = [c.key for c in model.__table__.columns]
|
|
self.form_columns = self.column_list
|
|
super(AdminView, self).__init__(model, *args, **kwargs)
|
|
|
|
|
|
class CommonView(ModelView):
|
|
def __init__(self, model, *args, **kwargs):
|
|
self.column_list = [c.key for c in model.__table__.columns]
|
|
super(CommonView, self).__init__(model, *args, **kwargs)
|
|
|
|
|
|
admin.add_view(AdminView(User, db.session))
|
|
|
|
admin.add_view(CommonView(CharacterRace, db.session))
|
|
admin.add_view(CommonView(CharacterClass, db.session))
|
|
admin.add_view(CommonView(Character, db.session))
|