app.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. from math import log
  2. from fastapi import FastAPI
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from starlette.middleware.sessions import SessionMiddleware
  5. from config.settings import DEVELOPMENT, SECRET_KEY
  6. from models import user
  7. from routes import sales
  8. from services.email_service import initialize_email_sender
  9. from middleware.no_cache import NoCacheMiddleware
  10. from logging import getLogger
  11. logger = getLogger(__name__)
  12. def create_app() -> FastAPI:
  13. """Create and configure FastAPI application"""
  14. logger.info("Creating FastAPI application")
  15. try:
  16. if DEVELOPMENT:
  17. logger.info("Creating FastAPI app in development mode")
  18. app = FastAPI(
  19. title="Web Pedidos Klein - FastAPI Backend",
  20. description="Backend for the Web Pedidos Klein application using FastAPI"
  21. )
  22. else:
  23. logger.info("Creating FastAPI app in production mode")
  24. app = FastAPI(
  25. title="Web Pedidos Klein - FastAPI Backend",
  26. description="Backend for the Web Pedidos Klein application using FastAPI",
  27. version="1.0.0"
  28. )
  29. # Initialize email sender
  30. logger.info("Initializing email sender")
  31. initialize_email_sender()
  32. logger.info("Email sender initialized")
  33. # Add CORS middleware
  34. logger.info("Adding CORS middleware")
  35. app.add_middleware(
  36. CORSMiddleware,
  37. allow_origins=["https://admin.kleinexpress.store"],
  38. allow_credentials=True,
  39. allow_methods=["*"],
  40. allow_headers=["*"],
  41. )
  42. # Add SessionMiddleware
  43. logger.info("Adding session middleware")
  44. app.add_middleware(
  45. SessionMiddleware,
  46. secret_key=SECRET_KEY,
  47. max_age=60 * 60 # max_age in seconds for Starlette
  48. )
  49. # Add NoCacheMiddleware
  50. logger.info("Adding no-cache middleware")
  51. app.add_middleware(NoCacheMiddleware)
  52. logger.info("FastAPI application created successfully")
  53. return app
  54. except Exception as e:
  55. error_msg = f"Error creating FastAPI application: {e}"
  56. logger.error(error_msg)
  57. raise
  58. def setup_routes(app: FastAPI):
  59. """Setup all application routes"""
  60. logger.info("Setting up application routes")
  61. try:
  62. from routes import chat, users, products, orders, static
  63. from fastapi import Depends
  64. from auth.security import get_current_user
  65. # Chat routes
  66. logger.info("Setting up chat routes")
  67. app.include_router(
  68. chat.chat_router,
  69. prefix="/api/chat",
  70. tags=["Chat"]
  71. )
  72. # User routes
  73. logger.info("Setting up user routes")
  74. app.include_router(users.user_router, prefix="/api/users", tags=["Users"])
  75. # Product routes
  76. logger.info("Setting up product routes")
  77. app.include_router(
  78. products.product_router,
  79. prefix="/api/products",
  80. tags=["Products"],
  81. dependencies=[Depends(get_current_user)]
  82. )
  83. # Order routes
  84. logger.info("Setting up order routes")
  85. app.include_router(
  86. orders.order_router,
  87. prefix="/api/orders",
  88. tags=["Orders"],
  89. dependencies=[Depends(get_current_user)]
  90. )
  91. # Sales routes
  92. logger.info("Setting up sales routes")
  93. app.include_router(
  94. sales.sales_router,
  95. prefix="/api/sales",
  96. tags=["Sales"],
  97. dependencies=[Depends(get_current_user)]
  98. )
  99. # Verification routes
  100. logger.info("Setting up verification routes")
  101. app.include_router(users.verify_router, prefix="/verify", tags=["Verification"])
  102. app.include_router(users.recovery_pin_router, prefix="/recovery", tags=["Recovery PIN"])
  103. # Static routes
  104. logger.info("Setting up static routes")
  105. from fastapi.responses import HTMLResponse
  106. app.add_api_route("/", static.serve_app_html, methods=["GET"],
  107. response_class=HTMLResponse, include_in_schema=False)
  108. app.add_api_route("/register", static.serve_register_html, methods=["GET"],
  109. response_class=HTMLResponse, include_in_schema=False)
  110. app.add_api_route("/images/{image_path:path}", static.serve_image, methods=["GET"],
  111. response_class=HTMLResponse, include_in_schema=False)
  112. # Mount static files
  113. logger.info("Mounting static file directories")
  114. static.mount_main_static_files(app)
  115. static.mount_register_static_files(app)
  116. logger.info("Application routes setup completed successfully")
  117. except Exception as e:
  118. error_msg = f"Error setting up application routes: - {e}"
  119. logger.error(error_msg)
  120. raise e