120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
import argparse
|
|
import datetime
|
|
import time
|
|
|
|
import botconfig
|
|
import src.settings as var
|
|
|
|
# Segue to logger, since src.gamemodes requires it
|
|
# TODO: throw this into a logger.py perhaps so we aren't breaking up imports with non-import stuff
|
|
def logger(file, write=True, display=True):
|
|
if file is not None:
|
|
open(file, "a").close() # create the file if it doesn't exist
|
|
def log(*output, write=write, display=display):
|
|
output = " ".join([str(x) for x in output]).replace("\u0002", "").replace("\\x02", "") # remove bold
|
|
if botconfig.DEBUG_MODE:
|
|
write = True
|
|
if botconfig.DEBUG_MODE or botconfig.VERBOSE_MODE:
|
|
display = True
|
|
timestamp = get_timestamp()
|
|
if display:
|
|
print(timestamp + output, file=utf8stdout)
|
|
if write and file is not None:
|
|
with open(file, "a", errors="replace") as f:
|
|
f.seek(0, 2)
|
|
f.write(timestamp + output + "\n")
|
|
|
|
return log
|
|
|
|
stream_handler = logger(None)
|
|
debuglog = logger("debug.log", write=False, display=False)
|
|
errlog = logger("errors.log")
|
|
plog = logger(None) # use this instead of print so that logs have timestamps
|
|
|
|
# Import the user-defined game modes
|
|
# These are not required, so failing to import it doesn't matter
|
|
# The file then imports our game modes
|
|
# Fall back to importing our game modes if theirs fail
|
|
|
|
try:
|
|
import gamemodes
|
|
except ImportError:
|
|
import src.gamemodes
|
|
|
|
# Handle launch parameters
|
|
|
|
# Argument --debug means start in debug mode
|
|
# --verbose means to print a lot of stuff (when not in debug mode)
|
|
# --normal means to override the above and use nothing
|
|
# Settings can be defined in the config, but launch argumentss override it
|
|
|
|
debug_mode = False
|
|
verbose = False
|
|
normal = False
|
|
|
|
# Carry over settings from botconfig into settings.py
|
|
|
|
for setting, value in botconfig.__dict__.items():
|
|
if not setting.isupper():
|
|
continue # Not a setting
|
|
if setting == "DEBUG_MODE":
|
|
debug_mode = value
|
|
if setting == "VERBOSE_MODE":
|
|
verbose = value
|
|
if setting == "NORMAL_MODE":
|
|
normal = value
|
|
if not setting in var.__dict__.keys():
|
|
continue # Don't carry over config-only settings
|
|
|
|
# If we got that far, it's valid
|
|
setattr(var, setting, value)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--debug', action='store_true')
|
|
parser.add_argument('--verbose', action='store_true')
|
|
parser.add_argument('--normal', action='store_true')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.debug: debug_mode = True
|
|
if args.verbose: verbose = True
|
|
if args.normal: normal = True
|
|
|
|
botconfig.DEBUG_MODE = debug_mode if not normal else False
|
|
botconfig.VERBOSE_MODE = verbose if not normal else False
|
|
|
|
# Initialize Database
|
|
|
|
var.init_db()
|
|
|
|
# Logger
|
|
|
|
# replace characters that can't be encoded with '?'
|
|
# since windows likes to use weird encodings by default
|
|
utf8stdout = open(1, 'w', errors="replace", closefd=False) # stdout
|
|
|
|
def get_timestamp(use_utc=None, ts_format=None):
|
|
"""Return a timestamp with timezone + offset from UTC."""
|
|
if use_utc is None:
|
|
use_utc = botconfig.USE_UTC
|
|
if ts_format is None:
|
|
ts_format = botconfig.TIMESTAMP_FORMAT
|
|
if use_utc:
|
|
tmf = datetime.datetime.utcnow().strftime(ts_format)
|
|
tz = "UTC"
|
|
offset = "+0000"
|
|
else:
|
|
tmf = time.strftime(ts_format)
|
|
tz = time.tzname[0]
|
|
offset = "+"
|
|
if datetime.datetime.utcnow().hour > datetime.datetime.now().hour:
|
|
offset = "-"
|
|
offset += str(time.timezone // 36).zfill(4)
|
|
return tmf.format(tzname=tz, tzoffset=offset).strip().upper() + " "
|
|
|
|
def stream(output, level="normal"):
|
|
if botconfig.VERBOSE_MODE or botconfig.DEBUG_MODE:
|
|
stream_handler(output)
|
|
elif level == "warning":
|
|
stream_handler(output)
|