Make the database upgrade system use logger

[cascardo/ipsilon.git] / ipsilon / install / ipsilon-upgrade-database

diff --git a/ipsilon/install/ipsilon-upgrade-database b/ipsilon/install/ipsilon-upgrade-database
index 7ee18fb..cb1d4fa 100755 (executable)
--- a/ipsilon/install/ipsilon-upgrade-database
+++ b/ipsilon/install/ipsilon-upgrade-database
@@ -5,10 +5,19 @@
 import sys
 from ipsilon import find_config
 from ipsilon.tools import dbupgrade
+import logging
 
+logger = logging.getLogger(__name__)
 
-try:
-    dbupgrade.execute_upgrade(find_config())
-except Exception as ex:
-    print ex
-    sys.exit(1)
+
+if __name__ == '__main__':
+    def_logger = logging.getLogger()
+    ch = logging.StreamHandler(sys.stdout)
+    ch.setLevel(logging.ERROR)
+    def_logger.addHandler(ch)
+
+    try:
+        dbupgrade.execute_upgrade(find_config())
+    except Exception as ex:
+        logger.error(ex)
+        sys.exit(1)