crud.py

# Description: This file contains the CRUD operations for talking to the database.

from typing import List, Optional, Union

from lnbits.db import Database
from lnbits.helpers import urlsafe_short_hash

from .models import CreateMyExtensionData, MyExtension

db = Database("ext_myextension")


async def create_myextension(data: CreateMyExtensionData) -> MyExtension:
    data.id = urlsafe_short_hash()
    await db.insert("myextension.maintable", data)
    return MyExtension(**data.dict())


async def get_myextension(myextension_id: str) -> Optional[MyExtension]:
    return await db.fetchone(
        "SELECT * FROM myextension.maintable WHERE id = :id",
        {"id": myextension_id},
        MyExtension,
    )


async def get_myextensions(wallet_ids: Union[str, List[str]]) -> List[MyExtension]:
    if isinstance(wallet_ids, str):
        wallet_ids = [wallet_ids]
    q = ",".join([f"'{w}'" for w in wallet_ids])
    return await db.fetchall(
        f"SELECT * FROM myextension.maintable WHERE wallet IN ({q}) ORDER BY id",
        model=MyExtension,
    )


async def update_myextension(data: CreateMyExtensionData) -> MyExtension:
    await db.update("myextension.maintable", data)
    return MyExtension(**data.dict())


async def delete_myextension(myextension_id: str) -> None:
    await db.execute(
        "DELETE FROM myextension.maintable WHERE id = :id", {"id": myextension_id}
    )