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

from typing import List, Optional, Union

from lnbits.db import open_ext_db
from lnbits.helpers import urlsafe_short_hash

from .models import Tickets, Events


#######TICKETS########


def create_ticket(payment_hash: str, wallet: str, event: str, name: str, email: str) -> Tickets:
    with open_ext_db("events") as db:
        db.execute(
            """
            INSERT INTO ticket (id, wallet, event, name, email, registered, paid)
            VALUES (?, ?, ?, ?, ?, ?, ?)
            """,
            (payment_hash, wallet, event, name, email, False, False),
        )

    return get_ticket(payment_hash)


def update_ticket(paid: bool, payment_hash: str) -> Tickets:
    with open_ext_db("events") as db:
        row = db.fetchone("SELECT * FROM ticket WHERE id = ?", (payment_hash,))
        if row[6] == True:
            return get_ticket(payment_hash)
        db.execute(
            """
            UPDATE ticket
            SET paid = ?
            WHERE id = ?
            """,
            (paid, payment_hash),
        )

        eventdata = get_event(row[2])
        sold = eventdata.sold + 1
        amount_tickets = eventdata.amount_tickets - 1
        db.execute(
            """
            UPDATE events
            SET sold = ?, amount_tickets = ?
            WHERE id = ?
            """,
            (sold, amount_tickets, row[2]),
        )
    return get_ticket(payment_hash)


def get_ticket(payment_hash: str) -> Optional[Tickets]:
    with open_ext_db("events") as db:
        row = db.fetchone("SELECT * FROM ticket WHERE id = ?", (payment_hash,))

    return Tickets(**row) if row else None


def get_tickets(wallet_ids: Union[str, List[str]]) -> List[Tickets]:
    if isinstance(wallet_ids, str):
        wallet_ids = [wallet_ids]

    with open_ext_db("events") as db:
        q = ",".join(["?"] * len(wallet_ids))
        rows = db.fetchall(f"SELECT * FROM ticket WHERE wallet IN ({q})", (*wallet_ids,))
    print("scrum")
    return [Tickets(**row) for row in rows]


def delete_ticket(payment_hash: str) -> None:
    with open_ext_db("events") as db:
        db.execute("DELETE FROM ticket WHERE id = ?", (payment_hash,))


########EVENTS#########


def create_event(
    *,
    wallet: str,
    name: str,
    info: str,
    closing_date: str,
    event_start_date: str,
    event_end_date: str,
    amount_tickets: int,
    price_per_ticket: int,
) -> Events:
    with open_ext_db("events") as db:
        event_id = urlsafe_short_hash()
        db.execute(
            """
            INSERT INTO events (id, wallet, name, info, closing_date, event_start_date, event_end_date, amount_tickets, price_per_ticket, sold)
            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
            """,
            (
                event_id,
                wallet,
                name,
                info,
                closing_date,
                event_start_date,
                event_end_date,
                amount_tickets,
                price_per_ticket,
                0,
            ),
        )
        print(event_id)

    return get_event(event_id)


def update_event(event_id: str, **kwargs) -> Events:
    q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()])
    with open_ext_db("events") as db:
        db.execute(f"UPDATE events SET {q} WHERE id = ?", (*kwargs.values(), event_id))

        row = db.fetchone("SELECT * FROM events WHERE id = ?", (event_id,))

    return Events(**row) if row else None


def get_event(event_id: str) -> Optional[Events]:
    with open_ext_db("events") as db:
        row = db.fetchone("SELECT * FROM events WHERE id = ?", (event_id,))

    return Events(**row) if row else None


def get_events(wallet_ids: Union[str, List[str]]) -> List[Events]:
    if isinstance(wallet_ids, str):
        wallet_ids = [wallet_ids]

    with open_ext_db("events") as db:
        q = ",".join(["?"] * len(wallet_ids))
        rows = db.fetchall(f"SELECT * FROM events WHERE wallet IN ({q})", (*wallet_ids,))

    return [Events(**row) for row in rows]


def delete_event(event_id: str) -> None:
    with open_ext_db("events") as db:
        db.execute("DELETE FROM events WHERE id = ?", (event_id,))


########EVENTTICKETS#########


def get_event_tickets(event_id: str, wallet_id: str) -> Tickets:

    with open_ext_db("events") as db:
        rows = db.fetchall("SELECT * FROM ticket WHERE wallet = ? AND event = ?", (wallet_id, event_id))
        print(rows)

    return [Tickets(**row) for row in rows]


def reg_ticket(ticket_id: str) -> Tickets:
    with open_ext_db("events") as db:
        db.execute("UPDATE ticket SET registered = ? WHERE id = ?", (True, ticket_id))
        ticket = db.fetchone("SELECT * FROM ticket WHERE id = ?", (ticket_id,))
        print(ticket[1])
        rows = db.fetchall("SELECT * FROM ticket WHERE event = ?", (ticket[1],))

    return [Tickets(**row) for row in rows]