aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions 1

make format

Browse source
This commit is contained in:
Tiago vasconcelos 2022-09-12 17:01:41 +01:00
parent f429037d63
commit 418cc7bf86
5 changed files with 25 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

11
lnbits/extensions/diagonalley/crud.py
View file

@ -274,12 +274,14 @@ async def get_diagonalley_order(order_id: str) -> Optional[Orders]:
) )
return Orders(**row) if row else None return Orders(**row) if row else None
async def get_diagonalley_order_invoiceid(invoice_id: str) -> Optional[Orders]: async def get_diagonalley_order_invoiceid(invoice_id: str) -> Optional[Orders]:
row = await db.fetchone( row = await db.fetchone(
"SELECT * FROM diagonalley.orders WHERE invoiceid = ?", (invoice_id,) "SELECT * FROM diagonalley.orders WHERE invoiceid = ?", (invoice_id,)
) )
return Orders(**row) if row else None return Orders(**row) if row else None
async def set_diagonalley_order_paid(payment_hash: str) -> Orders: async def set_diagonalley_order_paid(payment_hash: str) -> Orders:
await db.execute( await db.execute(
""" """
@ -290,9 +292,12 @@ async def set_diagonalley_order_paid(payment_hash: str) -> Orders:
(payment_hash,), (payment_hash,),
) )
async def update_diagonalley_product_stock(products): async def update_diagonalley_product_stock(products):
q = "\n".join([f"""WHEN id='{p.product_id}' THEN quantity - {p.quantity}""" for p in products]) q = "\n".join(
[f"""WHEN id='{p.product_id}' THEN quantity - {p.quantity}""" for p in products]
)
v = ",".join(["?"] * len(products)) v = ",".join(["?"] * len(products))
await db.execute( await db.execute(
@ -303,9 +308,10 @@ async def update_diagonalley_product_stock(products):
END) END)
WHERE id IN ({v}); WHERE id IN ({v});
""", """,
(*[p.product_id for p in products],) (*[p.product_id for p in products],),
) )
async def get_diagonalley_orders(wallet_ids: Union[str, List[str]]) -> List[Orders]: async def get_diagonalley_orders(wallet_ids: Union[str, List[str]]) -> List[Orders]:
if isinstance(wallet_ids, str): if isinstance(wallet_ids, str):
wallet_ids = [wallet_ids] wallet_ids = [wallet_ids]
@ -344,6 +350,7 @@ async def get_diagonalley_market_stalls(market_id: str):
return [{**row} for row in rows] return [{**row} for row in rows]
async def create_diagonalley_market(data: CreateMarket): async def create_diagonalley_market(data: CreateMarket):
market_id = urlsafe_short_hash() market_id = urlsafe_short_hash()

1
lnbits/extensions/diagonalley/models.py
View file

@ -104,5 +104,6 @@ class Market(BaseModel):
usr: str usr: str
name: Optional[str] name: Optional[str]
class CreateMarketStalls(BaseModel): class CreateMarketStalls(BaseModel):
stallid: str stallid: str

2
lnbits/extensions/diagonalley/tasks.py
View file

@ -37,5 +37,3 @@ async def on_invoice_paid(payment: Payment) -> None:
# deduct items sold from stock # deduct items sold from stock
details = await get_diagonalley_order_details(order.id) details = await get_diagonalley_order_details(order.id)
await update_diagonalley_product_stock(details) await update_diagonalley_product_stock(details)

1
lnbits/extensions/diagonalley/views_api.py
View file

@ -290,6 +290,7 @@ async def api_diagonalley_check_payment(payment_hash: str):
return {"paid": False} return {"paid": False}
return status return status
@diagonalley_ext.delete("/api/v1/orders/{order_id}") @diagonalley_ext.delete("/api/v1/orders/{order_id}")
async def api_diagonalley_order_delete( async def api_diagonalley_order_delete(
order_id: str, wallet: WalletTypeInfo = Depends(get_key_type) order_id: str, wallet: WalletTypeInfo = Depends(get_key_type)