From f97326bed03fefdb98b37072323eae2ae05d9d25 Mon Sep 17 00:00:00 2001 From: ben Date: Thu, 12 Jan 2023 22:42:49 +0000 Subject: [PATCH] Adde models --- lnbits/core/crud.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lnbits/core/crud.py b/lnbits/core/crud.py index 3979e791..45f192b9 100644 --- a/lnbits/core/crud.py +++ b/lnbits/core/crud.py @@ -9,7 +9,7 @@ from lnbits.db import COCKROACH, POSTGRES, Connection from lnbits.settings import AdminSettings, EditableSettings, SuperSettings, settings from . import db -from .models import BalanceCheck, Payment, TinyURL, User, Wallet +from .models import BalanceCheck, Payment, TinyURL, User, Wallet, TinyURL # accounts # -------- @@ -637,7 +637,7 @@ async def create_tinyurl(domain: str): 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( """ SELECT * @@ -649,7 +649,7 @@ async def get_tinyurl(tinyurl_id: str) -> Optional[BalanceCheck]: 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( """ SELECT *