Don't break on old mautrix-python versions

This commit is contained in:
Tulir Asokan 2022-04-30 21:24:02 +03:00
parent fa34d80c4f
commit e87f332e0e

View File

@ -23,7 +23,13 @@ from attr import dataclass
import attr
from mautrix.types import RoomID, UserID
from mautrix.util.async_db import Database, Scheme, SQLiteCursor
from mautrix.util.async_db import Database, Scheme
# TODO make this import unconditional after updating mautrix-python
try:
from mautrix.util.async_db import SQLiteCursor
except ImportError:
SQLiteCursor = None
@dataclass
@ -193,7 +199,8 @@ class DBManager:
info.link,
info.next_retry,
)
assert isinstance(cur, SQLiteCursor)
if SQLiteCursor is not None:
assert isinstance(cur, SQLiteCursor)
info.id = cur.lastrowid
else:
info.id = await self.db.fetchval(