From: martin f. krafft Date: Wed, 23 Aug 2023 05:02:29 +0000 (+1200) Subject: buildmimetree.py: allow tempdir to be controlled X-Git-Url: https://git.madduck.net/etc/neomutt.git/commitdiff_plain/5f583dd6f81b280645e008beca34c64aa63c6a0f buildmimetree.py: allow tempdir to be controlled --- diff --git a/.config/neomutt/buildmimetree.py b/.config/neomutt/buildmimetree.py index bd891a2..40bd856 100755 --- a/.config/neomutt/buildmimetree.py +++ b/.config/neomutt/buildmimetree.py @@ -7,7 +7,7 @@ # Configuration: # neomuttrc (needs to be a single line): # macro compose B "\ -# source '$my_confdir/buildmimetree.py setup|'\ +# source '$my_confdir/buildmimetree.py setup --tempdir $tempdir|'\ # sourc e \$my_mdwn_postprocess_cmd_file\ # " "Convert message into a modern MIME tree with inline images" # @@ -67,6 +67,12 @@ def parse_cli_args(*args, **kwargs): help="Turn on debug logging of commands generated to stderr", ) + parser_setup.add_argument( + "--tempdir", + default=None, + help="Specify temporary directory to use for attachments", + ) + parser_setup.add_argument( "--extension", "-x", @@ -96,6 +102,12 @@ def parse_cli_args(*args, **kwargs): help="Turn on debugging to stderr of the MIME tree walk", ) + parser_massage.add_argument( + "--tempdir", + default=None, + help="Specify temporary directory to use for attachments", + ) + parser_massage.add_argument( "--extensions", metavar="EXTENSIONS", @@ -241,7 +253,12 @@ def collect_inline_images( def convert_markdown_to_html( - origtext, draftpath, *, filewriter_fn=filewriter_fn, extensions=None + origtext, + draftpath, + *, + filewriter_fn=filewriter_fn, + tempdir=None, + extensions=None, ): origtext, htmltext, images = markdown_with_inline_image_support( origtext, extensions=extensions @@ -262,7 +279,9 @@ def convert_markdown_to_html( "alternative", [textpart, htmlpart], "Group of alternative content" ) - imgparts = collect_inline_images(images, filewriter_fn=filewriter_fn) + imgparts = collect_inline_images( + images, tempdir=tempdir, filewriter_fn=filewriter_fn + ) if imgparts: return Multipart( "relative", [altpart] + imgparts, "Group of related content" @@ -377,17 +396,24 @@ class MuttCommands: def do_setup( - extensions=None, *, out_f=sys.stdout, temppath=None, debug_commands=False + extensions=None, + *, + out_f=sys.stdout, + temppath=None, + tempdir=None, + debug_commands=False, ): extensions = extensions or [] temppath = temppath or pathlib.Path( - tempfile.mkstemp(prefix="muttmdwn-")[1] + tempfile.mkstemp(prefix="muttmdwn-", dir=tempdir)[1] ) cmds = MuttCommands(out_f, debug=debug_commands) editor = f"{sys.argv[0]} massage --write-commands-to {temppath}" if extensions: editor = f'{editor} --extensions {",".join(extensions)}' + if tempdir: + editor = f"{editor} --tempdir {tempdir}" if debug_commands: editor = f"{editor} --debug-commands" @@ -407,6 +433,7 @@ def do_massage( *, extensions=None, converter=convert_markdown_to_html, + tempdir=None, debug_commands=False, debug_walk=False, ): @@ -429,7 +456,7 @@ def do_massage( cmds.flush() extensions = extensions.split(",") if extensions else [] - tree = converter(draft_f.read(), draftpath, extensions=extensions) + tree = converter(draft_f.read(), draftpath, tempdir=tempdir, extensions=extensions) mimetree = MIMETreeDFWalker(debug=debug_walk) @@ -514,7 +541,11 @@ if __name__ == "__main__": if args.send_message: raise NotImplementedError() - do_setup(args.extensions, debug_commands=args.debug_commands) + do_setup( + args.extensions, + tempdir=args.tempdir, + debug_commands=args.debug_commands, + ) elif args.mode == "massage": with open(args.MAILDRAFT, "r") as draft_f, open( @@ -525,6 +556,7 @@ if __name__ == "__main__": pathlib.Path(args.MAILDRAFT), cmd_f, extensions=args.extensions, + tempdir=args.tempdir, debug_commands=args.debug_commands, debug_walk=args.debug_walk, ) @@ -686,7 +718,7 @@ try: return StringIO(text or const1) def test_do_massage_basic(self, const1, string_io, capsys): - def converter(drafttext, draftpath, extensions): + def converter(drafttext, draftpath, extensions, tempdir): return Part("text", "plain", draftpath, orig=True) do_massage( @@ -709,7 +741,7 @@ try: def test_do_massage_fulltree( self, string_io, const1, basic_mime_tree, capsys ): - def converter(drafttext, draftpath, extensions): + def converter(drafttext, draftpath, extensions, tempdir): return basic_mime_tree do_massage( @@ -864,7 +896,6 @@ try: "AAAABCAAAAAA6fptVAAAACklEQVQI12P4DwABAQEAG7buVgAA" ) - def test_markdown_inline_image_processor_base64(self, test_png): text = f"![1px white inlined]({test_png})" text, html, images = markdown_with_inline_image_support(text)