view python/tmpbuffer.py @ 824:16200adbd4f0

work around exit code for no changes
author Jeff Hammel <k0scist@gmail.com>
date Tue, 07 Feb 2017 10:28:00 -0800
parents e07b99607d27
children b8510769d001
line wrap: on
line source

#!/usr/bin/env python

# from http://k0s.org/hg/bitsyblog/file/5c04cf601aba/bitsyblog/blogme.py

import optparse
import os
import subprocess
import tempfile

def tmpbuffer(editor=None):
    """open an editor and retreive the resulting editted buffer"""

    if not editor:
        editor = os.environ.get('EDITOR')
        if not editor:
            raise Exception("tmpbuffer: editor not supplied and EDITOR not defined")
    tmpfile = tempfile.mktemp(suffix='.txt')
    cmdline = editor.split() # XXX shlex would be more powerful
    cmdline.append(tmpfile)
    edit = subprocess.call(cmdline)
    buffer = file(tmpfile).read().strip()
    os.remove(tmpfile)
    return buffer

if __name__ == '__main__':
    # purely for testing/illustration purposes
    contents = tmpbuffer()
    print contents