easyeditor/util/globals.py

from pathlib import Path

import logging
import os

import yaml


def get_handler(path, log_name):
    log_file_path = os.path.join(path, log_name)
    try:
        if not os.path.exists(path):
            print("We are creating the logger files")
            os.makedirs(path)
    except:
        pass
    file_handler = logging.FileHandler(log_file_path)
    file_handler.setLevel(logging.DEBUG)
    file_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))

    stream_handler = logging.StreamHandler()
    stream_handler.setLevel(logging.DEBUG)
    stream_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
    return file_handler, stream_handler


# def get_run_dir(dir_name):
#
#     alg_dir = RESULTS_DIR / dir_name
#     if alg_dir.exists():
#         id_list = [
#             int(str(x).split("_")[-1])
#             for x in alg_dir.iterdir()
#             if str(x).split("_")[-1].isnumeric()
#         ]
#         run_id = 0 if not id_list else max(id_list) + 1
#     else:
#         run_id = 0
#     run_dir = RESULTS_DIR / dir_name / f"run_{str(run_id).zfill(3)}"
#     run_dir.mkdir(parents=True, exist_ok=True)
#     print(f"Results will be stored at {run_dir}")
#
#     return run_dir