app.py

import os
import json

from flask import Flask, jsonify
from flask_cors import CORS
from flask_pymongo import PyMongo
from dotenv import load_dotenv

from werkzeug.exceptions import BadRequest, InternalServerError, MethodNotAllowed, NotFound, ServiceUnavailable, Unauthorized

# CONFIGURATIONS
from configurations.database import mongo, database_credentials
from configurations.auth import jwt, bcrypt

# BLUEPRINTS

# AUTH
from controllers.auth.login import login_blueprint

# USER
from controllers.user.create_user import create_user_blueprint
from controllers.user.get_users import get_users_blueprint
from controllers.user.update_user import update_user_blueprint
from controllers.user.delete_user import delete_user_blueprint

# CUSTOMER
from controllers.customer.add_favorite import add_favorite_blueprint

# SUPPLIER
from controllers.supplier.get_supplier_stock import get_supplier_stock_blueprint

# CATEGORY
from controllers.category.create_category import create_category_blueprint
from controllers.category.get_categories import get_categories_blueprint
from controllers.category.update_category import update_category_blueprint
from controllers.category.delete_category import delete_category_blueprint

# PRODUCT
from controllers.product.create_product import create_product_blueprint
from controllers.product.get_products import get_products_blueprint
from controllers.product.update_product import update_product_blueprint
from controllers.product.delete_product import delete_product_blueprint
from controllers.product.rate_product import rate_product_blueprint
from controllers.product.search_products import search_products_blueprint

app = Flask(__name__)

CORS(app)

def create_app():
    # Database Connection
    app.config["MONGO_URI"] = "mongodb+srv://{}:{}@{}.vfd0n.mongodb.net/{}?retryWrites=true&w=majority".format(
        database_credentials["user"],
        database_credentials["password"],
        database_credentials["cluster"],
        database_credentials["database"]
    )
    mongo.init_app(app)

    # JWT Token Configuration
    app.config["JWT_SECRET_KEY"] = os.getenv("JWT_SECRET_KEY")
    app.config["JWT_ACCESS_TOKEN_EXPIRES"] = 604800

    jwt.init_app(app)
    bcrypt.init_app(app)

    # BLUEPRINTS REGISTRATION
    # AUTH
    app.register_blueprint(login_blueprint)

    # USER
    app.register_blueprint(create_user_blueprint)
    app.register_blueprint(get_users_blueprint)
    app.register_blueprint(update_user_blueprint)
    app.register_blueprint(delete_user_blueprint)

    # CUSTOMER
    app.register_blueprint(add_favorite_blueprint)

    # CATEGORY
    app.register_blueprint(create_category_blueprint)
    app.register_blueprint(get_categories_blueprint)
    app.register_blueprint(update_category_blueprint)
    app.register_blueprint(delete_category_blueprint)

    # PRODUCT
    app.register_blueprint(create_product_blueprint)
    app.register_blueprint(get_products_blueprint)
    app.register_blueprint(update_product_blueprint)
    app.register_blueprint(delete_product_blueprint)
    app.register_blueprint(rate_product_blueprint)
    app.register_blueprint(search_products_blueprint)
    app.register_blueprint(get_supplier_stock_blueprint)

    # -----------------------------------------------------
    # EXCEPTIONS
    @app.errorhandler(InternalServerError)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "status": str(status_code),
            "message": "default_{}".format(status_code),
        })

    @app.errorhandler(ServiceUnavailable)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "status": str(status_code),
            "message": "default_{}".format(status_code),
        })

    @app.errorhandler(BadRequest)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "status": str(status_code),
            "message": "default_{}".format(status_code),
        })

    @app.errorhandler(NotFound)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "status": str(status_code),
            "message": "default_{}".format(status_code),
        })

    @app.errorhandler(MethodNotAllowed)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "message": "default_{}".format(status_code),
            "status": str(status_code)
        })

    @app.errorhandler(Unauthorized)
    def handle_exception(e):
        status_code = json.dumps(e.code)
        return jsonify({
            "message": "default_{}".format(status_code),
            "status": str(status_code)
        })

    # JWT CUSTOM BEHAVIOUR
    @jwt.expired_token_loader
    def expired_token_handler(*argv):
        return jsonify({
            "message": "token_expired",
            "status": "401"
        })

    @jwt.invalid_token_loader
    def invalid_token_handler(*argv):
        return jsonify({
            "message": "invalid_token",
            "status": "401"
        })

    @jwt.needs_fresh_token_loader
    def needs_fresh_token_handler(*argv):
        return jsonify({
            "message": "requires_fresh_token",
            "status": "401"
        })

    @jwt.revoked_token_loader
    def revoked_token_handler(*argv):
        return jsonify({
            "message": "token_revoked",
            "status": "401"
        })

    @jwt.token_verification_failed_loader
    def token_verification_failed_handler(*argv):
        return jsonify({
            "message": "token_verification_fails",
            "status": "401"
        })

    @jwt.unauthorized_loader
    def unauthorized_handler(*argv):
        return jsonify({
            "message": "token_required",
            "status": "401"
        })

    return app

app = create_app()

if __name__ == "__main__":
    app.run()