Correct pipe-handling behavior

1 files changed, 9 insertions, 1 deletions

diff --git a/vgstash b/vgstash
index 1196920..3ddd7ae 100755
--- a/vgstash
+++ b/vgstash
@@ -15,6 +15,13 @@ OWNERSHIP = 1
 PROGRESS = 1
 TABLE_WIDTH = 80
 
+def safe_print(line):
+    try:
+        print(line, flush=True)
+    # We're catching this in case the other end of a pipe exits early
+    except BrokenPipeError:
+        sys.stderr.close()
+
 def set_env():
     '''Ensures environment variables are respected. Sets defaults if
     they're not present. If the defaults are not usable, it throws an
@@ -343,7 +350,8 @@ def row_format(args, header):
     -------------------------------------------------
     1234 | This is a title | Wii U VC |  *  | F I B C
     """
-    print(" | ".join((gidstr, titlestr, systemstr, ownstr, statstr)))
+    safe_print(" | ".join((gidstr, titlestr, systemstr, ownstr, statstr)))
+
 
 def main():
     # Establish our important variables