Răsfoiți Sursa

Catch exceptions in main function

Kentaro Wada 6 ani în urmă
părinte
comite
ccd7a5e79f
1 a modificat fișierele cu 8 adăugiri și 5 ștergeri
  1. 8 5
      labelme/main.py

+ 8 - 5
labelme/main.py

@@ -14,6 +14,13 @@ from labelme.utils import newIcon
 
 
 def main():
+    try:
+        _main()
+    except Exception as e:
+        logger.error(e)
+
+
+def _main():
     parser = argparse.ArgumentParser()
     parser.add_argument(
         '--version', '-V', action='store_true', help='show version'
@@ -140,14 +147,10 @@ def main():
     )
 
     if reset_config:
-        print('Resetting Qt config: %s' % win.settings.fileName())
+        logger.info('Resetting Qt config: %s' % win.settings.fileName())
         win.settings.clear()
         sys.exit(0)
 
     win.show()
     win.raise_()
     sys.exit(app.exec_())
-
-
-if __name__ == '__main__':
-    main()