matrix/weechat-matrix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow saving attachment without running a plumber

Browse source 
Add a second (optional) positional argument to specify the destination
file. If a file has been specified, just save without running the
default plumber.

Specifying a custom plumber allows to both save and run the plumber.
This commit is contained in:
Yuri D'Elia 2020-10-12 01:41:05 +02:00
parent e43edbcbc9
commit fc69f77923
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
contrib/matrix_decrypt.py
View file

@ -44,6 +44,7 @@ def main():
description='Download and decrypt matrix attachments' description='Download and decrypt matrix attachments'
) )
parser.add_argument('url', help='the url of the attachment') parser.add_argument('url', help='the url of the attachment')
parser.add_argument('file', nargs='?', help='save attachment to <file>')
parser.add_argument('--plumber', parser.add_argument('--plumber',
help='program that gets called with the ' help='program that gets called with the '
'dowloaded file') 'dowloaded file')
@ -68,10 +69,18 @@ def main():
print("Error downloading file") print("Error downloading file")
return -2 return -2
plumber = args.plumber or "/usr/bin/rifle" plumber = args.plumber
plaintext = decrypt_attachment(request.content, key, hash, iv) plaintext = decrypt_attachment(request.content, key, hash, iv)
file_name = save_file(plaintext)
if args.file is None:
file_name = save_file(plaintext)
if plumber is None:
plumber = "/usr/bin/rifle"
else:
file_name = args.file
open(file_name, "wb").write(plaintext)
if plumber is not None:
subprocess.run([plumber, "{file}".format(file=file_name)]) subprocess.run([plumber, "{file}".format(file=file_name)])
return 0 return 0