app.py 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  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. docs_url=None,
  29. redoc_url=None
  30. )
  31. # Initialize email sender
  32. logger.info("Initializing email sender")
  33. initialize_email_sender()
  34. logger.info("Email sender initialized")
  35. # Add CORS middleware
  36. logger.info("Adding CORS middleware")
  37. app.add_middleware(
  38. CORSMiddleware,
  39. allow_origins=["https://admin.kleinexpress.store"],
  40. allow_credentials=True,
  41. allow_methods=["*"],
  42. allow_headers=["*"],
  43. )
  44. # Add SessionMiddleware
  45. logger.info("Adding session middleware")
  46. app.add_middleware(
  47. SessionMiddleware,
  48. secret_key=SECRET_KEY,
  49. max_age=60 * 60 # max_age in seconds for Starlette
  50. )
  51. # Add NoCacheMiddleware
  52. logger.info("Adding no-cache middleware")
  53. app.add_middleware(NoCacheMiddleware)
  54. logger.info("FastAPI application created successfully")
  55. return app
  56. except Exception as e:
  57. error_msg = f"Error creating FastAPI application: {e}"
  58. logger.error(error_msg)
  59. raise
  60. def setup_routes(app: FastAPI):
  61. """Setup all application routes"""
  62. logger.info("Setting up application routes")
  63. try:
  64. from routes import chat, users, products, orders, static
  65. from fastapi import Depends
  66. from auth.security import get_current_user
  67. # Chat routes
  68. logger.info("Setting up chat routes")
  69. app.include_router(
  70. chat.chat_router,
  71. prefix="/api/chat",
  72. tags=["Chat"]
  73. )
  74. # User routes
  75. logger.info("Setting up user routes")
  76. app.include_router(users.user_router, prefix="/api/users", tags=["Users"])
  77. # Product routes
  78. logger.info("Setting up product routes")
  79. app.include_router(
  80. products.product_router,
  81. prefix="/api/products",
  82. tags=["Products"],
  83. dependencies=[Depends(get_current_user)]
  84. )
  85. # Order routes
  86. logger.info("Setting up order routes")
  87. app.include_router(
  88. orders.order_router,
  89. prefix="/api/orders",
  90. tags=["Orders"],
  91. dependencies=[Depends(get_current_user)]
  92. )
  93. # Sales routes
  94. logger.info("Setting up sales routes")
  95. app.include_router(
  96. sales.sales_router,
  97. prefix="/api/sales",
  98. tags=["Sales"],
  99. dependencies=[Depends(get_current_user)]
  100. )
  101. # Verification routes
  102. logger.info("Setting up verification routes")
  103. app.include_router(users.verify_router, prefix="/verify", tags=["Verification"])
  104. app.include_router(users.recovery_pin_router, prefix="/recovery", tags=["Recovery PIN"])
  105. # Static routes
  106. logger.info("Setting up static routes")
  107. from fastapi.responses import HTMLResponse
  108. app.add_api_route("/", static.serve_app_html, methods=["GET"],
  109. response_class=HTMLResponse, include_in_schema=False)
  110. app.add_api_route("/register", static.serve_register_html, methods=["GET"],
  111. response_class=HTMLResponse, include_in_schema=False)
  112. app.add_api_route("/images/{image_path:path}", static.serve_image, methods=["GET"],
  113. response_class=HTMLResponse, include_in_schema=False)
  114. # Mount static files
  115. logger.info("Mounting static file directories")
  116. static.mount_main_static_files(app)
  117. static.mount_register_static_files(app)
  118. logger.info("Application routes setup completed successfully")
  119. except Exception as e:
  120. error_msg = f"Error setting up application routes: - {e}"
  121. logger.error(error_msg)
  122. raise e