format.py

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

proc = None
try:
    tmp = tempfile.NamedTemporaryFile(delete=False)
    tmp.write(sys.stdin.read().encode("UTF-8"))
    tmp.close()
    proc = subprocess.Popen([
        "pyformat",
        "-i",
        tmp.name
    ])
    proc.wait()
    with open(tmp.name, "r") as file_:
        sys.stdout.write(file_.read())
    os.unlink(tmp.name)
finally:
    proc.terminate()
sys.exit(proc.returncode)