diff options
| author | Patrick Cloke <clokep@users.noreply.github.com> | 2021-10-01 07:02:32 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-01 07:02:32 -0400 |
| commit | 7e440520c9b370ce008c6a65c5dd87a360a6457c (patch) | |
| tree | de7da69fea19d210f2f9b8650a140aaf97c95285 /synapse/storage | |
| parent | Clean-up registration tests (#10945) (diff) | |
| download | synapse-7e440520c9b370ce008c6a65c5dd87a360a6457c.tar.xz | |
Add type hints to filtering classes. (#10958)
Diffstat (limited to 'synapse/storage')
| -rw-r--r-- | synapse/storage/databases/main/filtering.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/synapse/storage/databases/main/filtering.py b/synapse/storage/databases/main/filtering.py index bb244a03c0..434986fa64 100644 --- a/synapse/storage/databases/main/filtering.py +++ b/synapse/storage/databases/main/filtering.py @@ -12,6 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. +from typing import Union + from canonicaljson import encode_canonical_json from synapse.api.errors import Codes, SynapseError @@ -22,7 +24,9 @@ from synapse.util.caches.descriptors import cached class FilteringStore(SQLBaseStore): @cached(num_args=2) - async def get_user_filter(self, user_localpart, filter_id): + async def get_user_filter( + self, user_localpart: str, filter_id: Union[int, str] + ) -> JsonDict: # filter_id is BIGINT UNSIGNED, so if it isn't a number, fail # with a coherent error message rather than 500 M_UNKNOWN. try: @@ -40,7 +44,7 @@ class FilteringStore(SQLBaseStore): return db_to_json(def_json) - async def add_user_filter(self, user_localpart: str, user_filter: JsonDict) -> str: + async def add_user_filter(self, user_localpart: str, user_filter: JsonDict) -> int: def_json = encode_canonical_json(user_filter) # Need an atomic transaction to SELECT the maximal ID so far then |