lain/megolm_filter
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

allow multiple rooms

Browse source
This commit is contained in:
Lain Iwakura 2025-01-24 17:00:07 -03:00
parent c077b09717
commit 746d8380cb
Signed by: lain
GPG key ID: 89686F4239E80508
2 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
README.md
View file

@ -25,11 +25,10 @@ options:
-p, --plain Returns the plain unencrypted content
```
Using the above example, let's say we want to only get session keys of the room
`!foo:matrix.org`. You can do this fairly easily with:
You can do this fairly easily with:
```bash
$ ./megolm_filter.py element-keys.txt '!foo:matrix.org'
$ ./megolm_filter.py element-keys.txt '!room1:matrix.org' '!room2:matrix.org' ...
```
You need to have PyCryptodome installed in order for this script to work.

6
megolm_filter.py
View file

@ -145,7 +145,7 @@ def dec_session_data(passphrase, session_data):
def main():
parser = argparse.ArgumentParser(description="Operate on megolm session backups.")
parser.add_argument("file", nargs='?', help="megolm session data")
parser.add_argument("room_id", nargs='?', help="Room id to filter (optional)")
parser.add_argument("room_ids", nargs='*', help="Room id to filter (optional)")
parser.add_argument("-o", "--output", help="Output to file")
parser.add_argument("-p", "--plain", dest="mode", const="plain", action="store_const",
help="Returns the plain unencrypted content")
@ -166,9 +166,9 @@ def main():
passphrase = getpass.getpass(f"Backup passphrase [mode={args.mode}]: ")
decrypted_data = dec_session_data(passphrase, data)
if args.room_id:
if args.room_ids:
json_data = json.loads(decrypted_data)
filtered_data = [ key for key in json_data if key['room_id'] == args.room_id ]
filtered_data = [ key for key in json_data if key['room_id'] in args.room_ids ]
if not filtered_data:
bail(f"No keys found for room {args.room_id}")