From ca8f649fd63e71b8ce655bb50214dfb1a4a06f00 Mon Sep 17 00:00:00 2001
From: ben <ben@arc.wales>
Date: Tue, 17 May 2022 22:02:54 +0100
Subject: [PATCH] Added check for error message

---
 lnbits/extensions/withdraw/models.py    | 2 +-
 lnbits/extensions/withdraw/views_api.py | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/lnbits/extensions/withdraw/models.py b/lnbits/extensions/withdraw/models.py
index 72b75fb8..a03c7db8 100644
--- a/lnbits/extensions/withdraw/models.py
+++ b/lnbits/extensions/withdraw/models.py
@@ -12,7 +12,7 @@ class CreateWithdrawData(BaseModel):
     title: str = Query(...)
     min_withdrawable: int = Query(..., ge=1)
     max_withdrawable: int = Query(..., ge=1)
-    uses: int = Query(..., ge=1, le=250)
+    uses: int = Query(..., ge=1)
     wait_time: int = Query(..., ge=1)
     is_unique: bool
 
diff --git a/lnbits/extensions/withdraw/views_api.py b/lnbits/extensions/withdraw/views_api.py
index 8dd9e340..c48a26b7 100644
--- a/lnbits/extensions/withdraw/views_api.py
+++ b/lnbits/extensions/withdraw/views_api.py
@@ -71,6 +71,11 @@ async def api_link_create_or_update(
     link_id: str = None,
     wallet: WalletTypeInfo = Depends(require_admin_key),
 ):
+    if data.uses > 250:
+        raise HTTPException(
+            detail="250 uses max.", status_code=HTTPStatus.BAD_REQUEST
+        )
+   
     if data.min_withdrawable < 1:
         raise HTTPException(
             detail="Min must be more than 1.", status_code=HTTPStatus.BAD_REQUEST