Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 2,054 Bytes
4d16728 4130a97 4d16728 |
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 |
import os
import time
import logging
from dotenv import load_dotenv
load_dotenv()
class Logger:
def __init__(self, name: str, see_time: bool = False, console_log: bool = False, level: int = logging.INFO):
self.is_dev = os.getenv("ENVIRONMENT", "development") == "development"
self.logger = logging.getLogger(name)
self.logger.setLevel(level)
if not self.is_dev:
self.logger.addHandler(logging.NullHandler())
return
os.makedirs("./logs", exist_ok=True)
formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s" if see_time else "%(message)s,"
)
file_handler = logging.FileHandler(f"./logs/{name}.log")
file_handler.setFormatter(formatter)
self.logger.addHandler(file_handler)
if console_log:
console_handler = logging.StreamHandler()
console_handler.setFormatter(formatter)
self.logger.addHandler(console_handler)
def log_message(self, message: str, level: int = logging.INFO):
if not self.is_dev:
return
if level == logging.INFO:
self.logger.info(message)
elif level == logging.ERROR:
self.logger.error(message)
elif level == logging.WARNING:
self.logger.warning(message)
elif level == logging.DEBUG:
self.logger.debug(message)
else:
self.logger.info(message)
def disable_logging(self):
self.logger.disabled = True
def log_time(func):
def wrapper(*args, **kwargs):
if os.getenv("ENV", "development") != "development":
return func(*args, **kwargs)
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
logger = Logger(func.__name__ + "_time", see_time=True, level=logging.INFO)
logger.log_message(f"Function: {func.__name__}, Execution time: {round(end_time - start_time, 5)} seconds")
return result
return wrapper
|