main.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. """
  2. [App documentation here]
  3. @author:[author], [year]
  4. """
  5. import sys
  6. import yaml
  7. from PyQt5.Qt import QApplication
  8. from PyQt5.QtWidgets import QMessageBox
  9. from core import db
  10. from core.constants import SETTING_PATH, SETTING_DIST_PATH
  11. from core.indexer import Indexer
  12. from core.logging_ import Logger
  13. from ui.window import MainWindow
  14. try:
  15. # Necessary to ensure that stacktraces are printed when using PyQt5
  16. # Tough, could make pythonw.exe crash
  17. import ipdb # @UnusedImport
  18. except:
  19. pass
  20. logger = Logger.get()
  21. main_window = None
  22. # Configure how errors are processed
  23. sys_err = sys.excepthook
  24. def err_handler(typ, value, trace):
  25. QApplication.restoreOverrideCursor()
  26. if main_window:
  27. QMessageBox.critical(main_window, "Erreur: {}".format(typ.__name__), """{}""".format(value))
  28. sys_err(typ, value, trace)
  29. sys.excepthook = err_handler
  30. # create settings file
  31. if not SETTING_PATH.exists():
  32. SETTING_DIST_PATH.copy(SETTING_PATH)
  33. settings = yaml.load(SETTING_PATH.text(), Loader=yaml.FullLoader)
  34. # Create db if not existing
  35. try:
  36. db.create()
  37. except FileExistsError:
  38. pass
  39. # Start UI
  40. logger.info("-- Start --")
  41. app = QApplication(sys.argv)
  42. indexer = Indexer()
  43. indexer.start()
  44. main_window = MainWindow(settings)
  45. main_window.show()
  46. try:
  47. r = app.exec_()
  48. finally:
  49. try:
  50. settings = main_window.currentSettings()
  51. with open(SETTING_PATH, 'w') as f:
  52. yaml.dump(settings, f)
  53. finally:
  54. indexer.stop()
  55. db.close()
  56. logger.info("-- Closed --")