From dda1d9bb3a9d45a97d9487ebe3959c3b1c5289b4 Mon Sep 17 00:00:00 2001 From: ben Date: Tue, 24 Jan 2023 21:29:52 +0000 Subject: [PATCH] mypy --- lnbits/core/views/api.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/lnbits/core/views/api.py b/lnbits/core/views/api.py index 485592a7..f614ba02 100644 --- a/lnbits/core/views/api.py +++ b/lnbits/core/views/api.py @@ -738,8 +738,9 @@ async def api_get_tinyurl( ): try: tinyurl = await get_tinyurl(tinyurl_id) - if tinyurl.wallet == wallet.wallet.inkey: - return tinyurl + if tinyurl: + if tinyurl.wallet == wallet.wallet.inkey: + return tinyurl raise HTTPException( status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided." ) @@ -755,9 +756,10 @@ async def api_delete_tinyurl( ): try: tinyurl = await get_tinyurl(tinyurl_id) - if tinyurl.wallet == wallet.wallet.inkey: - await delete_tinyurl(tinyurl_id) - return {"deleted": True} + if tinyurl: + if tinyurl.wallet == wallet.wallet.inkey: + await delete_tinyurl(tinyurl_id) + return {"deleted": True} raise HTTPException( status_code=HTTPStatus.FORBIDDEN, detail="Wrong key provided." )