Update some more checks

Author: tulir

mautrix/crypto/store/asyncpg/store.py

@@ -272,7 +272,7 @@ async def has_group_session(
     async def add_outbound_group_session(self, session: OutboundGroupSession) -> None:
         pickle = session.pickle(self.pickle_key)
         max_age = session.max_age
-        if self.db.scheme == "sqlite":
+        if self.db.scheme == Scheme.SQLITE:
             max_age = max_age.total_seconds()
         await self.db.execute(
             "INSERT INTO crypto_megolm_outbound_session (room_id, session_id, session, shared, "
@@ -316,7 +316,7 @@ async def get_outbound_group_session(self, room_id: RoomID) -> OutboundGroupSess
         if row is None:
             return None
         max_age = row["max_age"]
-        if self.db.scheme == "sqlite":
+        if self.db.scheme == Scheme.SQLITE:
             max_age = timedelta(seconds=max_age)
         return OutboundGroupSession.from_pickle(
             row["session"],

mautrix/util/async_db/connection.pyi

@@ -12,13 +12,17 @@ import asyncpg
 from mautrix.util.logging import TraceLogger
 
 from . import aiosqlite
+from .scheme import Scheme
 
 class LoggingConnection:
-    scheme: str
+    scheme: Scheme
     wrapped: aiosqlite.TxnConnection | asyncpg.Connection
     log: TraceLogger
     def __init__(
-        self, scheme: str, wrapped: aiosqlite.TxnConnection | asyncpg.Connection, log: TraceLogger
+        self,
+        scheme: Scheme,
+        wrapped: aiosqlite.TxnConnection | asyncpg.Connection,
+        log: TraceLogger,
     ) -> None: ...
     async def transaction(self) -> AsyncContextManager[None]: ...
     async def execute(self, query: str, *args: Any, timeout: float | None = None) -> str: ...