Don't use form of get_state_for_events with None state_key

This commit is contained in:
Erik Johnston 2016-01-19 17:12:46 +00:00
parent 40d9765123
commit 892ee473d9
1 changed files with 13 additions and 6 deletions

View File

@ -53,16 +53,23 @@ class BaseHandler(object):
self.event_builder_factory = hs.get_event_builder_factory()
@defer.inlineCallbacks
def _filter_events_for_clients(self, users, events):
def _filter_events_for_clients(self, user_tuples, events):
""" Returns dict of user_id -> list of events that user is allowed to
see.
"""
# If there is only one user, just get the state for that one user,
# otherwise just get all the state.
if len(user_tuples) == 1:
types = (
(EventTypes.RoomHistoryVisibility, ""),
(EventTypes.Member, user_tuples[0][0]),
)
else:
types = None
event_id_to_state = yield self.store.get_state_for_events(
frozenset(e.event_id for e in events),
types=(
(EventTypes.RoomHistoryVisibility, ""),
(EventTypes.Member, None),
)
types=types
)
forgotten = yield defer.gatherResults([
@ -122,7 +129,7 @@ class BaseHandler(object):
for event in events
if allowed(event, user_id, is_guest)
]
for user_id, is_guest in users
for user_id, is_guest in user_tuples
})
@defer.inlineCallbacks