83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
from http import HTTPStatus
|
|
from typing import List
|
|
|
|
from fastapi import Depends
|
|
from loguru import logger
|
|
from starlette.exceptions import HTTPException
|
|
|
|
from lnbits.core.crud import get_wallet, get_wallet_for_key
|
|
from lnbits.decorators import WalletTypeInfo, require_admin_key
|
|
|
|
from . import splitpayments_ext
|
|
from .crud import get_targets, set_targets
|
|
from .models import Target, TargetPutList
|
|
|
|
|
|
@splitpayments_ext.get("/api/v1/targets")
|
|
async def api_targets_get(
|
|
wallet: WalletTypeInfo = Depends(require_admin_key),
|
|
) -> List[Target]:
|
|
targets = await get_targets(wallet.wallet.id)
|
|
return targets or []
|
|
|
|
|
|
@splitpayments_ext.put("/api/v1/targets", status_code=HTTPStatus.OK)
|
|
async def api_targets_set(
|
|
target_put: TargetPutList,
|
|
source_wallet: WalletTypeInfo = Depends(require_admin_key),
|
|
) -> None:
|
|
try:
|
|
targets: List[Target] = []
|
|
for entry in target_put.targets:
|
|
|
|
if entry.wallet.find("@") < 0 and entry.wallet.find("LNURL") < 0:
|
|
wallet = await get_wallet(entry.wallet)
|
|
if not wallet:
|
|
wallet = await get_wallet_for_key(entry.wallet, "invoice")
|
|
if not wallet:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.BAD_REQUEST,
|
|
detail=f"Invalid wallet '{entry.wallet}'.",
|
|
)
|
|
|
|
if wallet.id == source_wallet.wallet.id:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.BAD_REQUEST, detail="Can't split to itself."
|
|
)
|
|
|
|
if entry.percent <= 0:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.BAD_REQUEST,
|
|
detail=f"Invalid percent '{entry.percent}'.",
|
|
)
|
|
|
|
targets.append(
|
|
Target(
|
|
wallet=entry.wallet,
|
|
source=source_wallet.wallet.id,
|
|
percent=entry.percent,
|
|
alias=entry.alias,
|
|
)
|
|
)
|
|
|
|
percent_sum = sum([target.percent for target in targets])
|
|
if percent_sum > 100:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.BAD_REQUEST, detail="Splitting over 100%"
|
|
)
|
|
|
|
await set_targets(source_wallet.wallet.id, targets)
|
|
|
|
except Exception as ex:
|
|
logger.warning(ex)
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
detail="Cannot set targets.",
|
|
)
|
|
|
|
|
|
@splitpayments_ext.delete("/api/v1/targets", status_code=HTTPStatus.OK)
|
|
async def api_targets_delete(
|
|
source_wallet: WalletTypeInfo = Depends(require_admin_key),
|
|
) -> None:
|
|
await set_targets(source_wallet.wallet.id, [])
|