-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
83 lines (69 loc) · 2.74 KB
/
app.py
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
74
75
76
77
78
79
80
81
82
83
import os
from flask import Flask, jsonify
from flask_smorest import Api
from flask_jwt_extended import JWTManager
from flask_admin import Admin
from flask_migrate import Migrate
from models import *
from db import db
from resources.item import blp as ItemBlueprint
from resources.store import blp as StoreBlueprint
from resources.tag import blp as TagBlueprint
from resources.user import blp as UserBlueprint
from resources.cart import blp as CartBlueprint
from resources.order import blp as OrderBluePrint
from dotenv import load_dotenv
from flask_admin.contrib.sqla import ModelView
def create_app(db_url=None):
app = Flask(__name__)
load_dotenv()
app.config["API_TITLE"] = "Stores REST API"
app.config["API_VERSION"] = "v1"
app.config["OPENAPI_VERSION"] = "3.0.3"
app.config["OPENAPI_URL_PREFIX"] = "/"
app.config["OPENAPI_SWAGGER_UI_PATH"] = "/"
app.config[
"OPENAPI_SWAGGER_UI_URL"
] = "https://cdn.jsdelivr.net/npm/swagger-ui-dist/"
app.config["SQLALCHEMY_DATABASE_URI"] = db_url or os.getenv(
"DATABASE_URL", "sqlite:///data.db")
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
app.config["PROPAGATE_EXCEPTIONS"] = True
app.config["FLASK_ADMIN_SWATCH"] = "cerulean"
app.config["SESSION_TYPE"] = "filesystem"
app.secret_key = os.getenv("SECRET_KEY", "super-secret-key")
db.init_app(app)
migrate = Migrate(app, db)
api = Api(app)
app.config["JWT_SECRET_KEY"] = os.getenv(
"JWT_SECRET_KEY", "super-secret-key")
jwt = JWTManager(app)
@jwt.additional_claims_loader
def add_claims_to_jwt(identity):
if identity == 1:
return {"is_admin": True}
return {"is_admin": False}
@jwt.expired_token_loader
def expired_token_callback(jwt_header, jwt_payload):
return (
jsonify({"message": "The token has expired.",
"error": "token_expired"}),
401,
)
api.register_blueprint(ItemBlueprint)
api.register_blueprint(StoreBlueprint)
api.register_blueprint(TagBlueprint)
api.register_blueprint(UserBlueprint)
api.register_blueprint(CartBlueprint)
api.register_blueprint(OrderBluePrint)
admin = Admin(app, name='ُُE-Commerce-API ', template_mode='bootstrap3')
admin.add_view(ModelView(UserModel, db.session))
admin.add_view(ModelView(ItemModel, db.session))
admin.add_view(ModelView(StoreModel, db.session))
admin.add_view(ModelView(TagModel, db.session))
admin.add_view(ModelView(CollectionModel, db.session))
admin.add_view(ModelView(CartModel, db.session))
admin.add_view(ModelView(CartItemModel, db.session))
admin.add_view(ModelView(OrderItemModel, db.session))
admin.add_view(ModelView(OrderModel, db.session))
return app