cura_app.py 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138
  1. #!/usr/bin/env python3
  2. # Copyright (c) 2018 Ultimaker B.V.
  3. # Cura is released under the terms of the LGPLv3 or higher.
  4. import argparse
  5. import faulthandler
  6. import os
  7. import sys
  8. from UM.Platform import Platform
  9. parser = argparse.ArgumentParser(prog = "cura",
  10. add_help = False)
  11. parser.add_argument("--debug",
  12. action="store_true",
  13. default = False,
  14. help = "Turn on the debug mode by setting this option."
  15. )
  16. parser.add_argument("--trigger-early-crash",
  17. dest = "trigger_early_crash",
  18. action = "store_true",
  19. default = False,
  20. help = "FOR TESTING ONLY. Trigger an early crash to show the crash dialog."
  21. )
  22. known_args = vars(parser.parse_known_args()[0])
  23. if not known_args["debug"]:
  24. def get_cura_dir_path():
  25. if Platform.isWindows():
  26. return os.path.expanduser("~/AppData/Roaming/cura")
  27. elif Platform.isLinux():
  28. return os.path.expanduser("~/.local/share/cura")
  29. elif Platform.isOSX():
  30. return os.path.expanduser("~/Library/Logs/cura")
  31. if hasattr(sys, "frozen"):
  32. dirpath = get_cura_dir_path()
  33. os.makedirs(dirpath, exist_ok = True)
  34. sys.stdout = open(os.path.join(dirpath, "stdout.log"), "w", encoding = "utf-8")
  35. sys.stderr = open(os.path.join(dirpath, "stderr.log"), "w", encoding = "utf-8")
  36. # WORKAROUND: GITHUB-88 GITHUB-385 GITHUB-612
  37. if Platform.isLinux(): # Needed for platform.linux_distribution, which is not available on Windows and OSX
  38. # For Ubuntu: https://bugs.launchpad.net/ubuntu/+source/python-qt4/+bug/941826
  39. # The workaround is only needed on Ubuntu+NVidia drivers. Other drivers are not affected, but fine with this fix.
  40. try:
  41. import ctypes
  42. from ctypes.util import find_library
  43. libGL = find_library("GL")
  44. ctypes.CDLL(libGL, ctypes.RTLD_GLOBAL)
  45. except:
  46. # GLES-only systems (e.g. ARM Mali) do not have libGL, ignore error
  47. pass
  48. # When frozen, i.e. installer version, don't let PYTHONPATH mess up the search path for DLLs.
  49. if Platform.isWindows() and hasattr(sys, "frozen"):
  50. try:
  51. del os.environ["PYTHONPATH"]
  52. except KeyError:
  53. pass
  54. # WORKAROUND: GITHUB-704 GITHUB-708
  55. # It looks like setuptools creates a .pth file in
  56. # the default /usr/lib which causes the default site-packages
  57. # to be inserted into sys.path before PYTHONPATH.
  58. # This can cause issues such as having libsip loaded from
  59. # the system instead of the one provided with Cura, which causes
  60. # incompatibility issues with libArcus
  61. if "PYTHONPATH" in os.environ.keys(): # If PYTHONPATH is used
  62. PYTHONPATH = os.environ["PYTHONPATH"].split(os.pathsep) # Get the value, split it..
  63. PYTHONPATH.reverse() # and reverse it, because we always insert at 1
  64. for PATH in PYTHONPATH: # Now beginning with the last PATH
  65. PATH_real = os.path.realpath(PATH) # Making the the path "real"
  66. if PATH_real in sys.path: # This should always work, but keep it to be sure..
  67. sys.path.remove(PATH_real)
  68. sys.path.insert(1, PATH_real) # Insert it at 1 after os.curdir, which is 0.
  69. def exceptHook(hook_type, value, traceback):
  70. from cura.CrashHandler import CrashHandler
  71. from cura.CuraApplication import CuraApplication
  72. has_started = False
  73. if CuraApplication.Created:
  74. has_started = CuraApplication.getInstance().started
  75. #
  76. # When the exception hook is triggered, the QApplication may not have been initialized yet. In this case, we don't
  77. # have an QApplication to handle the event loop, which is required by the Crash Dialog.
  78. # The flag "CuraApplication.Created" is set to True when CuraApplication finishes its constructor call.
  79. #
  80. # Before the "started" flag is set to True, the Qt event loop has not started yet. The event loop is a blocking
  81. # call to the QApplication.exec_(). In this case, we need to:
  82. # 1. Remove all scheduled events so no more unnecessary events will be processed, such as loading the main dialog,
  83. # loading the machine, etc.
  84. # 2. Start the Qt event loop with exec_() and show the Crash Dialog.
  85. #
  86. # If the application has finished its initialization and was running fine, and then something causes a crash,
  87. # we run the old routine to show the Crash Dialog.
  88. #
  89. from PyQt5.Qt import QApplication
  90. if CuraApplication.Created:
  91. _crash_handler = CrashHandler(hook_type, value, traceback, has_started)
  92. if CuraApplication.splash is not None:
  93. CuraApplication.splash.close()
  94. if not has_started:
  95. CuraApplication.getInstance().removePostedEvents(None)
  96. _crash_handler.early_crash_dialog.show()
  97. sys.exit(CuraApplication.getInstance().exec_())
  98. else:
  99. _crash_handler.show()
  100. else:
  101. application = QApplication(sys.argv)
  102. application.removePostedEvents(None)
  103. _crash_handler = CrashHandler(hook_type, value, traceback, has_started)
  104. # This means the QtApplication could be created and so the splash screen. Then Cura closes it
  105. if CuraApplication.splash is not None:
  106. CuraApplication.splash.close()
  107. _crash_handler.early_crash_dialog.show()
  108. sys.exit(application.exec_())
  109. # Set exception hook to use the crash dialog handler
  110. sys.excepthook = exceptHook
  111. # Enable dumping traceback for all threads
  112. faulthandler.enable(all_threads = True)
  113. # Workaround for a race condition on certain systems where there
  114. # is a race condition between Arcus and PyQt. Importing Arcus
  115. # first seems to prevent Sip from going into a state where it
  116. # tries to create PyQt objects on a non-main thread.
  117. import Arcus #@UnusedImport
  118. import Savitar #@UnusedImport
  119. from cura.CuraApplication import CuraApplication
  120. app = CuraApplication()
  121. app.run()