Skip to content

Instantly share code, notes, and snippets.

@heyibrahimkhan
Last active November 14, 2021 11:12
Show Gist options
  • Save heyibrahimkhan/6a6ff61c49eab55dc1764aac5167d7a9 to your computer and use it in GitHub Desktop.
Save heyibrahimkhan/6a6ff61c49eab55dc1764aac5167d7a9 to your computer and use it in GitHub Desktop.
Add some additional functionality to the script. Inclduing logging and arparse
import logging
import colorlog
def create_log_file(log_file_name):
with open(log_file_name, 'w') as o: pass
def setup_logger(log_fmt="%(log_color)s%(asctime)s:%(levelname)s:%(message)s", log_file_name=".output.log", level='DEBUG'):
# a new log file is created each time.
# no space issues are caused.
create_log_file(log_file_name)
formatter = colorlog.ColoredFormatter(
log_fmt,
datefmt='%DT%H:%M:%SZ'
)
logger = logging.getLogger()
handler2 = logging.FileHandler(log_file_name)
handler = logging.StreamHandler()
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.addHandler(handler2)
logger.setLevel(level)
return logger
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment