From e7fb656397ec967390a85ae57a39aa73ef1c70a2 Mon Sep 17 00:00:00 2001 From: Patrick Cloke Date: Wed, 2 Sep 2020 14:17:49 -0400 Subject: [PATCH] Add an overload for simple_select_one_onecol_txn. --- changelog.d/8235.misc | 1 + synapse/storage/database.py | 24 ++++++++++++++++++++++++ 2 files changed, 25 insertions(+) create mode 100644 changelog.d/8235.misc diff --git a/changelog.d/8235.misc b/changelog.d/8235.misc new file mode 100644 index 000000000000..3a7a352c4f6d --- /dev/null +++ b/changelog.d/8235.misc @@ -0,0 +1 @@ +Add type hints to `StreamStore`. diff --git a/synapse/storage/database.py b/synapse/storage/database.py index 8851710d4718..78ca6d8346d8 100644 --- a/synapse/storage/database.py +++ b/synapse/storage/database.py @@ -1149,6 +1149,30 @@ async def simple_select_one_onecol( allow_none=allow_none, ) + @overload + @classmethod + def simple_select_one_onecol_txn( + cls, + txn: LoggingTransaction, + table: str, + keyvalues: Dict[str, Any], + retcol: Iterable[str], + allow_none: Literal[False] = False, + ) -> Any: + ... + + @overload + @classmethod + def simple_select_one_onecol_txn( + cls, + txn: LoggingTransaction, + table: str, + keyvalues: Dict[str, Any], + retcol: Iterable[str], + allow_none: Literal[True] = True, + ) -> Optional[Any]: + ... + @classmethod def simple_select_one_onecol_txn( cls,