This commit is contained in:
ben 2023-01-24 21:29:52 +00:00
parent c0c4c86808
commit dda1d9bb3a

View file

@ -738,8 +738,9 @@ async def api_get_tinyurl(
): ):
try: try:
tinyurl = await get_tinyurl(tinyurl_id) tinyurl = await get_tinyurl(tinyurl_id)
if tinyurl.wallet == wallet.wallet.inkey: if tinyurl:
return tinyurl if tinyurl.wallet == wallet.wallet.inkey:
return tinyurl
raise HTTPException( raise HTTPException(
status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided." status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided."
) )
@ -755,9 +756,10 @@ async def api_delete_tinyurl(
): ):
try: try:
tinyurl = await get_tinyurl(tinyurl_id) tinyurl = await get_tinyurl(tinyurl_id)
if tinyurl.wallet == wallet.wallet.inkey: if tinyurl:
await delete_tinyurl(tinyurl_id) if tinyurl.wallet == wallet.wallet.inkey:
return {"deleted": True} await delete_tinyurl(tinyurl_id)
return {"deleted": True}
raise HTTPException( raise HTTPException(
status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided." status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided."
) )