mirror of
https://github.com/resf/distro-tools.git
synced 2024-12-26 20:50:54 +00:00
77 lines
2 KiB
Python
77 lines
2 KiB
Python
from typing import TypeVar, Generic, Optional
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from fastapi.exceptions import HTTPException
|
|
from fastapi_pagination import Params
|
|
from fastapi_pagination.links import Page
|
|
from fastapi_pagination.ext.tortoise import paginate
|
|
|
|
from apollo.db import Advisory, RedHatIndexState
|
|
from apollo.db.serialize import Advisory_Pydantic
|
|
|
|
router = APIRouter(tags=["advisories"])
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class Pagination(Page[T], Generic[T]):
|
|
last_updated_at: Optional[str]
|
|
|
|
class Config:
|
|
allow_population_by_field_name = True
|
|
fields = {"items": {"alias": "advisories"}}
|
|
|
|
|
|
@router.get(
|
|
"/",
|
|
response_model=Pagination[Advisory_Pydantic],
|
|
)
|
|
async def list_advisories(
|
|
params: Params = Depends(),
|
|
product: Optional[str] = None,
|
|
before_raw: Optional[str] = None,
|
|
after_raw: Optional[str] = None,
|
|
cve: Optional[str] = None,
|
|
synopsis: Optional[str] = None,
|
|
keyword: Optional[str] = None,
|
|
severity: Optional[str] = None,
|
|
kind: Optional[str] = None,
|
|
):
|
|
advisories = await paginate(
|
|
Advisory.all().prefetch_related(
|
|
"red_hat_advisory",
|
|
"packages",
|
|
"cves",
|
|
"fixes",
|
|
"affected_products",
|
|
).order_by("-published_at"),
|
|
)
|
|
|
|
state = await RedHatIndexState.first()
|
|
advisories.last_updated_at = state.last_indexed_at.isoformat("T").replace(
|
|
"+00:00",
|
|
"",
|
|
) + "Z"
|
|
|
|
return advisories
|
|
|
|
|
|
@router.get(
|
|
"/{advisory_name}",
|
|
response_model=Advisory_Pydantic,
|
|
)
|
|
async def get_advisory(advisory_name: str):
|
|
advisory = await Advisory.filter(name=advisory_name).prefetch_related(
|
|
"packages",
|
|
"cves",
|
|
"fixes",
|
|
"affected_products",
|
|
"packages",
|
|
"packages__supported_product",
|
|
"packages__supported_products_rh_mirror",
|
|
).get_or_none()
|
|
|
|
if advisory is None:
|
|
raise HTTPException(404)
|
|
|
|
return await Advisory_Pydantic.from_tortoise_orm(advisory)
|