cometrics.py

import os
import sys
# Custom library imports
import traceback
from tkinter import messagebox

# Setup environment variables
cwd = os.getcwd()
ffmpeg_path = os.path.join(cwd, r'external_bin\ffmpeg\ffmpeg-win64-v4.2.2.exe')
if os.path.exists(ffmpeg_path):
    os.environ['IMAGEIO_FFMPEG_EXE'] = os.path.join(cwd, r'external_bin\ffmpeg\ffmpeg-win64-v4.2.2.exe')

import imageio_ffmpeg
from tkinter_utils import LoadingWindow
from config_utils import ConfigUtils
from session_manager_ui import SessionManagerWindow
from logger_util import CreateLogger, log_startup
from logger_util.logger_util import get_process_memory
from project_setup_ui import ProjectSetupWindow
from ui_params import cometrics_data_root, cometrics_ver_root, cometrics_version


def main(config_file, first_time_user):
    project_setup = ProjectSetupWindow(config_file, first_time_user)
    return SessionManagerWindow(config_file, project_setup)


if __name__ == "__main__":
    # Check root dir
    if not os.path.exists(cometrics_data_root):
        os.mkdir(cometrics_data_root)
    if not os.path.exists(cometrics_ver_root):
        os.mkdir(cometrics_ver_root)
    # Reroute stdout and stderr
    CreateLogger(os.path.join(cometrics_ver_root, 'logs'))
    # Log computer information
    log_startup()
    print(f"STARTUP: {cometrics_version}")
    print(f"INFO: imageio_ffmpeg exe location - {imageio_ffmpeg.get_ffmpeg_exe()}")
    # Load our configuration
    config = ConfigUtils()
    config.set_cwd(cwd)
    config.set_logs_dir(os.path.join(cometrics_ver_root, 'logs'))
    if not os.path.exists(f"{cometrics_ver_root}/Projects"):
        os.mkdir(f"{cometrics_ver_root}/Projects")
    while True:
        first_time = config.get_first_time()
        setup = ProjectSetupWindow(config, first_time)
        if setup.setup_complete:
            while True:
                try:
                    config = ConfigUtils()
                    manager = SessionManagerWindow(config, setup)
                    get_process_memory()
                    setup_again = manager.setup_again
                    if manager.close_program:
                        break
                    LoadingWindow(objects=manager)
                    if setup_again:
                        break
                except Exception as e:
                    messagebox.showerror("Error", f"Exception encountered:\n{str(e)}\n{traceback.print_exc()}")
                    manager = None
                    sys.exit(1)
            if manager:
                if manager.close_program:
                    sys.exit(0)
        else:
            break
    sys.exit(0)