From 7c8e3de261d49f5600e4b900249528c3dcdafc10 Mon Sep 17 00:00:00 2001
From: Tiago vasconcelos <talvasconcelos@gmail.com>
Date: Fri, 19 Aug 2022 13:24:18 +0100
Subject: [PATCH] update product stock tests

---
 lnbits/extensions/diagonalley/crud.py  | 19 ++++++++++++++++---
 lnbits/extensions/diagonalley/tasks.py | 15 +--------------
 lnbits/extensions/diagonalley/views.py | 10 +++++++++-
 3 files changed, 26 insertions(+), 18 deletions(-)

diff --git a/lnbits/extensions/diagonalley/crud.py b/lnbits/extensions/diagonalley/crud.py
index 0a825ca39..402fb1f5b 100644
--- a/lnbits/extensions/diagonalley/crud.py
+++ b/lnbits/extensions/diagonalley/crud.py
@@ -1,10 +1,7 @@
-import re
 from base64 import urlsafe_b64encode
 from typing import List, Optional, Union
 from uuid import uuid4
 
-import httpx
-
 # from lnbits.db import open_ext_db
 from lnbits.db import SQLITE
 from lnbits.helpers import urlsafe_short_hash
@@ -291,6 +288,22 @@ async def set_diagonalley_order_paid(payment_hash: str) -> Orders:
             (payment_hash,),
         )
 
+async def update_diagonalley_product_stock(products):
+    
+    q = "\n".join([f"""WHEN id='{p["product_id"]}' THEN {p["quantity"]}""" for p in products])
+    v = ",".join(["?"] * len(products))
+    
+    await db.execute(
+        f"""
+            UPDATE diagonalley.products
+            SET quantity=(CASE
+                        {q}
+                        END)
+            WHERE id IN ({v});
+        """,
+        (*[p["product_id"] for p in products],)
+    )
+
 async def get_diagonalley_orders(wallet_ids: Union[str, List[str]]) -> List[Orders]:
     if isinstance(wallet_ids, str):
         wallet_ids = [wallet_ids]
diff --git a/lnbits/extensions/diagonalley/tasks.py b/lnbits/extensions/diagonalley/tasks.py
index 6b64203ec..22ce08193 100644
--- a/lnbits/extensions/diagonalley/tasks.py
+++ b/lnbits/extensions/diagonalley/tasks.py
@@ -30,19 +30,6 @@ async def on_invoice_paid(payment: Payment) -> None:
     await set_diagonalley_order_paid(payment.payment_hash)
     
     # deduct items sold from stock
+    
     # TODO
 
-    
-    """
-    if "lnticket" != payment.extra.get("tag"):
-        # not a lnticket invoice
-        return
-
-    ticket = await get_ticket(payment.checking_id)
-    if not ticket:
-        print("this should never happen", payment)
-        return
-
-    await payment.set_pending(False)
-    await set_ticket_paid(payment.payment_hash)
-    """
diff --git a/lnbits/extensions/diagonalley/views.py b/lnbits/extensions/diagonalley/views.py
index af93d60c0..df65c5411 100644
--- a/lnbits/extensions/diagonalley/views.py
+++ b/lnbits/extensions/diagonalley/views.py
@@ -17,6 +17,7 @@ from .crud import (
     get_diagonalley_stall,
     get_diagonalley_zone,
     get_diagonalley_zones,
+    update_diagonalley_product_stock,
 )
 
 templates = Jinja2Templates(directory="templates")
@@ -31,6 +32,13 @@ async def index(request: Request, user: User = Depends(check_user_exists)):
 
 @diagonalley_ext.get("/{stall_id}", response_class=HTMLResponse)
 async def display(request: Request, stall_id):
+    # test_qty = 99    
+    # test = [
+    #     {"product_id": "55vpVjWAuQExHsJxT28MYe", "quantity": test_qty},
+    #     {"product_id": "f2eGNsEWgbLJbfAApd3Jw5", "quantity": test_qty},
+    #     {"product_id": "FVqZLZdemWCsiqe9gafvsC", "quantity": test_qty},
+    # ]    
+    # await update_diagonalley_product_stock(test)
     stall = await get_diagonalley_stall(stall_id)
     products = await get_diagonalley_products(stall_id)
     zones = []
@@ -43,7 +51,7 @@ async def display(request: Request, stall_id):
         raise HTTPException(
             status_code=HTTPStatus.NOT_FOUND, detail="Stall does not exist."
         )
-
+    
     stall = stall.dict()
     del stall["privatekey"]
     stall["zones"] = zones