import os
import logging


def setup_logger(CURRENT_PATH, CURRENT_TIME):
    print(
        f"Logging to {os.path.join(CURRENT_PATH, 'out', str(CURRENT_TIME), 'logs', 'log.txt')}")
    logging.basicConfig(filename=os.path.join(CURRENT_PATH, "out", str(CURRENT_TIME), "logs", "log.txt"), filemode='w',
                        format='%(asctime)s - %(levelname)s - %(message)s',
                        level=logging.INFO)


def create_output_dir(CURRENT_PATH, CURRENT_TIME):
    os.mkdir(os.path.join(CURRENT_PATH,
             "out", str(CURRENT_TIME)))
    os.mkdir(os.path.join(CURRENT_PATH, "out",
             str(CURRENT_TIME), "logs"))
    os.mkdir(os.path.join(CURRENT_PATH, "out",
                          str(CURRENT_TIME), "models"))