From 084ecb1c08d01e8380846480034b56d2c67c1dd0 Mon Sep 17 00:00:00 2001 From: Tiago vasconcelos Date: Mon, 8 Nov 2021 15:55:23 +0000 Subject: [PATCH] fix missing number on model --- lnbits/extensions/withdraw/crud.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/lnbits/extensions/withdraw/crud.py b/lnbits/extensions/withdraw/crud.py index 911a8e635..18a057f34 100644 --- a/lnbits/extensions/withdraw/crud.py +++ b/lnbits/extensions/withdraw/crud.py @@ -56,7 +56,10 @@ async def get_withdraw_link(link_id: str, num=0) -> Optional[WithdrawLink]: if not row: return None - return WithdrawLink(**row) if row else None + link = dict(**row) + link["number"] = num + + return WithdrawLink.parse_obj(link) async def get_withdraw_link_by_hash(unique_hash: str, num=0) -> Optional[WithdrawLink]: @@ -65,7 +68,11 @@ async def get_withdraw_link_by_hash(unique_hash: str, num=0) -> Optional[Withdra ) if not row: return None - return WithdrawLink(**row) if row else None + + link = dict(**row) + link["number"] = num + + return WithdrawLink.parse_obj(link) async def get_withdraw_links(wallet_ids: Union[str, List[str]]) -> List[WithdrawLink]: