mirror of
https://github.com/Starlio-app/Starlio-web.git
synced 2024-11-05 17:03:58 +03:00
change statuscode condition
This commit is contained in:
parent
aaf6afa449
commit
c3918c3d35
1 changed files with 6 additions and 2 deletions
|
@ -4,6 +4,7 @@ from fastapi.templating import Jinja2Templates
|
|||
|
||||
import requests
|
||||
from datetime import datetime
|
||||
from http import HTTPStatus
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
@ -15,7 +16,9 @@ async def today_wallpaper(request: Request):
|
|||
date = datetime.today().strftime('%Y-%m-%d')
|
||||
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 RedirectResponse(f'/wallpaper/{date}')
|
||||
|
@ -25,7 +28,8 @@ async def today_wallpaper(request: Request):
|
|||
async def wallpaper(request: Request, 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 template.TemplateResponse(
|
||||
|
|
Loading…
Reference in a new issue