update to work with newer code, and fix formatting

This commit is contained in:
Hubert Chathi 2019-10-22 21:44:58 -04:00
parent 41ad35b523
commit 1fabf82d50
3 changed files with 7 additions and 6 deletions

View file

@ -468,7 +468,7 @@ class DeviceHandler(DeviceWorkerHandler):
"stream_id": stream_id, "stream_id": stream_id,
"devices": devices, "devices": devices,
"master_key": master_key, "master_key": master_key,
"self_signing_key": self_signing_key "self_signing_key": self_signing_key,
} }
@defer.inlineCallbacks @defer.inlineCallbacks

View file

@ -260,7 +260,7 @@ class E2eKeysHandler(object):
Returns: Returns:
defer.Deferred[dict[str, dict[str, dict]]]: map from defer.Deferred[dict[str, dict[str, dict]]]: map from
(master|self_signing|user_signing) -> user_id -> key (master_keys|self_signing_keys|user_signing_keys) -> user_id -> key
""" """
master_keys = {} master_keys = {}
self_signing_keys = {} self_signing_keys = {}
@ -355,10 +355,11 @@ class E2eKeysHandler(object):
ret = {"device_keys": res} ret = {"device_keys": res}
# add in the cross-signing keys # add in the cross-signing keys
cross_signing_keys = yield self.query_cross_signing_keys(device_keys_query, None) cross_signing_keys = yield self.get_cross_signing_keys_from_cache(
device_keys_query, None
)
for key, value in iteritems(cross_signing_keys): ret.update(cross_signing_keys)
ret[key + "_keys"] = value
return ret return ret

View file

@ -214,7 +214,7 @@ class DeviceWorkerStore(SQLBaseStore):
# figure out which cross-signing keys were changed by intersecting the # figure out which cross-signing keys were changed by intersecting the
# update list with the master/self-signing key by user maps # update list with the master/self-signing key by user maps
cross_signing_keys_by_user = {} cross_signing_keys_by_user = {}
for user_id, device_id, stream in updates: for user_id, device_id, stream, _opentracing_context in updates:
if device_id == master_key_by_user.get(user_id, {}).get("pubkey", None): if device_id == master_key_by_user.get(user_id, {}).get("pubkey", None):
result = cross_signing_keys_by_user.setdefault(user_id, {}) result = cross_signing_keys_by_user.setdefault(user_id, {})
result["master_key"] = master_key_by_user[user_id]["key_info"] result["master_key"] = master_key_by_user[user_id]["key_info"]