blob: 070c02ac6dea8edef2713ce96096eb231e8afa92 (
plain)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
import click
from flask import Flask
from flask.cli import AppGroup
from config import Config
from app.extensions import db, Migrate, LoginManager
import flask_login
from flask_admin import Admin
from flask_admin.contrib.sqla import ModelView
class ModelView(ModelView):
def is_accessible(self):
is_admin = False
try:
is_admin = flask_login.current_user.admin
except AttributeError:
is_admin = False
return is_admin
def create_app(config_class=Config):
app = Flask(__name__)
app.config.from_object(config_class)
# Flask Extensions
db.init_app(app)
migrate = Migrate(app, db)
login_manager = LoginManager()
login_manager.login_view = 'auth.login'
login_manager.login_message = "You need to be logged in to view the page..."
login_manager.init_app(app)
from app.models.user import User
@login_manager.user_loader
def load_user(user_id):
return User.query.get(int(user_id))
admin = Admin(app, name="Flask-App", template_mode="bootstrap4")
admin.add_view(ModelView(User, db.session))
# CLI Commands
database_cli = AppGroup('database')
@database_cli.command('create')
def create_database():
db.create_all()
@database_cli.command('delete')
def delete_database():
db.drop_all()
@database_cli.command('superuser')
@click.argument('user_email')
def make_superuser(user_email):
user = User.query.filter_by(email=user_email).first()
if user:
print(f"Making {user.name} into admin")
user.admin = True
db.session.commit()
else:
print("User not found with email f{user_email}")
app.cli.add_command(database_cli)
# Blueprints
from app.auth import bp as auth_routes_bp
app.register_blueprint(auth_routes_bp)
from app.main import bp as main_bp
app.register_blueprint(main_bp)
return app
|
