fix little issues after refactor :)

This commit is contained in:
dni ⚡ 2023-01-12 08:05:05 +01:00
parent 85afb39388
commit bb57831c53
2 changed files with 5 additions and 4 deletions

View file

@ -123,7 +123,7 @@ async def set_email_paid(payment_hash: str) -> bool:
email = await get_email_by_payment_hash(payment_hash)
if email and email.paid == False:
await db.execute(
f"UPDATE smtp.email SET paid = true WHERE payment_hash = {payment_hash}"
f"UPDATE smtp.email SET paid = true WHERE payment_hash = ?", (payment_hash,)
)
return True
return False
@ -131,13 +131,13 @@ async def set_email_paid(payment_hash: str) -> bool:
async def get_email_by_payment_hash(payment_hash: str) -> Optional[Email]:
row = await db.fetchone(
f"SELECT * FROM smtp.email WHERE payment_hash = {payment_hash}"
f"SELECT * FROM smtp.email WHERE payment_hash = ?", (payment_hash,)
)
return Email(**row) if row else None
async def get_email(id: str) -> Optional[Email]:
row = await db.fetchone(f"SELECT * FROM smtp.email WHERE id = {id}")
row = await db.fetchone(f"SELECT * FROM smtp.email WHERE id = ?", (id,))
return Email(**row) if row else None

View file

@ -13,6 +13,7 @@ from .crud import (
delete_email,
delete_emailaddress,
get_email,
get_email_by_payment_hash,
get_emailaddress,
get_emailaddresses,
get_emails,
@ -37,7 +38,7 @@ async def api_email(
@smtp_ext.get("/api/v1/email/{payment_hash}")
async def api_smtp_send_email(payment_hash):
email = await get_email(payment_hash)
email = await get_email_by_payment_hash(payment_hash)
if not email:
raise HTTPException(
status_code=HTTPStatus.BAD_REQUEST, detail="paymenthash is wrong"