Handle getting logged out the same way in all cases

This commit is contained in:
Tulir Asokan
2023-01-03 21:45:24 +02:00
parent f1f0b86696
commit 9fae4f14d2
6 changed files with 91 additions and 35 deletions
+3 -3
View File
@@ -24,7 +24,7 @@ from asyncpg import Record
from attr import dataclass
from mautrix.types import UserID
from mautrix.util.async_db import Database
from mautrix.util.async_db import Connection, Database
from ..types import TelegramID
@@ -169,8 +169,8 @@ class Backfill:
)
@classmethod
async def delete_all(cls, user_mxid: UserID) -> None:
await cls.db.execute("DELETE FROM backfill_queue WHERE user_mxid=$1", user_mxid)
async def delete_all(cls, user_mxid: UserID, conn: Connection | None = None) -> None:
await (conn or cls.db).execute("DELETE FROM backfill_queue WHERE user_mxid=$1", user_mxid)
@classmethod
async def delete_for_portal(cls, tgid: int, tg_receiver: int) -> None: