PR tweaks: set earlier on and use 'as json' for compat

This commit is contained in:
Kegan Dougal 2015-03-02 09:53:00 +00:00
parent 806a6c886a
commit ebc4830666
3 changed files with 12 additions and 16 deletions

View File

@ -219,20 +219,17 @@ class ApplicationServiceStore(SQLBaseStore):
# get all rooms matching the room ID regex.
room_entries = yield self.get_all_rooms() # RoomEntry list
matching_room_id_list = [
matching_room_list = set([
r.room_id for r in room_entries if
service.is_interested_in_room(r.room_id)
]
])
# resolve room IDs for matching room alias regex.
room_alias_mappings = yield self.get_all_associations()
matching_alias_list = [
matching_room_list |= set([
r.room_id for r in room_alias_mappings if
service.is_interested_in_alias(r.room_alias)
]
room_ids_matching_alias_or_id = set(
matching_room_id_list + matching_alias_list
)
])
# get all rooms for every user for this AS. This is scoped to users on
# this HS only.
@ -241,11 +238,10 @@ class ApplicationServiceStore(SQLBaseStore):
u["name"] for u in user_list if
service.is_interested_in_user(u["name"])
]
rooms_for_user_matching_user_id = [] # RoomsForUser list
rooms_for_user_matching_user_id = set() # RoomsForUser list
for user_id in user_list:
rooms_for_user = yield self.get_rooms_for_user(user_id)
rooms_for_user_matching_user_id += rooms_for_user
rooms_for_user_matching_user_id = set(rooms_for_user_matching_user_id)
rooms_for_user_matching_user_id |= set(rooms_for_user)
# make RoomsForUser tuples for room ids and aliases which are not in the
# main rooms_for_user_list - e.g. they are rooms which do not have AS
@ -253,7 +249,7 @@ class ApplicationServiceStore(SQLBaseStore):
known_room_ids = [r.room_id for r in rooms_for_user_matching_user_id]
missing_rooms_for_user = [
RoomsForUser(r, service.sender, "join") for r in
room_ids_matching_alias_or_id if r not in known_room_ids
matching_room_list if r not in known_room_ids
]
rooms_for_user_matching_user_id |= set(missing_rooms_for_user)

View File

@ -93,7 +93,7 @@ class RegistrationStore(SQLBaseStore):
)
def get_all_users(self):
query = ("SELECT users.name FROM users")
query = "SELECT users.name FROM users"
return self._execute(
self.cursor_to_dict,
query

View File

@ -43,7 +43,7 @@ from synapse.util.logutils import log_function
from collections import namedtuple
import logging
import simplejson
import simplejson as json
logger = logging.getLogger(__name__)
@ -178,7 +178,7 @@ class StreamStore(SQLBaseStore):
# interested in was invited to a room. We'll be passing this
# through _get_events_txn later, so ignore the fact that this
# may be a redacted event.
event_content = simplejson.loads(row["unrecognized_keys"])
event_content = json.loads(row["unrecognized_keys"])
if (service.is_interested_in_user(
event_content.get("state_key"))):
return True
@ -202,7 +202,7 @@ class StreamStore(SQLBaseStore):
self._set_before_and_after(ret, rows)
if rows:
key = "s%d" % max([r["stream_ordering"] for r in rows])
key = "s%d" % max(r["stream_ordering"] for r in rows)
else:
# Assume we didn't get anything because there was nothing to
@ -271,7 +271,7 @@ class StreamStore(SQLBaseStore):
self._set_before_and_after(ret, rows)
if rows:
key = "s%d" % max([r["stream_ordering"] for r in rows])
key = "s%d" % max(r["stream_ordering"] for r in rows)
else:
# Assume we didn't get anything because there was nothing to