From cd6f4c909f9693a11ac7f34f8f83a7c8544468b9 Mon Sep 17 00:00:00 2001
From: ben <ben@arc.wales>
Date: Fri, 23 Dec 2022 22:38:19 +0000
Subject: [PATCH] swaps pos id for number

---
 lnbits/extensions/lnurldevice/crud.py   | 6 ++++--
 lnbits/extensions/lnurldevice/models.py | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/lnbits/extensions/lnurldevice/crud.py b/lnbits/extensions/lnurldevice/crud.py
index d761edf42..708827b08 100644
--- a/lnbits/extensions/lnurldevice/crud.py
+++ b/lnbits/extensions/lnurldevice/crud.py
@@ -11,8 +11,10 @@ from .models import createLnurldevice, lnurldevicepayment, lnurldevices
 async def create_lnurldevice(
     data: createLnurldevice,
 ) -> lnurldevices:
-    if data.device == "pos":
+    if data.device == "pos" or data.device == "atm":
         lnurldevice_id = str(await get_lnurldeviceposcount())
+    else:
+        lnurldevice_id = urlsafe_short_hash()
     lnurldevice_key = urlsafe_short_hash()
     await db.execute(
         """
@@ -82,7 +84,7 @@ async def update_lnurldevice(lnurldevice_id: str, **kwargs) -> Optional[lnurldev
 
 async def get_lnurldeviceposcount() -> int:
     row = await db.fetchall(
-        "SELECT * FROM lnurldevice.lnurldevices WHERE device"
+        "SELECT * FROM lnurldevice.lnurldevices WHERE device = ? OR device = ?", ("pos", "atm",)
     )
     return len(row) + 1
 
diff --git a/lnbits/extensions/lnurldevice/models.py b/lnbits/extensions/lnurldevice/models.py
index c27470b73..66b215f27 100644
--- a/lnbits/extensions/lnurldevice/models.py
+++ b/lnbits/extensions/lnurldevice/models.py
@@ -17,8 +17,8 @@ class createLnurldevice(BaseModel):
     wallet: str
     currency: str
     device: str
-    profit: float
-    amount: int
+    profit: float = 0
+    amount: Optional[int] = 0
     pin: int = 0
     profit1: float = 0
     amount1: int = 0