From 4ae6a24702e3b8bf79b0cdf0b6c070ec9618c4b9 Mon Sep 17 00:00:00 2001 From: drunkendog Date: Mon, 25 Sep 2023 20:39:51 +0100 Subject: [PATCH] Add `action` as a parameter --- api.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/api.py b/api.py index 8284c5a..25efb4f 100644 --- a/api.py +++ b/api.py @@ -32,15 +32,15 @@ async def root(): @app.get("/set_badge/") -async def set_badge(repo: str, new_badge: str, branch: str=None, api_key: str = Security(get_api_key)): +async def set_badge(repo: str, new_badge: str, branch: str=None, action: str=None, api_key: str = Security(get_api_key)): if api_key.rsplit("-", 1)[0] != repo: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="No permissions for repo", ) - if branch is not None: - badge_dict[repo, branch] = new_badge + if branch is not None or action is not None: + badge_dict[repo, branch, action] = new_badge else: badge_dict[repo] = new_badge with open("badges.yaml", "w") as f: @@ -48,10 +48,10 @@ async def set_badge(repo: str, new_badge: str, branch: str=None, api_key: str = @app.get("/get_badge/") -async def get_badge(repo: str, branch: str=None): +async def get_badge(repo: str, branch: str=None, action: str=None): if repo 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] if branch 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])