Uk national

pyproject.toml

@@ -29,8 +29,10 @@ dependencies = [
     "numpy >= 1.25.0",
     "sentry-sdk >= 2.1.1",
     "pyhocon>=0.3.61",
+    "sqlalchemy>=2.0.44",
     "apitally[fastapi]>=0.22.3",
     "auth0-fastapi-api>=1.0.0b5",
+    "fastapi-cache2[memcache]>=0.2.2",
 ]
 
 [dependency-groups]
@@ -46,11 +48,13 @@ dev = [
     "types-pytz>=2025.2.0.20251108",
     "pandas-stubs>=2.3.2.250926",
     "pytest-asyncio>=1.3.0",
+    "freezegun>=1.5.5",
 ]
 
 [tool.uv.sources]
 dp-sdk = { url = "https://github.com/openclimatefix/data-platform/releases/download/v0.18.2/dp_sdk-0.18.2-py3-none-any.whl" }
 
+
 [project.urls]
 repository = "https://github.com/openclimatefix/quartz-api"
 

src/quartz_api/cmd/main.py

@@ -16,6 +16,8 @@
 from fastapi import FastAPI, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.utils import get_openapi
+from fastapi_cache import FastAPICache
+from fastapi_cache.backends.inmemory import InMemoryBackend
 from grpclib.client import Channel
 from pydantic import BaseModel
 from pyhocon import ConfigFactory, ConfigTree
@@ -29,6 +31,7 @@
 from ._logging import setup_json_logging
 
 log = logging.getLogger(__name__)
+# set hpack to warning log level
 logging.getLogger("hpack").setLevel(logging.WARNING)
 
 static_dir = pathlib.Path(__file__).parent.parent / "static"
@@ -67,7 +70,6 @@ def _custom_openapi(server: FastAPI) -> dict[str, Any]:
 
     return openapi_schema
 
-
 @asynccontextmanager
 async def _lifespan(server: FastAPI, conf: ConfigTree) -> Generator[None]:
     """Configure FastAPI app instance with startup and shutdown events."""
@@ -120,6 +122,9 @@ def _create_server(conf: ConfigTree) -> FastAPI:
         },
     )
 
+    # set up cache
+    FastAPICache.init(InMemoryBackend(), expire=120, prefix="fastapi-cache")
+
     # Add the default routes
     server.mount("/static", StaticFiles(directory=static_dir.as_posix()), name="static")