Philipp Hörist pushed to branch master at gajim / python-nbxmpp

Commits:

2 changed files:

Changes:

  • nbxmpp/modules/omemo.py
    ... ... @@ -161,7 +161,7 @@ class OMEMO(BaseModule):
    161 161
             if not items:
    
    162 162
                 yield task.set_result(None)
    
    163 163
     
    
    164
    -        yield _parse_bundle(items[0])
    
    164
    +        yield _parse_bundle(items[0], device_id)
    
    165 165
     
    
    166 166
     
    
    167 167
     def _parse_omemo_message(stanza):
    
    ... ... @@ -235,7 +235,7 @@ def _parse_omemo_message(stanza):
    235 235
         return OMEMOMessage(sid=sid, iv=iv, keys=keys, payload=payload)
    
    236 236
     
    
    237 237
     
    
    238
    -def _parse_bundle(item):
    
    238
    +def _parse_bundle(item, device_id):
    
    239 239
         '''
    
    240 240
         <item id='current'>
    
    241 241
           <bundle xmlns='eu.siacs.conversations.axolotl'>
    
    ... ... @@ -317,7 +317,9 @@ def _parse_bundle(item):
    317 317
     
    
    318 318
             result['otpks'].append({'key': key, 'id': id_})
    
    319 319
     
    
    320
    -    return OMEMOBundle(**result)
    
    320
    +    return OMEMOBundle(**result,
    
    321
    +                       device_id=device_id,
    
    322
    +                       namespace=Namespace.OMEMO_TEMP)
    
    321 323
     
    
    322 324
     
    
    323 325
     def _make_bundle(bundle):
    

  • nbxmpp/structs.py
    ... ... @@ -732,6 +732,8 @@ class OMEMOBundle(NamedTuple):
    732 732
         spk_signature: bytes
    
    733 733
         ik: bytes
    
    734 734
         otpks: list[dict[str, str]]
    
    735
    +    device_id: int = -1
    
    736
    +    namespace: str = Namespace.OMEMO_TEMP
    
    735 737
     
    
    736 738
         def pick_prekey(self) -> dict[str, str]:
    
    737 739
             return random.SystemRandom().choice(self.otpks)