|
1 | 1 | from datetime import datetime, timedelta, timezone |
2 | | -from typing import Optional, Union |
3 | 2 |
|
4 | 3 | from lnbits.db import Database |
5 | 4 | from lnbits.helpers import urlsafe_short_hash |
@@ -33,15 +32,15 @@ async def update_ticket(ticket: Ticket) -> Ticket: |
33 | 32 | return ticket |
34 | 33 |
|
35 | 34 |
|
36 | | -async def get_ticket(payment_hash: str) -> Optional[Ticket]: |
| 35 | +async def get_ticket(payment_hash: str) -> Ticket | None: |
37 | 36 | return await db.fetchone( |
38 | 37 | "SELECT * FROM events.ticket WHERE id = :id", |
39 | 38 | {"id": payment_hash}, |
40 | 39 | Ticket, |
41 | 40 | ) |
42 | 41 |
|
43 | 42 |
|
44 | | -async def get_tickets(wallet_ids: Union[str, list[str]]) -> list[Ticket]: |
| 43 | +async def get_tickets(wallet_ids: str | list[str]) -> list[Ticket]: |
45 | 44 | if isinstance(wallet_ids, str): |
46 | 45 | wallet_ids = [wallet_ids] |
47 | 46 | q = ",".join([f"'{wallet_id}'" for wallet_id in wallet_ids]) |
@@ -84,15 +83,15 @@ async def update_event(event: Event) -> Event: |
84 | 83 | return event |
85 | 84 |
|
86 | 85 |
|
87 | | -async def get_event(event_id: str) -> Optional[Event]: |
| 86 | +async def get_event(event_id: str) -> Event | None: |
88 | 87 | return await db.fetchone( |
89 | 88 | "SELECT * FROM events.events WHERE id = :id", |
90 | 89 | {"id": event_id}, |
91 | 90 | Event, |
92 | 91 | ) |
93 | 92 |
|
94 | 93 |
|
95 | | -async def get_events(wallet_ids: Union[str, list[str]]) -> list[Event]: |
| 94 | +async def get_events(wallet_ids: str | list[str]) -> list[Event]: |
96 | 95 | if isinstance(wallet_ids, str): |
97 | 96 | wallet_ids = [wallet_ids] |
98 | 97 | q = ",".join([f"'{wallet_id}'" for wallet_id in wallet_ids]) |
|
0 commit comments