drunkendog
/
gitea-actions-badges
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add branch option to api

main
drunkendog 2023-09-16 16:38:48 +01:00
parent a1b7d6320b
commit 9ad46bc733
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api.py
View File

@ -32,23 +32,26 @@ async def root():
@app.get("/set_badge/") @app.get("/set_badge/")
async def set_badge(repo: str, new_badge: str, api_key: str = Security(get_api_key)): async def set_badge(repo: str, new_badge: str, branch: str=None, api_key: str = Security(get_api_key)):
if api_key.rsplit("-", 1)[0] != repo: if api_key.rsplit("-", 1)[0] != repo:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_401_UNAUTHORIZED,
detail="No permissions for repo", detail="No permissions for repo",
) )
if branch is not None:
badge_dict[repo, branch] = new_badge
else:
badge_dict[repo] = new_badge badge_dict[repo] = new_badge
with open("badges.yaml", "w") as f: with open("badges.yaml", "w") as f:
yaml.safe_dump(badge_dict, f) yaml.safe_dump(badge_dict, f)
@app.get("/get_badge/") @app.get("/get_badge/")
async def get_badge(repo: str): async def get_badge(repo: str, branch: str=None):
if repo not in badge_dict.keys(): if 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="Repo not found",
) )
return RedirectResponse("https://img.shields.io/badge/" + badge_dict[repo]) return RedirectResponse("https://img.shields.io/badge/" + badge_dict[repo, branch] if branch is not None else badge_dict[repo])