Adde models
This commit is contained in:
parent
6e33784061
commit
f97326bed0
1 changed files with 3 additions and 3 deletions
|
|
@ -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 *
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue