main: Handle BrokenPipeError cleanly

Whilst not recommended, quarg may happen to be used in conjunction with
head(1), tail(1), or any other program that will send SIGPIPE. This
commit makes sure that we close the database connection properly if we
receive that signal.

1 files changed, 1 insertions, 1 deletions

diff --git a/quarg/main.py b/quarg/main.py
index 29138e3..5e0aaf8 100644
--- a/quarg/main.py
+++ b/quarg/main.py
@@ -144,7 +144,7 @@ def main():
             print(formatter.format_from(row))
     except exc.SQLAlchemyError as err:
         errx(err)
-    except KeyboardInterrupt:
+    except (KeyboardInterrupt, BrokenPipeError):
         session.rollback()
         session.close()
         sys.exit(1)