forked from zietzm/Helmholtz_Test_Bench
Martin Zietz
3b62e41f5a
- moved csv files and unused code files to separate folders - comments and cleanup in config_handling
76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
from os.path import exists
|
|
import traceback
|
|
from tkinter import messagebox
|
|
|
|
import cage_func as func
|
|
from User_Interface import HelmholtzGUI
|
|
from User_Interface import ui_print
|
|
import User_Interface as ui
|
|
import globals as g
|
|
import config_handling as config
|
|
import csv_logging as log
|
|
|
|
|
|
def program_end(): # called on exception or when user closes application
|
|
g.exitFlag = True # tell everything else the application has been closed
|
|
if g.app is not None:
|
|
if g.app.pages[ui.ExecuteCSVMode].csv_thread is not None: # end possible csv execution thread
|
|
g.app.pages[ui.ExecuteCSVMode].csv_thread.stop() # stop thread
|
|
func.shut_down_all() # shut down devices
|
|
|
|
if log.unsaved_data: # There is logged data that has not been saved yet
|
|
# open pop-up to ask user if he wants to save the data:
|
|
save_log = messagebox.askquestion("Save log data?", "There seems to be unsaved logging data. "
|
|
"Do you wish to write it to a file now?")
|
|
if save_log == 'yes': # user has chosen yes
|
|
filepath = log.select_file() # let user select a file to write to
|
|
log.write_to_file(log.log_data, filepath) # write the data to the chosen file
|
|
if g.app is not None:
|
|
g.app.destroy() # close application
|
|
|
|
|
|
try: # start normal operations
|
|
|
|
config.CONFIG_FILE = 'config.ini'
|
|
# ToDo: remember what the last config file was
|
|
if not exists(config.CONFIG_FILE):
|
|
print("Config file not found, creating new from defaults.")
|
|
config.reset_config_to_default()
|
|
config.write_config_to_file(config.CONFIG_OBJECT)
|
|
|
|
config.CONFIG_OBJECT = config.get_config_from_file(config.CONFIG_FILE)
|
|
|
|
print("Starting setup...")
|
|
func.setup_all() # initiate communication, set handles
|
|
|
|
print("\nOpening User Interface...")
|
|
|
|
g.app = HelmholtzGUI()
|
|
g.exitFlag = False
|
|
g.app.state('zoomed') # open maximized
|
|
g.app.StatusDisplay.continuous_label_update(g.app, 500) # initiate regular Status Display updates (ms)
|
|
ui_print("Program Initialized")
|
|
config.check_config(config.CONFIG_OBJECT) # check config for values exceeding limits
|
|
|
|
ui_print("\nStarting setup...") # do it again, so it is printed in the UI console ToDo: do it only once
|
|
func.setup_all() # initiate communication, set handles
|
|
|
|
g.app.protocol("WM_DELETE_WINDOW", program_end) # call program_end function if user closes the application
|
|
|
|
g.app.mainloop()
|
|
|
|
|
|
except Exception as e: # if there is an error, print what happened
|
|
print("\nAn error occurred, Shutting down.")
|
|
# shop pup-up error message:
|
|
message = "%s.\nSee python console traceback for more details. " \
|
|
"\nShutting down devices, check equipment to confirm." % e
|
|
messagebox.showerror("Error!", message)
|
|
print(traceback.print_exc())
|
|
program_end() # safely close everything and shut down devices
|
|
|
|
# ToDo: rework window closing code https://bytes.com/topic/python/answers/431323-detect-tkinter-window-being-closed
|
|
# https://stackoverflow.com/questions/14694408/runtimeerror-main-thread-is-not-in-main-loop
|
|
|
|
# ToDo: logging
|