|
| 1 | +"""Packages""" |
| 2 | + |
| 3 | +import argparse |
| 4 | +import os |
| 5 | +import time |
| 6 | +from typing import Any |
| 7 | + |
| 8 | +import flask |
| 9 | +import flask_cors # type: ignore |
| 10 | +from flask import Flask, Response |
| 11 | +from flask_cors import cross_origin |
| 12 | +from werkzeug.exceptions import HTTPException |
| 13 | + |
| 14 | +from opengeodeweb_back import utils_functions, app_config |
| 15 | +from opengeodeweb_back.routes import blueprint_routes |
| 16 | +from opengeodeweb_back.routes.models import blueprint_models |
| 17 | +from opengeodeweb_microservice.database.connection import init_database |
| 18 | + |
| 19 | + |
| 20 | +""" Global config """ |
| 21 | +app: Flask = flask.Flask(__name__) |
| 22 | + |
| 23 | +""" Config variables """ |
| 24 | +FLASK_DEBUG = True if os.environ.get("FLASK_DEBUG", default=None) == "True" else False |
| 25 | + |
| 26 | +if FLASK_DEBUG == False: |
| 27 | + app.config.from_object(app_config.ProdConfig) |
| 28 | +else: |
| 29 | + app.config.from_object(app_config.DevConfig) |
| 30 | + |
| 31 | +DEFAULT_HOST: str = app.config.get("DEFAULT_HOST") or "localhost" |
| 32 | +DEFAULT_PORT: int = int(app.config.get("DEFAULT_PORT") or 5000) |
| 33 | +DEFAULT_DATA_FOLDER_PATH: str = app.config.get("DEFAULT_DATA_FOLDER_PATH") or "./data" |
| 34 | +ORIGINS: Any = app.config.get("ORIGINS") |
| 35 | +TIMEOUT: int = int(app.config.get("MINUTES_BEFORE_TIMEOUT") or 30) |
| 36 | +SSL: Any = app.config.get("SSL") |
| 37 | +SECONDS_BETWEEN_SHUTDOWNS: float = float( |
| 38 | + app.config.get("SECONDS_BETWEEN_SHUTDOWNS") or 60.0 |
| 39 | +) |
| 40 | + |
| 41 | + |
| 42 | +app.register_blueprint( |
| 43 | + blueprint_routes.routes, |
| 44 | + url_prefix="/opengeodeweb_back", |
| 45 | + name="opengeodeweb_back", |
| 46 | +) |
| 47 | + |
| 48 | +app.register_blueprint( |
| 49 | + blueprint_models.routes, |
| 50 | + url_prefix="/opengeodeweb_back/models", |
| 51 | + name="opengeodeweb_models", |
| 52 | +) |
| 53 | + |
| 54 | +if FLASK_DEBUG == False: |
| 55 | + utils_functions.set_interval( |
| 56 | + utils_functions.kill_task, SECONDS_BETWEEN_SHUTDOWNS, app |
| 57 | + ) |
| 58 | + |
| 59 | + |
| 60 | +@app.errorhandler(HTTPException) |
| 61 | +def errorhandler(e: HTTPException) -> tuple[dict[str, Any], int] | Response: |
| 62 | + return utils_functions.handle_exception(e) |
| 63 | + |
| 64 | + |
| 65 | +@app.route( |
| 66 | + "/error", |
| 67 | + methods=["POST"], |
| 68 | +) |
| 69 | +def return_error() -> Response: |
| 70 | + flask.abort(500, f"Test") |
| 71 | + return flask.make_response({}, 500) |
| 72 | + |
| 73 | + |
| 74 | +@app.route("/", methods=["POST"]) |
| 75 | +@cross_origin() |
| 76 | +def root() -> Response: |
| 77 | + return flask.make_response({}, 200) |
| 78 | + |
| 79 | + |
| 80 | +@app.route("/kill", methods=["POST"]) |
| 81 | +@cross_origin() |
| 82 | +def kill() -> None: |
| 83 | + print("Manual server kill, shutting down...", flush=True) |
| 84 | + os._exit(0) |
| 85 | + |
| 86 | + |
| 87 | +def run_server() -> None: |
| 88 | + parser = argparse.ArgumentParser( |
| 89 | + prog="OpenGeodeWeb-Back", description="Backend server for OpenGeodeWeb" |
| 90 | + ) |
| 91 | + parser.add_argument("--host", type=str, default=DEFAULT_HOST, help="Host to run on") |
| 92 | + parser.add_argument( |
| 93 | + "-p", "--port", type=int, default=DEFAULT_PORT, help="Port to listen on" |
| 94 | + ) |
| 95 | + parser.add_argument( |
| 96 | + "-d", |
| 97 | + "--debug", |
| 98 | + default=FLASK_DEBUG, |
| 99 | + help="Whether to run in debug mode", |
| 100 | + action="store_true", |
| 101 | + ) |
| 102 | + parser.add_argument( |
| 103 | + "-dfp", |
| 104 | + "--data_folder_path", |
| 105 | + type=str, |
| 106 | + default=DEFAULT_DATA_FOLDER_PATH, |
| 107 | + help="Path to the folder where data is stored", |
| 108 | + ) |
| 109 | + parser.add_argument( |
| 110 | + "-ufp", |
| 111 | + "--upload_folder_path", |
| 112 | + type=str, |
| 113 | + default=DEFAULT_DATA_FOLDER_PATH, |
| 114 | + help="Path to the folder where uploads are stored", |
| 115 | + ) |
| 116 | + parser.add_argument( |
| 117 | + "-origins", |
| 118 | + "--allowed_origins", |
| 119 | + default=ORIGINS, |
| 120 | + help="Origins that are allowed to connect to the server", |
| 121 | + ) |
| 122 | + parser.add_argument( |
| 123 | + "-t", |
| 124 | + "--timeout", |
| 125 | + default=TIMEOUT, |
| 126 | + help="Number of minutes before the server times out", |
| 127 | + ) |
| 128 | + args = parser.parse_args() |
| 129 | + |
| 130 | + app.config.update(DATA_FOLDER_PATH=args.data_folder_path) |
| 131 | + app.config.update(UPLOAD_FOLDER=args.upload_folder_path) |
| 132 | + app.config.update(MINUTES_BEFORE_TIMEOUT=args.timeout) |
| 133 | + |
| 134 | + flask_cors.CORS(app, origins=args.allowed_origins) |
| 135 | + |
| 136 | + print( |
| 137 | + f"Host: {args.host}, Port: {args.port}, Debug: {args.debug}, " |
| 138 | + f"Data folder path: {args.data_folder_path}, Timeout: {args.timeout}, " |
| 139 | + f"Origins: {args.allowed_origins}", |
| 140 | + flush=True, |
| 141 | + ) |
| 142 | + |
| 143 | + db_filename: str = app.config.get("DATABASE_FILENAME") or "database.db" |
| 144 | + db_path = os.path.join(args.data_folder_path, db_filename) |
| 145 | + os.makedirs(os.path.dirname(db_path), exist_ok=True) |
| 146 | + app.config["SQLALCHEMY_DATABASE_URI"] = f"sqlite:///{db_path}" |
| 147 | + app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False |
| 148 | + init_database(app, db_filename) |
| 149 | + print(f"Database initialized at: {db_path}", flush=True) |
| 150 | + |
| 151 | + app.run(debug=args.debug, host=args.host, port=args.port, ssl_context=SSL) |
| 152 | + |
| 153 | + |
| 154 | +# ''' Main ''' |
| 155 | +if __name__ == "__main__": |
| 156 | + run_server() |
0 commit comments