Adde models

This commit is contained in:
ben
2023-01-12 22:42:49 +00:00
parent 6e33784061
commit f97326bed0

View File

@@ -9,7 +9,7 @@ from lnbits.db import COCKROACH, POSTGRES, Connection
from lnbits.settings import AdminSettings, EditableSettings, SuperSettings, settings from lnbits.settings import AdminSettings, EditableSettings, SuperSettings, settings
from . import db from . import db
from .models import BalanceCheck, Payment, TinyURL, User, Wallet from .models import BalanceCheck, Payment, TinyURL, User, Wallet, TinyURL
# accounts # accounts
# -------- # --------
@@ -637,7 +637,7 @@ async def create_tinyurl(domain: str):
return await get_tinyurl(tinyurl_id) return await get_tinyurl(tinyurl_id)
async def get_tinyurl(tinyurl_id: str) -> Optional[BalanceCheck]: async def get_tinyurl(tinyurl_id: str) -> Optional[TinyURL]:
row = await db.fetchone( row = await db.fetchone(
""" """
SELECT * SELECT *
@@ -649,7 +649,7 @@ async def get_tinyurl(tinyurl_id: str) -> Optional[BalanceCheck]:
return TinyURL.from_row(row) if row else None return TinyURL.from_row(row) if row else None
async def get_tinyurl_by_url(url: str) -> Optional[BalanceCheck]: async def get_tinyurl_by_url(url: str) -> Optional[TinyURL]:
row = await db.fetchone( row = await db.fetchone(
""" """
SELECT * SELECT *