Fix bug with if statements

main
drunkendog 2023-09-25 20:57:21 +01:00
parent e25c048717
commit ca5dc87e82
1 changed files with 2 additions and 7 deletions

9
api.py
View File

@ -49,14 +49,9 @@ async def set_badge(repo: str, new_badge: str, branch: str=None, action: str=Non
@app.get("/get_badge/") @app.get("/get_badge/")
async def get_badge(repo: str, branch: str=None, action: str=None): async def get_badge(repo: str, branch: str=None, action: str=None):
if branch is None and action is None and repo not in badge_dict.keys(): if (repo, branch, action) not in badge_dict.keys() or repo not in badge_dict.keys():
raise HTTPException( raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, status_code=status.HTTP_404_NOT_FOUND,
detail="Repo not found", detail="Badge not found",
)
elif (repo, branch, action) not in badge_dict.keys():
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail="Repo not found",
) )
return RedirectResponse("https://img.shields.io/badge/" + (badge_dict[repo, branch, action] if branch is not None or action is not None else badge_dict[repo])) return RedirectResponse("https://img.shields.io/badge/" + (badge_dict[repo, branch, action] if branch is not None or action is not None else badge_dict[repo]))