view python/tmpbuffer.py @ 659:e5be32fea639

https://gist.github.com/lelandbatey/8677901
author Jeff Hammel <k0scist@gmail.com>
date Thu, 03 Apr 2014 11:53:23 -0700
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