simple-text-generator-ui.py

import hashlib
import logging
import os
import pathlib
import subprocess
import sys

from simpletextgenerator.logging_setup import setup_logging
from simpletextgenerator.ui.menu import draw_main_menu

version = "0.3.0"
logger = logging.getLogger("ui")


def _get_exe_hashes() -> str:
    try:
        hash1 = _get_simple_text_generator_ui_exe_hash()
        hash2 = _get_run_training_exe_hash()
    except Exception as e:
        logger.debug(e)
        return ""

    return f"simple-text-generator md5sum: {hash1}; run_training.exe: {hash2}"


def _get_simple_text_generator_ui_exe_hash() -> str:
    """Gets md5sum of Windows UI binary if present."""
    return hashlib.md5(open('simple-text-generator-ui.exe', 'rb').read()).hexdigest()


def _get_run_training_exe_hash() -> str:
    """Gets md5sum of Windows training binary if present."""
    return hashlib.md5(open('run_training.exe', 'rb').read()).hexdigest()


def _get_git_hash_or_empty() -> str:
    """Returns current git commit hash if running from checked out repository."""
    try:
        return f"git hash: {_get_git_revision_hash()}"
    except Exception as e:
        logger.debug(e)
        return ""


def _get_git_revision_hash():
    return subprocess.check_output(['git', 'rev-parse', 'HEAD'])


def _create_dir_if_not_exists(directory):
    if not os.path.exists(directory):
        os.makedirs(directory)


def _log_environment():
    logger.info(sys.version)
    logger.info(f"Platform: {sys.platform}")
    logger.info(f"simple-text-generator code version: {version}")
    commit = _get_git_hash_or_empty()
    if commit != "":
        logger.info(f"commit: {commit}")
    hashes = _get_exe_hashes()
    if hashes != "":
        logger.info(f"{hashes}")


if __name__ == "__main__":
    pathlib.Path('logs').mkdir(exist_ok=True)

    log_file_name = "logs/simple-text-generator-ui.log"
    previous_log_file_name = "logs/simple-text-generator-ui_previous.log"
    setup_logging(logger, log_file_name, previous_log_file_name)
    _log_environment()

    _create_dir_if_not_exists("projects")
    _create_dir_if_not_exists("projects/archive")

    try:
        window_manager = draw_main_menu()
    except Exception as e:
        logger.error(e)
        raise e