mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-26 23:51:55 +01:00
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from typing import List, Optional
|
|
from pydantic import BaseModel
|
|
from quart import g, jsonify
|
|
from http import HTTPStatus
|
|
|
|
from lnbits.decorators import api_check_wallet_key, api_validate_post_request
|
|
from lnbits.core.crud import get_wallet, get_wallet_for_key
|
|
|
|
from . import splitpayments_ext
|
|
from .crud import get_targets, set_targets
|
|
from .models import Target
|
|
|
|
|
|
@splitpayments_ext.get("/api/v1/targets")
|
|
@api_check_wallet_key("admin")
|
|
async def api_targets_get():
|
|
targets = await get_targets(g.wallet.id)
|
|
return [target._asdict() for target in targets] or []
|
|
|
|
class SchemaData(BaseModel):
|
|
wallet: str
|
|
alias: str
|
|
percent: int
|
|
|
|
@splitpayments_ext.put("/api/v1/targets")
|
|
@api_check_wallet_key("admin")
|
|
async def api_targets_set(targets: Optional[List[SchemaData]]
|
|
= None):
|
|
targets = []
|
|
|
|
for entry in targets:
|
|
wallet = await get_wallet(entry["wallet"])
|
|
if not wallet:
|
|
wallet = await get_wallet_for_key(entry["wallet"], "invoice")
|
|
if not wallet:
|
|
return (
|
|
{"message": f"Invalid wallet '{entry['wallet']}'."},
|
|
HTTPStatus.BAD_REQUEST,
|
|
)
|
|
|
|
if wallet.id == g.wallet.id:
|
|
return (
|
|
{"message": "Can't split to itself."},
|
|
HTTPStatus.BAD_REQUEST,
|
|
)
|
|
|
|
if entry["percent"] < 0:
|
|
return (
|
|
{"message": f"Invalid percent '{entry['percent']}'."},
|
|
HTTPStatus.BAD_REQUEST,
|
|
)
|
|
|
|
targets.append(
|
|
Target(wallet.id, g.wallet.id, entry["percent"], entry["alias"] or "")
|
|
)
|
|
|
|
percent_sum = sum([target.percent for target in targets])
|
|
if percent_sum > 100:
|
|
return {"message": "Splitting over 100%."}, HTTPStatus.BAD_REQUEST
|
|
|
|
await set_targets(g.wallet.id, targets)
|
|
return "", HTTPStatus.OK
|