change statuscode condition

This commit is contained in:
Данил 2024-09-30 16:55:16 +03:00
parent aaf6afa449
commit c3918c3d35

View file

@ -4,6 +4,7 @@ from fastapi.templating import Jinja2Templates
import requests import requests
from datetime import datetime from datetime import datetime
from http import HTTPStatus
router = APIRouter() router = APIRouter()
@ -15,7 +16,9 @@ async def today_wallpaper(request: Request):
date = datetime.today().strftime('%Y-%m-%d') date = datetime.today().strftime('%Y-%m-%d')
res = requests.get(f'https://api.starlio.space/last') res = requests.get(f'https://api.starlio.space/last')
if res.status_code != 200: if not HTTPStatus(res.status_code).is_redirection or \
not HTTPStatus(res.status_code).is_success:
return FileResponse('./src/web/html/error/404.html') return FileResponse('./src/web/html/error/404.html')
return RedirectResponse(f'/wallpaper/{date}') return RedirectResponse(f'/wallpaper/{date}')
@ -25,7 +28,8 @@ async def today_wallpaper(request: Request):
async def wallpaper(request: Request, day): async def wallpaper(request: Request, day):
res = requests.get(f'https://api.starlio.space/wallpaper/{day}') res = requests.get(f'https://api.starlio.space/wallpaper/{day}')
if res.status_code != 200: if not HTTPStatus(res.status_code).is_redirection or \
not HTTPStatus(res.status_code).is_success:
return FileResponse('./src/web/html/error/404.html') return FileResponse('./src/web/html/error/404.html')
return template.TemplateResponse( return template.TemplateResponse(