mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-23 14:40:47 +01:00
350 lines
10 KiB
Python
350 lines
10 KiB
Python
from quart import g, jsonify, request
|
|
from http import HTTPStatus
|
|
|
|
from lnbits.core.crud import get_user
|
|
from lnbits.decorators import api_check_wallet_key, api_validate_post_request
|
|
|
|
from lnbits.extensions.diagonalley import diagonalley_ext
|
|
from .crud import (
|
|
create_diagonalleys_product,
|
|
get_diagonalleys_product,
|
|
get_diagonalleys_products,
|
|
delete_diagonalleys_product,
|
|
create_diagonalleys_indexer,
|
|
update_diagonalleys_indexer,
|
|
get_diagonalleys_indexer,
|
|
get_diagonalleys_indexers,
|
|
delete_diagonalleys_indexer,
|
|
create_diagonalleys_order,
|
|
get_diagonalleys_order,
|
|
get_diagonalleys_orders,
|
|
update_diagonalleys_product,
|
|
)
|
|
from lnbits.core.services import create_invoice
|
|
from base64 import urlsafe_b64encode
|
|
from uuid import uuid4
|
|
from lnbits.db import open_ext_db
|
|
|
|
### Products
|
|
|
|
|
|
@diagonalley_ext.get("/api/v1/diagonalley/products")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_products():
|
|
wallet_ids = [g.wallet.id]
|
|
|
|
if "all_wallets" in request.args:
|
|
wallet_ids = get_user(g.wallet.user).wallet_ids
|
|
|
|
return (
|
|
jsonify(
|
|
[product._asdict() for product in get_diagonalleys_products(wallet_ids)]
|
|
),
|
|
HTTPStatus.OK,
|
|
)
|
|
|
|
class CreateData(BaseModel):
|
|
product: str
|
|
categories: str
|
|
description: str
|
|
image: str
|
|
price: int = Query(ge=0)
|
|
quantity: int = Query(ge=0)
|
|
|
|
@diagonalley_ext.post("/api/v1/diagonalley/products")
|
|
@diagonalley_ext.put("/api/v1/diagonalley/products{product_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_product_create(product_id=None):
|
|
|
|
if product_id:
|
|
product = get_diagonalleys_indexer(product_id)
|
|
|
|
if not product:
|
|
return (
|
|
jsonify({"message": "Withdraw product does not exist."}),
|
|
HTTPStatus.NOT_FOUND,
|
|
)
|
|
|
|
if product.wallet != g.wallet.id:
|
|
return (
|
|
jsonify({"message": "Not your withdraw product."}),
|
|
HTTPStatus.FORBIDDEN,
|
|
)
|
|
|
|
product = update_diagonalleys_product(product_id, **g.data)
|
|
else:
|
|
product = create_diagonalleys_product(wallet_id=g.wallet.id, **g.data)
|
|
|
|
return (
|
|
jsonify(product._asdict()),
|
|
HTTPStatus.OK if product_id else HTTPStatus.CREATED,
|
|
)
|
|
|
|
|
|
@diagonalley_ext.delete("/api/v1/diagonalley/products/{product_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_products_delete(product_id):
|
|
product = get_diagonalleys_product(product_id)
|
|
|
|
if not product:
|
|
return jsonify({"message": "Product does not exist."}), HTTPStatus.NOT_FOUND
|
|
|
|
if product.wallet != g.wallet.id:
|
|
return jsonify({"message": "Not your Diagon Alley."}), HTTPStatus.FORBIDDEN
|
|
|
|
delete_diagonalleys_product(product_id)
|
|
|
|
return "", HTTPStatus.NO_CONTENT
|
|
|
|
|
|
###Indexers
|
|
|
|
|
|
@diagonalley_ext.get("/api/v1/diagonalley/indexers")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_indexers():
|
|
wallet_ids = [g.wallet.id]
|
|
|
|
if "all_wallets" in request.args:
|
|
wallet_ids = get_user(g.wallet.user).wallet_ids
|
|
|
|
return (
|
|
jsonify(
|
|
[indexer._asdict() for indexer in get_diagonalleys_indexers(wallet_ids)]
|
|
),
|
|
HTTPStatus.OK,
|
|
)
|
|
|
|
class CreateData(BaseModel):
|
|
shopname: str
|
|
indexeraddress: str
|
|
shippingzone1: str
|
|
shippingzone2: str
|
|
email: str
|
|
zone1cost: int = Query(ge=0)
|
|
zone2cost: int = Query(ge=0)
|
|
|
|
@diagonalley_ext.post("/api/v1/diagonalley/indexers")
|
|
@diagonalley_ext.put("/api/v1/diagonalley/indexers{indexer_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_indexer_create(data: CreateData, indexer_id=None):
|
|
|
|
if indexer_id:
|
|
indexer = get_diagonalleys_indexer(indexer_id)
|
|
|
|
if not indexer:
|
|
return (
|
|
jsonify({"message": "Withdraw indexer does not exist."}),
|
|
HTTPStatus.NOT_FOUND,
|
|
)
|
|
|
|
if indexer.wallet != g.wallet.id:
|
|
return (
|
|
jsonify({"message": "Not your withdraw indexer."}),
|
|
HTTPStatus.FORBIDDEN,
|
|
)
|
|
|
|
indexer = update_diagonalleys_indexer(indexer_id, **data)
|
|
else:
|
|
indexer = create_diagonalleys_indexer(wallet_id=g.wallet.id, **data)
|
|
|
|
return (
|
|
indexer._asdict(),
|
|
HTTPStatus.OK if indexer_id else HTTPStatus.CREATED,
|
|
)
|
|
|
|
|
|
@diagonalley_ext.delete("/api/v1/diagonalley/indexers/{indexer_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_indexer_delete(indexer_id):
|
|
indexer = get_diagonalleys_indexer(indexer_id)
|
|
|
|
if not indexer:
|
|
return {"message": "Indexer does not exist."}, HTTPStatus.NOT_FOUND
|
|
|
|
if indexer.wallet != g.wallet.id:
|
|
return {"message": "Not your Indexer."}, HTTPStatus.FORBIDDEN
|
|
|
|
delete_diagonalleys_indexer(indexer_id)
|
|
|
|
return "", HTTPStatus.NO_CONTENT
|
|
|
|
|
|
###Orders
|
|
|
|
|
|
@diagonalley_ext.get("/api/v1/diagonalley/orders")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_orders():
|
|
wallet_ids = [g.wallet.id]
|
|
|
|
if "all_wallets" in request.args:
|
|
wallet_ids = get_user(g.wallet.user).wallet_ids
|
|
|
|
return (
|
|
[order._asdict() for order in get_diagonalleys_orders(wallet_ids)],
|
|
HTTPStatus.OK,
|
|
)
|
|
|
|
class CreateData(BaseModel):
|
|
id: str
|
|
address: str
|
|
email: str
|
|
quantity: int
|
|
shippingzone: int
|
|
|
|
@diagonalley_ext.post("/api/v1/diagonalley/orders")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
|
|
async def api_diagonalley_order_create(data: CreateData):
|
|
order = create_diagonalleys_order(wallet_id=g.wallet.id, **data)
|
|
return order._asdict(), HTTPStatus.CREATED
|
|
|
|
|
|
@diagonalley_ext.delete("/api/v1/diagonalley/orders/{order_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalley_order_delete(order_id):
|
|
order = get_diagonalleys_order(order_id)
|
|
|
|
if not order:
|
|
return {"message": "Indexer does not exist."}, HTTPStatus.NOT_FOUND
|
|
|
|
if order.wallet != g.wallet.id:
|
|
return {"message": "Not your Indexer."}, HTTPStatus.FORBIDDEN
|
|
|
|
delete_diagonalleys_indexer(order_id)
|
|
|
|
return "", HTTPStatus.NO_CONTENT
|
|
|
|
|
|
@diagonalley_ext.get("/api/v1/diagonalley/orders/paid/{order_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalleys_order_paid(order_id):
|
|
with open_ext_db("diagonalley") as db:
|
|
db.execute(
|
|
"UPDATE diagonalley.orders SET paid = ? WHERE id = ?",
|
|
(
|
|
True,
|
|
order_id,
|
|
),
|
|
)
|
|
return "", HTTPStatus.OK
|
|
|
|
|
|
@diagonalley_ext.get("/api/v1/diagonalley/orders/shipped/{order_id}")
|
|
@api_check_wallet_key(key_type="invoice")
|
|
async def api_diagonalleys_order_shipped(order_id):
|
|
with open_ext_db("diagonalley") as db:
|
|
db.execute(
|
|
"UPDATE diagonalley.orders SET shipped = ? WHERE id = ?",
|
|
(
|
|
True,
|
|
order_id,
|
|
),
|
|
)
|
|
order = db.fetchone(
|
|
"SELECT * FROM diagonalley.orders WHERE id = ?", (order_id,)
|
|
)
|
|
|
|
return (
|
|
jsonify(
|
|
[order._asdict() for order in get_diagonalleys_orders(order["wallet"])]
|
|
),
|
|
HTTPStatus.OK,
|
|
)
|
|
|
|
|
|
###List products based on indexer id
|
|
|
|
|
|
@diagonalley_ext.get(
|
|
"/api/v1/diagonalley/stall/products/{indexer_id}"
|
|
)
|
|
async def api_diagonalleys_stall_products(indexer_id):
|
|
with open_ext_db("diagonalley") as db:
|
|
rows = db.fetchone(
|
|
"SELECT * FROM diagonalley.indexers WHERE id = ?", (indexer_id,)
|
|
)
|
|
print(rows[1])
|
|
if not rows:
|
|
return {"message": "Indexer does not exist."}, HTTPStatus.NOT_FOUND
|
|
|
|
products = db.fetchone(
|
|
"SELECT * FROM diagonalley.products WHERE wallet = ?", (rows[1],)
|
|
)
|
|
if not products:
|
|
return {"message": "No products"}, HTTPStatus.NOT_FOUND
|
|
|
|
return (
|
|
[products._asdict() for products in get_diagonalleys_products(rows[1])],
|
|
HTTPStatus.OK,
|
|
)
|
|
|
|
|
|
###Check a product has been shipped
|
|
|
|
|
|
@diagonalley_ext.get(
|
|
"/api/v1/diagonalley/stall/checkshipped/{checking_id}"
|
|
)
|
|
async def api_diagonalleys_stall_checkshipped(checking_id):
|
|
with open_ext_db("diagonalley") as db:
|
|
rows = db.fetchone(
|
|
"SELECT * FROM diagonalley.orders WHERE invoiceid = ?", (checking_id,)
|
|
)
|
|
|
|
return {"shipped": rows["shipped"]}, HTTPStatus.OK
|
|
|
|
|
|
###Place order
|
|
|
|
|
|
@diagonalley_ext.post("/api/v1/diagonalley/stall/order/{indexer_id}")
|
|
@api_validate_post_request(
|
|
schema={
|
|
"id": {"type": "string", "empty": False, "required": True},
|
|
"email": {"type": "string", "empty": False, "required": True},
|
|
"address": {"type": "string", "empty": False, "required": True},
|
|
"quantity": {"type": "integer", "empty": False, "required": True},
|
|
"shippingzone": {"type": "integer", "empty": False, "required": True},
|
|
}
|
|
)
|
|
async def api_diagonalley_stall_order(indexer_id):
|
|
product = get_diagonalleys_product(g.data["id"])
|
|
shipping = get_diagonalleys_indexer(indexer_id)
|
|
|
|
if g.data["shippingzone"] == 1:
|
|
shippingcost = shipping.zone1cost
|
|
else:
|
|
shippingcost = shipping.zone2cost
|
|
|
|
checking_id, payment_request = create_invoice(
|
|
wallet_id=product.wallet,
|
|
amount=shippingcost + (g.data["quantity"] * product.price),
|
|
memo=g.data["id"],
|
|
)
|
|
selling_id = urlsafe_b64encode(uuid4().bytes_le).decode("utf-8")
|
|
with open_ext_db("diagonalley") as db:
|
|
db.execute(
|
|
"""
|
|
INSERT INTO diagonalley.orders (id, productid, wallet, product, quantity, shippingzone, address, email, invoiceid, paid, shipped)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
""",
|
|
(
|
|
selling_id,
|
|
g.data["id"],
|
|
product.wallet,
|
|
product.product,
|
|
g.data["quantity"],
|
|
g.data["shippingzone"],
|
|
g.data["address"],
|
|
g.data["email"],
|
|
checking_id,
|
|
False,
|
|
False,
|
|
),
|
|
)
|
|
return (
|
|
{"checking_id": checking_id, "payment_request": payment_request},
|
|
HTTPStatus.OK,
|
|
)
|