-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.py
65 lines (53 loc) · 1.77 KB
/
run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# Import required packages.
import logging
import logging.config
import utils.logging_handlers
import time
import os
import rich
import yaml
from interface import main_window
# Define constants.
LOGGING_LEVEL = "INFO" # Choices are: "DEBUG", "INFO", "WARN", "CRITICAL", "ERROR"
def setup_logger(level) -> logging.Logger:
"""
Sets up the built-in python logger with the appropriate handlers and formatting.
Parameters:
-----------
level - The level/depth at which information is logged.
Returns:
--------
Logger - The logger object to interface with.
"""
# Create log output directory.
os.makedirs("logs", exist_ok=True)
# Load config file.
log_config = yaml.safe_load(open("logging_config.yaml", "r", encoding="utf-8").read())
logging.config.dictConfig(log_config)
# Loop through the configured handlers in the yaml file and set their level.
for handler in logging.getLogger().handlers:
# Check if handler is an actual text console one.
if isinstance(handler, type(rich.logging.RichHandler())):
handler.setLevel(level)
return logging.getLogger()
def main() -> None:
"""
Main program method.
"""
try:
# Initialize logger.
logger = setup_logger(LOGGING_LEVEL)
# Start UI.
interface = main_window.MainUI()
interface.initialize_window()
# Main loop that runs as long as the main window is open.
while interface.get_is_window_open():
# Update window.
interface.update_window()
# Sleep
time.sleep(1/120)
except Exception as exception:
logger.critical("MAIN THREAD CRASH:", exc_info=exception, stack_info=True)
if __name__ == "__main__":
# Call main function.
main()