Add typing to SyncHandler (#6821)

Co-authored-by: Richard van der Hoff <1389908+richvdh@users.noreply.github.com>
This commit is contained in:
Erik Johnston 2020-02-03 18:05:44 +00:00 committed by GitHub
parent ae5b3104f0
commit b9391c9575
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 377 additions and 345 deletions

1
changelog.d/6821.misc Normal file
View File

@ -0,0 +1 @@
Add type hints to `SyncHandler`.

View File

@ -189,8 +189,14 @@ class EventBase(object):
redacts = _event_dict_property("redacts", None) redacts = _event_dict_property("redacts", None)
room_id = _event_dict_property("room_id") room_id = _event_dict_property("room_id")
sender = _event_dict_property("sender") sender = _event_dict_property("sender")
state_key = _event_dict_property("state_key")
type = _event_dict_property("type")
user_id = _event_dict_property("sender") user_id = _event_dict_property("sender")
@property
def event_id(self) -> str:
raise NotImplementedError()
@property @property
def membership(self): def membership(self):
return self.content["membership"] return self.content["membership"]
@ -281,10 +287,7 @@ class FrozenEvent(EventBase):
else: else:
frozen_dict = event_dict frozen_dict = event_dict
self.event_id = event_dict["event_id"] self._event_id = event_dict["event_id"]
self.type = event_dict["type"]
if "state_key" in event_dict:
self.state_key = event_dict["state_key"]
super(FrozenEvent, self).__init__( super(FrozenEvent, self).__init__(
frozen_dict, frozen_dict,
@ -294,6 +297,10 @@ class FrozenEvent(EventBase):
rejected_reason=rejected_reason, rejected_reason=rejected_reason,
) )
@property
def event_id(self) -> str:
return self._event_id
def __str__(self): def __str__(self):
return self.__repr__() return self.__repr__()
@ -332,9 +339,6 @@ class FrozenEventV2(EventBase):
frozen_dict = event_dict frozen_dict = event_dict
self._event_id = None self._event_id = None
self.type = event_dict["type"]
if "state_key" in event_dict:
self.state_key = event_dict["state_key"]
super(FrozenEventV2, self).__init__( super(FrozenEventV2, self).__init__(
frozen_dict, frozen_dict,

File diff suppressed because it is too large Load Diff

View File

@ -238,8 +238,11 @@ class RedactionTestCase(unittest.HomeserverTestCase):
@defer.inlineCallbacks @defer.inlineCallbacks
def build(self, prev_event_ids): def build(self, prev_event_ids):
built_event = yield self._base_builder.build(prev_event_ids) built_event = yield self._base_builder.build(prev_event_ids)
built_event.event_id = self._event_id
built_event._event_id = self._event_id
built_event._event_dict["event_id"] = self._event_id built_event._event_dict["event_id"] = self._event_id
assert built_event.event_id == self._event_id
return built_event return built_event
@property @property

View File

@ -180,6 +180,7 @@ commands = mypy \
synapse/api \ synapse/api \
synapse/config/ \ synapse/config/ \
synapse/federation/transport \ synapse/federation/transport \
synapse/handlers/sync.py \
synapse/handlers/ui_auth \ synapse/handlers/ui_auth \
synapse/logging/ \ synapse/logging/ \
synapse/module_api \ synapse/module_api \