This repository was archived by the owner on Apr 26, 2024. It is now read-only.
File tree 2 files changed +5
-7
lines changed
2 files changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -510,7 +510,7 @@ async def _get_to_device_messages(
510
510
return []
511
511
512
512
# Retrieve the to-device messages for each user
513
- recipient_user_id_device_id_to_messages = await self .store .get_new_messages (
513
+ recipient_device_to_messages = await self .store .get_new_messages (
514
514
users_appservice_is_interested_in ,
515
515
from_key ,
516
516
new_token ,
@@ -526,7 +526,7 @@ async def _get_to_device_messages(
526
526
for (
527
527
user_id ,
528
528
device_id ,
529
- ), messages in recipient_user_id_device_id_to_messages .items ():
529
+ ), messages in recipient_device_to_messages .items ():
530
530
for message_json in messages :
531
531
# Remove 'message_id' from the to-device message, as it's an internal ID
532
532
message_json .pop ("message_id" , None )
Original file line number Diff line number Diff line change @@ -191,20 +191,18 @@ def get_new_messages_txn(txn: LoggingTransaction):
191
191
192
192
# Create a dictionary of (user ID, device ID) -> list of messages that
193
193
# that device is meant to receive.
194
- recipient_user_id_device_id_to_messages : Dict [
195
- Tuple [str , str ], List [JsonDict ]
196
- ] = {}
194
+ recipient_device_to_messages : Dict [Tuple [str , str ], List [JsonDict ]] = {}
197
195
198
196
for row in txn :
199
197
recipient_user_id = row [0 ]
200
198
recipient_device_id = row [1 ]
201
199
message_dict = db_to_json (row [2 ])
202
200
203
- recipient_user_id_device_id_to_messages .setdefault (
201
+ recipient_device_to_messages .setdefault (
204
202
(recipient_user_id , recipient_device_id ), []
205
203
).append (message_dict )
206
204
207
- return recipient_user_id_device_id_to_messages
205
+ return recipient_device_to_messages
208
206
209
207
return await self .db_pool .runInteraction (
210
208
"get_new_messages" , get_new_messages_txn
You can’t perform that action at this time.
0 commit comments