From 8a4382f74bc25152106050c9dded232f2c03573b Mon Sep 17 00:00:00 2001 From: Ethan White Date: Sat, 25 Jan 2014 19:48:04 -0500 Subject: [PATCH] Cleanup database connection following command line installation Previously lib.engine.final_cleanup was only triggered by the GUI. This change results in the CLI also triggering a final cleanup after installing a dataset. This has the added benefit of allowing text file based output formats to add any necessary end of file information. --- __main__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/__main__.py b/__main__.py index 8b59e9a7a..363ec2dbb 100644 --- a/__main__.py +++ b/__main__.py @@ -112,6 +112,7 @@ def main(): print "=> Installing", dataset.name try: dataset.download(engine, debug=debug) + dataset.engine.final_cleanup() except KeyboardInterrupt: pass except Exception as e: