|
1 | 1 | from flask_admin import Admin, AdminIndexView |
2 | 2 | from flask_admin.contrib.sqla import ModelView |
3 | | -from flask import Flask, url_for, redirect, render_template, request, abort |
| 3 | +from flask import url_for, redirect, request |
4 | 4 | from app import db |
5 | 5 | from .models import Resource, Category, Language, User, Role |
6 | 6 | from flask_security import current_user |
|
9 | 9 | class AdminView(ModelView): |
10 | 10 | def is_accessible(self): |
11 | 11 | return current_user.has_role("admin") |
12 | | - |
| 12 | + |
13 | 13 | def inaccessible_callback(self): |
14 | 14 | return redirect(url_for("security.login", next=request.url)) |
15 | 15 |
|
| 16 | + |
16 | 17 | class HomeAdminView(AdminIndexView): |
17 | 18 | def is_accessible(self): |
18 | 19 | return current_user.has_role("admin") |
19 | | - |
| 20 | + |
20 | 21 | def inaccessible_callback(self, name): |
21 | 22 | return redirect(url_for("security.login", next=request.url)) |
22 | 23 |
|
23 | 24 |
|
24 | 25 | def run_flask_admin(app): |
25 | | - admin = Admin(app, name="Resources_api", url='/', index_view=HomeAdminView(name="Home")) |
| 26 | + admin = Admin(app, name="Resources_api", url='/', |
| 27 | + index_view=HomeAdminView(name="Home")) |
26 | 28 | admin.add_view(AdminView(Role, db.session)) |
27 | 29 | admin.add_view(AdminView(User, db.session)) |
28 | 30 | admin.add_view(AdminView(Resource, db.session)) |
29 | 31 | admin.add_view(AdminView(Category, db.session)) |
30 | 32 | admin.add_view(AdminView(Language, db.session)) |
31 | 33 | return admin |
32 | | - |
|
0 commit comments