diff options
| author | David Robertson <davidr@element.io> | 2021-10-08 15:25:16 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-08 14:25:16 +0000 |
| commit | 51a5da74ccd383806378b53ee8a09e27a8829f31 (patch) | |
| tree | 23b1f1e5aebd931ce9e3faba8e8f301d1f079595 /synapse/storage/databases/main/pusher.py | |
| parent | Relax `ignore-missing-imports` for modules that have stubs now and update myp... (diff) | |
| download | synapse-51a5da74ccd383806378b53ee8a09e27a8829f31.tar.xz | |
Annotate synapse.storage.util (#10892)
Also mark `synapse.streams` as having has no untyped defs Co-authored-by: Sean Quah <8349537+squahtx@users.noreply.github.com>
Diffstat (limited to 'synapse/storage/databases/main/pusher.py')
| -rw-r--r-- | synapse/storage/databases/main/pusher.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/synapse/storage/databases/main/pusher.py b/synapse/storage/databases/main/pusher.py index a93caae8d0..b73ce53c91 100644 --- a/synapse/storage/databases/main/pusher.py +++ b/synapse/storage/databases/main/pusher.py @@ -18,8 +18,7 @@ from typing import TYPE_CHECKING, Any, Dict, Iterable, Iterator, List, Optional, from synapse.push import PusherConfig, ThrottleParams from synapse.storage._base import SQLBaseStore, db_to_json -from synapse.storage.database import DatabasePool -from synapse.storage.types import Connection +from synapse.storage.database import DatabasePool, LoggingDatabaseConnection from synapse.storage.util.id_generators import StreamIdGenerator from synapse.types import JsonDict from synapse.util import json_encoder @@ -32,7 +31,12 @@ logger = logging.getLogger(__name__) class PusherWorkerStore(SQLBaseStore): - def __init__(self, database: DatabasePool, db_conn: Connection, hs: "HomeServer"): + def __init__( + self, + database: DatabasePool, + db_conn: LoggingDatabaseConnection, + hs: "HomeServer", + ): super().__init__(database, db_conn, hs) self._pushers_id_gen = StreamIdGenerator( db_conn, "pushers", "id", extra_tables=[("deleted_pushers", "stream_id")] |