Reports

Action: file_editor view /app/backend/server.py

Observation: /app/backend/server.py:

1|from fastapi import FastAPI, APIRouter

2|from dotenv import load_dotenv

3|from starlette.middleware.cors import CORSMiddleware

4|from motor.motor_asyncio import AsyncIOMotorClient

5|import os

6|import logging

7|from pathlib import Path

8|from pydantic import BaseModel, Field, ConfigDict

9|from typing import List

10|import uuid

11|from datetime import datetime, timezone

12|

13|

14|ROOT_DIR = Path(_file_).parent

15|load_dotenv(ROOT_DIR / '.env')

16|

17|# MongoDB connection

18|mongo_url = os.environ['MONGO_URL']

19|client = AsyncIOMotorClient(mongo_url)

20|db = client[os.environ['DB_NAME']]

21|

22|# Create the main app without a prefix

23|app = FastAPI()

24|

25|# Create a router with the /api prefix

26|api_router = APIRouter(prefix="/api")

27|

28|

29|# Define Models

30|class StatusCheck(BaseModel):

31| model_config = ConfigDict(extra="ignore") # Ignore MongoDB's _id field

32|

33| id: str = Field(default_factory=lambda: str(uuid.uuid4()))

34| client_name: str

35| timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))

36|

37|class StatusCheckCreate(BaseModel):

38| client_name: str

39|

40|# Add your routes to the router instead of directly to app

41|@api_router.get("/")

42|async def root():

43| return {"message": "Hello World"}

44|

45|@api_router.post("/status", response_model=StatusCheck)

46|async def create_status_check(input: StatusCheckCreate):

47| status_dict = input.model_dump()

48| status_obj = StatusCheck(**status_dict)

49|

50| # Convert to dict and serialize datetime to ISO string for MongoDB

51| doc = status_obj.model_dump()

52| doc['timestamp'] = doc['timestamp'].isoformat()

53|

54| _ = await db.status_checks.insert_one(doc)

55| return status_obj

56|

57|@api_router.get("/status", response_model=List[StatusCheck])

58|async def get_status_checks():

59| # Exclude MongoDB's _id field from the query results

60| status_checks = await db.status_checks.find({}, {"_id": 0}).to_list(1000)

61|

62| # Convert ISO string timestamps back to datetime objects

63| for check in status_checks:

64| if isinstance(check['timestamp'], str):

65| check['timestamp'] = datetime.fromisoformat(check['timestamp'])

66|

67| return status_checks

68|

69|# Include the router in the main app

70|app.include_router(api_router)

71|

72|app.add_middleware(

73| CORSMiddleware,

74| allow_credentials=True,

75| allow_origins=os.environ.get('CORS_ORIGINS', '*').split(','),

76| allow_methods=["*"],

77| allow_headers=["*"],

78|)

79|

80|# Configure logging

81|logging.basicConfig(

82| level=logging.INFO,

83| format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'

84|)

85|logger = logging.getLogger(_name_)

86|