lnbits-legend/lnbits/extensions/splitpayments/crud.py

from typing import List

from . import db
from .models import Target
from loguru import logger


async def get_targets(source_wallet: str) -> List[Target]:
    rows = await db.fetchall(
        "SELECT * FROM splitpayments.targets WHERE source = ?", (source_wallet,)
    )
    return [Target(**dict(row)) for row in rows]


async def set_targets(source_wallet: str, targets: List[Target]):
    logger.debug(targets)
    async with db.connect() as conn:
        await conn.execute(
            "DELETE FROM splitpayments.targets WHERE source = ?", (source_wallet,)
        )
        for target in targets:
            await conn.execute(
                """
                INSERT INTO splitpayments.targets
                  (source, wallet, percent, tag, alias)
                VALUES (?, ?, ?, ?, ?)
            """,
                (source_wallet, target.wallet, target.percent, target.tag, target.alias),
            )