Move initializations to main() (#409)

To enable starting InVesalius by importing app.py and calling main().

Move initializations to main() (#409)
To enable starting InVesalius by importing app.py and calling main().
Olli-Pekka Kahilakoski · GitHub
1 parent df13f481
Showing 1 changed file with 35 additions and 27 deletions Show diff stats
app.py
@@ -490,9 +490,41 @@ def print_events(topic=Publisher.AUTO_TOPIC, **msg_data):
     utils.debug("%s\n\tParameters: %s" % (topic, msg_data))
+def init():
+    """
+    Initialize InVesalius.
+
+    Mostly file-system related initializations.
+    """
+    #Is needed because of pyinstaller
+    multiprocessing.freeze_support()
+
+    #Needed in win 32 exe
+    if hasattr(sys,"frozen") and sys.platform.startswith('win'):
+
+        #Click in the .inv3 file support
+        root = winreg.HKEY_CLASSES_ROOT
+        key = "InVesalius 3.1\InstallationDir"
+        hKey = winreg.OpenKey (root, key, 0, winreg.KEY_READ)
+        value, type_ = winreg.QueryValueEx (hKey, "")
+        path = os.path.join(value,'dist')
+
+        os.chdir(path)
+
+    if not inv_paths.USER_INV_DIR.exists():
+        inv_paths.create_conf_folders()
+        if inv_paths.OLD_USER_INV_DIR.exists():
+            inv_paths.copy_old_files()
+
+    if hasattr(sys,"frozen") and sys.frozen == "windows_exe":
+        # Set system standard error output to file
+        path = inv_paths.USER_LOG_DIR.join("stderr.log")
+        sys.stderr = open(path, "w")
+
+
 def main(connection=None):
     """
-    Initialize InVesalius GUI
+    Start InVesalius.
     Parameters:
         connection: An object to communicate with the outside world.
@@ -505,6 +537,8 @@ def main(connection=None):
           functionality, InVesalius needs to be started by calling the main
           function directly with a proper connection object.
     """
+    init()
+
     options, args = parse_command_line()
     session = ses.Session()
@@ -536,30 +570,4 @@ def main(connection=None):
 if __name__ == '__main__':
-    #Is needed because of pyinstaller
-    multiprocessing.freeze_support()
-
-    #Needed in win 32 exe
-    if hasattr(sys,"frozen") and sys.platform.startswith('win'):
-
-        #Click in the .inv3 file support
-        root = winreg.HKEY_CLASSES_ROOT
-        key = "InVesalius 3.1\InstallationDir"
-        hKey = winreg.OpenKey (root, key, 0, winreg.KEY_READ)
-        value, type_ = winreg.QueryValueEx (hKey, "")
-        path = os.path.join(value,'dist')
-
-        os.chdir(path)
-
-    if not inv_paths.USER_INV_DIR.exists():
-        inv_paths.create_conf_folders()
-        if inv_paths.OLD_USER_INV_DIR.exists():
-            inv_paths.copy_old_files()
-
-    if hasattr(sys,"frozen") and sys.frozen == "windows_exe":
-        # Set system standard error output to file
-        path = inv_paths.USER_LOG_DIR.join("stderr.log")
-        sys.stderr = open(path, "w")
-
-    # Init application
     main()