Add some type hints to datastore. (#12477)

matrix-org · May 10, 2022 · 989fa33 · 989fa33
1 parent 147f098
commit 989fa33
Show file tree

Hide file tree

Showing 4 changed files with 122 additions and 71 deletions.
diff --git a/changelog.d/12477.misc b/changelog.d/12477.misc
@@ -0,0 +1 @@
+Add some type hints to datastore.
diff --git a/synapse/events/snapshot.py b/synapse/events/snapshot.py
@@ -15,6 +15,7 @@
 
 import attr
 from frozendict import frozendict
+from typing_extensions import Literal
 
 from twisted.internet.defer import Deferred
 
@@ -106,7 +107,7 @@ class EventContext:
             incomplete state.
     """
 
-    rejected: Union[bool, str] = False
+    rejected: Union[Literal[False], str] = False
     _state_group: Optional[int] = None
     state_group_before_event: Optional[int] = None
     prev_group: Optional[int] = None