dodalem backend dzialajacy

backend/my_fastapi.py

@@ -0,0 +1,114 @@
+import re
+import os
+from typing import Annotated
+
+from fastapi import Depends, FastAPI, HTTPException, Query, Request
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+
+from mayo.exceptions import (
+    MayoAuthError,
+    MayoConnectionError,
+    MayoError,
+    MayoOrderNotFound,
+    MayoParseError,
+    MayoResponseError,
+    MayoSessionError,
+)
+from mayo import MayoClient, MayoResponse
+
+app = FastAPI(title="Mayo Integration API")
+
+MAYO_BASE_URL = os.getenv("MAYO_BASE_URL", "http://10.8.0.6/mayo2")
+MAYO_LOGIN = os.getenv("MAYO_LOGIN", "nowakb")
+MAYO_PASSWORD = os.getenv("MAYO_PASSWORD", "def")
+ORDER_ID_RE = re.compile(
+    r"^(?P<order_num>\d{4})/(?P<year>\d{4})/(?P<item_idx>\d{1,2})$"
+)
+
+
+class OrderId(BaseModel):
+    order_num: str
+    year: str
+    item_idx: str
+
+
+@app.exception_handler(MayoOrderNotFound)
+async def mayo_order_not_found_handler(request: Request, exc: MayoOrderNotFound):
+    return JSONResponse(status_code=404, content={"detail": str(exc)})
+
+
+@app.exception_handler(MayoConnectionError)
+async def mayo_connection_error_handler(request: Request, exc: MayoConnectionError):
+    return JSONResponse(
+        status_code=503,
+        content={"detail": "Mayo service is currently unavailable."},
+    )
+
+
+@app.exception_handler(MayoAuthError)
+async def mayo_auth_error_handler(request: Request, exc: MayoAuthError):
+    return JSONResponse(
+        status_code=502,
+        content={"detail": "Authentication with Mayo failed."},
+    )
+
+
+@app.exception_handler(MayoSessionError)
+async def mayo_session_error_handler(request: Request, exc: MayoSessionError):
+    return JSONResponse(
+        status_code=502,
+        content={"detail": "Mayo session is invalid or has expired."},
+    )
+
+
+@app.exception_handler(MayoResponseError)
+async def mayo_response_error_handler(request: Request, exc: MayoResponseError):
+    return JSONResponse(
+        status_code=502,
+        content={"detail": "Mayo returned an unexpected response."},
+    )
+
+
+@app.exception_handler(MayoParseError)
+async def mayo_parse_error_handler(request: Request, exc: MayoParseError):
+    return JSONResponse(
+        status_code=502,
+        content={"detail": "Failed to parse data returned by Mayo."},
+    )
+
+
+@app.exception_handler(MayoError)
+async def mayo_error_handler(request: Request, exc: MayoError):
+    return JSONResponse(
+        status_code=500,
+        content={"detail": "Unexpected Mayo integration error."},
+    )
+
+
+def parse_order_id(
+    order_id: Annotated[str, Query(description="Format: XXXX/YYYY/ZZ")],
+) -> OrderId:
+    match = ORDER_ID_RE.fullmatch(order_id)
+    if not match:
+        raise HTTPException(
+            status_code=422,
+            detail="order_id must have format XXXX/YYYY/ZZ",
+        )
+
+    return OrderId(**match.groupdict())
+
+async def get_mayo_client():
+    async with MayoClient(MAYO_BASE_URL, MAYO_LOGIN, MAYO_PASSWORD) as client:
+        yield client
+
+
+@app.get("/orders/", response_model=MayoResponse)
+async def get_order(order: Annotated[OrderId, Depends(parse_order_id)], client: MayoClient = Depends(get_mayo_client)):
+    return await client.get_full_order_data(order.order_num, order.year, order.item_idx)
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=8001)