init

2024-02-28 11:04:05 +01:00
commit db2082311f
49 changed files with 587 additions and 0 deletions
--- a/input_to_browser.py
+++ b/input_to_browser.py
@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# Based on:
+# https://docs.python.org/3/library/email-examples.html
+
+import sys
+import os
+import tempfile
+import mimetypes
+import webbrowser
+from email import policy
+from email.parser import BytesParser
+
+
+raw = sys.stdin.buffer.read()
+
+if not os.isatty(0):
+    fd = os.open('/dev/tty', os.O_RDONLY)
+    if fd < 0:
+        sys.stderr.write('Unable to open an input tty.\n')
+        sys.exit(-1)
+    else:
+        os.dup2(fd, 0)
+        os.close(fd)
+
+msg = BytesParser(policy=policy.default).parsebytes(raw)
+
+# We can extract the richest alternative in order to display it:
+richest = msg.get_body()
+partfiles = {}
+if richest['content-type'].maintype == 'text':
+    if richest['content-type'].subtype == 'plain':
+        for line in richest.get_content().splitlines():
+            print(line)
+        sys.exit()
+    elif richest['content-type'].subtype == 'html':
+        body = richest
+    else:
+        print("Don't know how to display {}".format(richest.get_content_type()))
+        sys.exit()
+elif richest['content-type'].content_type == 'multipart/related':
+    body = richest.get_body(preferencelist=('html'))
+    for part in richest.iter_attachments():
+        fn = part.get_filename()
+        if fn:
+            extension = os.path.splitext(part.get_filename())[1]
+        else:
+            extension = mimetypes.guess_extension(part.get_content_type())
+        with tempfile.NamedTemporaryFile(suffix=extension, delete=False) as f:
+            f.write(part.get_content())
+            # again strip the <> to go from email form of cid to html form.
+            partfiles[part['content-id'][1:-1]] = f.name
+else:
+    print("Don't know how to display {}".format(richest.get_content_type()))
+    sys.exit()
+with tempfile.NamedTemporaryFile(mode='w', delete=False) as f:
+    # The magic_html_parser has to rewrite the href="cid:...." attributes to
+    # point to the filenames in partfiles.  It also has to do a safety-sanitize
+    # of the html.  It could be written using html.parser.
+    template = """
+    <!DOCTYPE html><html>
+    <head><meta charset="utf-8"></head>
+    <body>{}</body>
+    </html>
+    """
+    f.write(template.format(body.get_content()))
+webbrowser.open(f.name)
+# os.remove(f.name)
+for fn in partfiles.values():
+    os.remove(fn)