.BANCHO. Add logHelper

helpers/generalFunctions.py

@@ -1,5 +1,6 @@
 """Some functions that don't fit in any other file"""
 from constants import mods
+from time import gmtime, strftime
 
 def stringToBool(s):
 	"""
@@ -55,3 +56,10 @@ def readableMods(__mods):
 
 def strContains(s, w):
 	return (' ' + w + ' ') in (' ' + s + ' ')
+
+def getTimestamp():
+	"""
+	Return current time in YYYY-MM-DD HH:MM:SS format.
+	Used in logs.
+	"""
+	return strftime("%Y-%m-%d %H:%M:%S", gmtime())

helpers/logHelper.py

@@ -0,0 +1,41 @@
+from constants import bcolors
+from helpers import discordBotHelper
+from helpers import generalFunctions
+
+def logMessage(message, alertType, messageColor, discord = False, alertDev = False, of = None):
+	if alertType == "INFO":
+		typeColor = bcolors.GREEN
+	elif alertType == "WARNING":
+		typeColor = bcolors.YELLOW
+	elif typeColor == "ERROR":
+		typeColor = bcolors.RED
+	else:
+		typeColor = bcolors.ENDC
+
+	finalMessage = "[{time}] {type} - {message}".format(time=generalFunctions.getTimestamp(), type=alertType, message=message)
+	finalMessageConsole = "{typeColor}[{time}] {type}{endc} - {messageColor}{message}{endc}".format(
+		time=generalFunctions.getTimestamp(),
+		type=alertType,
+		message=message,
+
+		typeColor=typeColor,
+		messageColor=messageColor,
+		endc=bcolors.ENDC)
+
+	# Always log to console
+	print(finalMessageConsole)
+
+	# Log to discord if needed
+	if discord == True:
+		discordBotHelper.sendConfidential(message, alertDev)
+
+	# Log to file if needed
+	if of != None:
+		with open(".data/{}".format(of), "a") as f:
+			f.write(finalMessage)
+
+def warning(message, discord = False, alertDev = False):
+	logMessage(message, "WARNING", bcolors.YELLOW, discord, alertDev, "warnings.txt")
+
+def error(message, discord = False, alertDev = True):
+	logMessage(message, "ERROR", bcolors.RED, discord, alertDev, "errors.txt")