backend: remove deprecation warnings

Author: pablogsal

backend/app/crud.py

@@ -257,7 +257,7 @@ async def create_benchmark_result(
         allocation_histogram=result.result_json.allocation_histogram,
         total_allocated_bytes=result.result_json.total_allocated_bytes,
         top_allocating_functions=[
-            func.dict() for func in result.result_json.top_allocating_functions
+            func.model_dump() for func in result.result_json.top_allocating_functions
         ],
         flamegraph_html=result.flamegraph_html,
     )

backend/app/factory.py

@@ -3,6 +3,8 @@
 import uuid
 import time
 import logging
+from contextlib import asynccontextmanager
+from typing import AsyncIterator
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -31,13 +33,44 @@ def create_app(settings=None) -> FastAPI:
     if settings is None:
         settings = get_settings()
 
+    @asynccontextmanager
+    async def lifespan(app: FastAPI) -> AsyncIterator[None]:
+        # Configure logging using the app state before the app starts serving.
+        app.state.logging_manager.configure_logging()
+
+        # Disable uvicorn access logs to avoid duplication
+        uvicorn_logger = logging.getLogger("uvicorn.access")
+        uvicorn_logger.disabled = True
+
+        logger = get_logger("api.startup")
+        logger.info(
+            "Application starting up",
+            extra={
+                "log_level": settings.log_level,
+                "log_format": settings.log_format,
+                "api_version": settings.api_version,
+            },
+        )
+        await create_tables()
+        logger.info("Database tables created successfully")
+
+        # Ensure initial admin user exists
+        from .database import AsyncSessionLocal
+        from .crud import ensure_initial_admin
+
+        async with AsyncSessionLocal() as db:
+            await ensure_initial_admin(db, settings.admin_initial_username)
+
+        yield
+
     # Create FastAPI instance
     app = FastAPI(
         title=settings.api_title,
         version=settings.api_version,
         docs_url="/api/docs",
         redoc_url="/api/redoc",
         openapi_url="/api/openapi.json",
+        lifespan=lifespan,
     )
 
     # Store dependencies in app state
@@ -133,35 +166,6 @@ async def log_requests(request: Request, call_next):
 
         return response
 
-    # Configure startup event
-    @app.on_event("startup")
-    async def startup_event():
-        # Configure logging using the app state
-        app.state.logging_manager.configure_logging()
-
-        # Disable uvicorn access logs to avoid duplication
-        uvicorn_logger = logging.getLogger("uvicorn.access")
-        uvicorn_logger.disabled = True
-
-        logger = get_logger("api.startup")
-        logger.info(
-            "Application starting up",
-            extra={
-                "log_level": settings.log_level,
-                "log_format": settings.log_format,
-                "api_version": settings.api_version,
-            },
-        )
-        await create_tables()
-        logger.info("Database tables created successfully")
-
-        # Ensure initial admin user exists
-        from .database import AsyncSessionLocal
-        from .crud import ensure_initial_admin
-
-        async with AsyncSessionLocal() as db:
-            await ensure_initial_admin(db, settings.admin_initial_username)
-
     # Include routers
     app.include_router(health.router)
     app.include_router(commits.router)

backend/app/models.py

@@ -11,8 +11,7 @@
     Boolean,
     UniqueConstraint,
 )
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import relationship
+from sqlalchemy.orm import declarative_base, relationship
 from datetime import datetime, UTC
 
 Base = declarative_base()

backend/app/schemas.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 from typing import List, Optional, Dict, Any, Tuple
 from datetime import datetime
 
@@ -22,8 +22,7 @@ class CommitCreate(CommitBase):
 
 
 class Commit(CommitBase):
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 class BinaryBase(BaseModel):
@@ -45,8 +44,7 @@ class BinaryCreate(BinaryBase):
 
 
 class Binary(BinaryBase):
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 class EnvironmentBase(BaseModel):
@@ -60,8 +58,7 @@ class EnvironmentCreate(EnvironmentBase):
 
 
 class Environment(EnvironmentBase):
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 class RunBase(BaseModel):
@@ -78,8 +75,7 @@ class RunCreate(RunBase):
 
 
 class Run(RunBase):
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 class TopAllocatingFunction(BaseModel):
@@ -111,8 +107,7 @@ class BenchmarkResultCreate(BaseModel):
 
 
 class BenchmarkResult(BenchmarkResultBase):
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 # Worker upload schemas
@@ -181,8 +176,7 @@ class AdminUserPublic(BaseModel):
     added_at: datetime
     is_active: bool
 
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
 
 class MemrayFailureReport(BaseModel):
@@ -204,5 +198,4 @@ class MemrayFailurePublic(BaseModel):
     failure_timestamp: datetime
     commit_timestamp: datetime
 
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)