Source code for pumapy.utilities.logger

import datetime
import os


[docs]def print_warning(warning_text):
    print(Colors.WARNING + warning_text + Colors.ENDC)


[docs]class Colors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'


[docs]class Logger:

    def __init__(self):

        self.log = ""
        self.location = ""

        if not os.path.exists('logs'):
            os.mkdir('logs')

        now = datetime.datetime.now()
        month = str(now.month) if len(str(now.month)) == 2 else "0" + str(now.month)
        day = str(now.day) if len(str(now.day)) == 2 else "0" + str(now.day)
        hour = str(now.hour) if len(str(now.hour)) == 2 else "0" + str(now.hour)
        minute = str(now.minute) if len(str(now.minute)) == 2 else "0" + str(now.minute)
        second = str(now.second) if len(str(now.second)) == 2 else "0" + str(now.second)
        self.location = "logs/pumapy_log_" + str(now.year) + month + day + "_" + hour + minute + second + ".txt"

[docs]    def set_location(self, location):
        self.location = location

[docs]    def log_section(self, name):
        self.log += "\n---------------------------- \n " + str(name) + "\n---------------------------- \n"

[docs]    def log_line(self, val):
        self.log += str(val) + "\n"

[docs]    def log_item(self, val):
        self.log += str(val) + " "

[docs]    def new_line(self):
        self.log += "\n"

[docs]    def log_bool(self, var_name, val):
        if val:
            self.log_line(str(var_name) + ": True")
        else:
            self.log_line(str(var_name) + ": False")

[docs]    def log_value(self, var_name, val):
        self.log_line(str(var_name) + ": " + str(val))

[docs]    def write_log(self):
        if self.location == "":
            return
        txt_file = open(self.location, "w")
        txt_file.write(self.log)
        txt_file.close()