Add methods for set cache_age in openapi schema

[Nottezz]

Автоматическое проставления cache-age в OpenAPI схему.

[chud0]

bugbot run

[chud0]

а как она обычно подставляется? мы не ломаем какие то едж кейсы про генерацию схемы?

[Nottezz]

Скорее всего ещё надо в начале прокинуть проверку схему. Доке ФастАПИ есть пример кастомной схемы - https://fastapi.tiangolo.com/how-to/extending-openapi/#normal-fastapi

Там прям так и идёт:

app.openapi_schema = openapi_schema
return app.openapi_schema

Единственное я ничего не возвращаю.

[chud0]

а в app.openapi_schema когда метод выполняется ничего нет? может схема уже готова

[chud0]

бездушная машина тоже говорит ... by always calling get_openapi(), which bypasses any custom OpenAPI schema generation logic

[Nottezz]

Там логика такая: мы запрашиваем уже готовую схему от app, её редактируем, добавляем новые поля в ручки и передаём обновлённую схему. В прошлом варианте мы получали эту схему и целиком перезаписывали её, тем самым все пользовательские схемы перетирали.

[chud0]

bugbot run

[cursor]

Bug: Lifespan Event Missing Type Check

The on_lifespan method calls set_cache_age_in_openapi_schema with scope["app"] without a FastAPI type check, unlike the on_http method, which can lead to runtime errors if scope["app"] is not a FastAPI instance. Furthermore, the _openapi_initialized flag is not set to True in on_lifespan after this call, causing duplicate OpenAPI schema processing on the first HTTP request.

fast_cache_middleware/middleware.py#L231-L245

FastCacheMiddleware/fast_cache_middleware/middleware.py

Lines 231 to 245 in 5ebd29f

	async def on_lifespan(self, scope: Scope, _: Receive, __: Send) -> bool | None:
	app_routes = get_app_routes(scope["app"])
	set_cache_age_in_openapi_schema(scope["app"])
	self._routes_info = self._extract_routes_info(app_routes)
	return None
	async def on_http(self, scope: Scope, receive: Receive, send: Send) -> bool | None:
	request = Request(scope, receive)
	if not self._openapi_initialized:
	fastapi_app = scope["app"]
	if isinstance(fastapi_app, FastAPI):
	set_cache_age_in_openapi_schema(fastapi_app)
	self._openapi_initialized = True

Fix in Cursor

---

Bug: Cache Age Setting and Schema Overwrite Issues

The set_cache_age_in_openapi_schema function has two issues: it incorrectly accesses route dependencies via route.dependant.dependencies instead of route.dependencies, preventing x-cache-age from being set for CacheConfig dependencies in the OpenAPI schema. Additionally, it unconditionally overwrites app.openapi_schema by always calling get_openapi(), which bypasses any custom OpenAPI schema generation logic.

fast_cache_middleware/_helpers.py#L6-L33

FastCacheMiddleware/fast_cache_middleware/_helpers.py

Lines 6 to 33 in 5ebd29f

	def set_cache_age_in_openapi_schema(app: FastAPI) -> None:
	openapi_schema = get_openapi(
	title=app.title,
	version=app.version,
	description=app.description,
	routes=app.routes,
	)
	for route in app.routes:
	if isinstance(route, routing.APIRoute):
	path = route.path
	methods = route.methods
	for dependency in route.dependant.dependencies:
	dep = dependency.call
	if isinstance(dep, CacheConfig):
	max_age = dep.max_age
	for method in methods:
	method = method.lower()
	try:
	operation = openapi_schema["paths"][path][method]
	operation.setdefault("x-cache-age", max_age)
	except KeyError:
	continue
	app.openapi_schema = openapi_schema

Fix in Cursor

---

Was this report helpful? Give feedback by reacting with 👍 or 👎

[Nottezz]

bugbot run

[Nottezz]

Создал conftest, надеюсь не против. А то фикстуры разрастаются, становится тяжко. И свои тесты по схеме запихнул в отдельный модуль.

[Nottezz]

Мне никак не победить mypy с его ошибкой Mypy: Cannot assign to a method [method-assign].

[Nottezz]

Добавил тест, который проверяет схему на "добавленные пользователем поля", работаю с функцией из utils, всё перезаписывалось. Переделал что б мы работали чисто со схемой приложения и всё нормально, перезаписей нету.

[chud0]

а в app.openapi_schema когда метод выполняется ничего нет? может схема уже готова

[chud0]

бездушная машина тоже говорит ... by always calling get_openapi(), which bypasses any custom OpenAPI schema generation logic