view unixSoft/bin/patchpipe @ 439:2325dea339ca

smailq: vendor script to ease handling outgoing mail From commit 5b83ca873f1dc9117a9b3590f0aa07fe2806fce9 of http://git.sthu.org/repos/smailq.git - documented at https://www.sthu.org/code/smailq.html.
author Augie Fackler <raf@durin42.com>
date Sat, 15 Jul 2017 12:57:17 -0400
parents c7a9cd793e37
children a5a4f9e12c9f
line wrap: on
line source

#!/usr/bin/env python
import base64
import os
import subprocess
import sys
import tempfile

_PATCHHDR = '# HG changeset patch'

_BASE64_MAGIC = 'Content-Transfer-Encoding: base64'


def main(argv):
    tf = tempfile.NamedTemporaryFile(suffix='.diff')
    d = sys.stdin.read()
    if _PATCHHDR in d:
        junk, d = d.split(_PATCHHDR, 1)
        d = _PATCHHDR + d
    elif _BASE64_MAGIC in d:
        junk, d = d.split('\n\n', 1)
        d = d.decode('base64')
    tf.write(d)
    tf.flush()
    subprocess.check_call([os.environ['EDITOR'], tf.name],
                          stdout=sys.stderr)
    sys.stdout.write(open(tf.name).read())


if __name__ == '__main__':
    main(argv=sys.argv)