init commit

transportmanager/server/app.py

@@ -0,0 +1,93 @@
+import os
+from flask import Flask, jsonify
+from routes.auth import auth_bp
+from flask_jwt_extended import JWTManager
+from routes.profile import profile_bp
+from routes.clients import clients_bp
+from routes.transporters import transporters_bp
+from routes.destinations import destinations_bp
+from routes.orders_out import orders_bp
+from routes.ouders_in import orders_in_bp
+from routes.report import report_bp
+from admin.subscription import admin_subscription_bp
+from routes.subscription import subscription_bp
+from admin.tenants import admin_user_bp
+
+from apscheduler.schedulers.background import BackgroundScheduler
+from models.subscription import Subscription
+
+from flask_cors import CORS
+
+app = Flask(__name__)
+CORS(
+    app,
+    resources={r"/*": {"origins": [os.getenv("WEB_ORIGIN", "*")]}},
+    allow_headers=["Authorization", "Content-Type"],
+    expose_headers=["Content-Type"],
+)
+
+@app.get("/db/check")
+def db_check():
+    try:
+        import psycopg  # psycopg3 client
+    except Exception as e:
+        return {"ok": False, "error": f"psycopg not available: {e}"}, 500
+
+    dsn = os.getenv("DATABASE_URL")
+    if not dsn:
+        return {"ok": False, "error": "DATABASE_URL not set"}, 500
+
+    try:
+        with psycopg.connect(dsn, connect_timeout=5) as conn:
+            with conn.cursor() as cur:
+                cur.execute("SELECT version();")
+                ver = cur.fetchone()[0]
+        return {"ok": True, "version": ver}, 200
+    except Exception as e:
+        return {"ok": False, "error": str(e)}, 500
+
+@app.get("/health")
+def health():
+    return {"ok": True}, 200
+
+app.config["JWT_SECRET_KEY"] = os.environ.get("JWT_SECRET_KEY", "your-jwt-secret")
+app.config["JWT_TOKEN_LOCATION"] = ["headers", "query_string"]
+app.config["JWT_QUERY_STRING_NAME"] = "token"
+jwt = JWTManager(app)
+
+env = os.environ.get("FLASK_ENV", "development")
+
+# Register blueprints
+app.register_blueprint(auth_bp, url_prefix="/auth")
+app.register_blueprint(profile_bp, url_prefix="/profile")
+app.register_blueprint(clients_bp)
+app.register_blueprint(transporters_bp, url_prefix="/transporters")
+app.register_blueprint(destinations_bp, url_prefix="/destinations")
+app.register_blueprint(orders_bp, url_prefix="/orders")
+app.register_blueprint(orders_in_bp, url_prefix="/orders_in")
+app.register_blueprint(report_bp, url_prefix="/report")
+app.register_blueprint(admin_subscription_bp)
+app.register_blueprint(subscription_bp)
+app.register_blueprint(admin_user_bp)
+
+
+def update_subscription_statuses_job():
+    print("[Scheduler] Running daily subscription status check...")
+    subscription_model = Subscription()
+    subscription_model.update_subscription_statuses()
+
+RUN_SCHEDULER = os.getenv("RUN_SCHEDULER", "1") == "1"
+
+if RUN_SCHEDULER:
+    scheduler = BackgroundScheduler(daemon=True)
+    scheduler.add_job(func=update_subscription_statuses_job, trigger="interval", days=1)
+    scheduler.start()
+
+
+if __name__ == "__main__":
+    if env != "production":
+        # Avoid running the scheduler twice in development mode
+        import logging
+        logging.getLogger("apscheduler").setLevel(logging.DEBUG)
+
+    app.run(debug=(env == "development"), use_reloader=False)