mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
130 lines
4.3 KiB
Python
130 lines
4.3 KiB
Python
# coding=utf8
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
"""
|
|
This is the SpiceBot Logs system.
|
|
|
|
This Class stores logs in an easy to access manner
|
|
"""
|
|
|
|
import threading
|
|
import os
|
|
import sys
|
|
import sopel_modules.spicemanip as spicemanip
|
|
|
|
from .Config import config as botconfig
|
|
|
|
|
|
# TODO timestamps
|
|
|
|
class BotLogs():
|
|
"""This Is a contained source of log information"""
|
|
def __init__(self):
|
|
self.setup_logs()
|
|
self.lock = threading.Lock()
|
|
self.dict = {
|
|
"list": {"Sopel_systemd": [], "Sopel_stdio": []},
|
|
"queue": []
|
|
}
|
|
|
|
def setup_logs(self):
|
|
botconfig.core.logs_stdio = os.path.join(botconfig.core.logdir, 'stdio.log')
|
|
# botconfig.core.logs_stdio = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.stdio.log')
|
|
botconfig.core.logs_exceptions = os.path.join(botconfig.core.logdir, 'exceptions.log')
|
|
# botconfig.core.logs_exceptions = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.exceptions.log')
|
|
botconfig.core.logs_raw = os.path.join(botconfig.core.logdir, 'raw.log')
|
|
# botconfig.core.logs_raw = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.raw.log')
|
|
|
|
def botstderr(self, logmessage):
|
|
sys.stderr.write(logmessage)
|
|
|
|
def log(self, logtype, logentry, stdio=False):
|
|
|
|
self.lock.acquire()
|
|
|
|
logtitle = "[" + logtype + "]"
|
|
logmessage = logtitle + " " + logentry
|
|
|
|
if botconfig.logging_channel:
|
|
self.dict["queue"].append(logmessage)
|
|
|
|
if stdio:
|
|
self.botstderr(logmessage + "\n")
|
|
|
|
if logtype not in list(self.dict["list"].keys()):
|
|
self.dict["list"][logtype] = []
|
|
self.dict["list"][logtype].append(logentry)
|
|
|
|
self.lock.release()
|
|
|
|
def get_logs(self, logtype):
|
|
if logtype == "Sopel_systemd":
|
|
logindex = self.systemd_logs_fetch()
|
|
elif logtype == "Sopel_stdio":
|
|
logindex = self.stdio_logs_fetch()
|
|
else:
|
|
logindex = self.dict["list"][logtype]
|
|
return logindex
|
|
|
|
def stdio_logs_fetch(self):
|
|
|
|
stdio_ignore = []
|
|
for logtype in list(self.dict["list"].keys()):
|
|
stdio_ignore.append("[" + logtype + "]")
|
|
|
|
try:
|
|
log_file_lines = []
|
|
log_file = open(botconfig.logs_stdio, 'r')
|
|
lines = log_file.readlines()
|
|
for line in lines:
|
|
log_file_lines.append(line)
|
|
log_file.close()
|
|
|
|
currentline, linenumindex = 0, []
|
|
for line in log_file_lines:
|
|
if line.startswith("Welcome to Sopel. Loading modules..."):
|
|
linenumindex.append(currentline)
|
|
currentline += 1
|
|
last_start = max(linenumindex)
|
|
filelines = log_file_lines[last_start:]
|
|
except Exception as e:
|
|
debuglines = e
|
|
filelines = []
|
|
|
|
debuglines = []
|
|
loadedmodules = []
|
|
for line in filelines:
|
|
if line.startswith("Loaded:"):
|
|
loadedmodules.append(str(line).split("Loaded:")[-1])
|
|
else:
|
|
if not line.startswith(tuple(stdio_ignore)) and not line.isspace():
|
|
debuglines.append(str(line))
|
|
loadedmodules = "Loaded: " + spicemanip.main(loadedmodules, 'andlist')
|
|
debuglines.insert(1, loadedmodules)
|
|
|
|
return debuglines
|
|
|
|
def systemd_logs_fetch(self):
|
|
servicepid = self.get_running_pid()
|
|
debuglines = []
|
|
for line in os.popen(str("sudo journalctl _PID=" + str(servicepid))).read().split('\n'):
|
|
line = str(line).split(str(os.uname()[1] + " "))[-1]
|
|
lineparts = str(line).split(": ")
|
|
del lineparts[0]
|
|
line = spicemanip.main(lineparts, 0)
|
|
if not line.isspace():
|
|
debuglines.append(str(line))
|
|
return debuglines
|
|
|
|
def get_running_pid(self):
|
|
try:
|
|
filename = "/run/sopel/sopel-" + str(botconfig.nick) + ".pid"
|
|
with open(filename, 'r') as pid_file:
|
|
pidnum = int(pid_file.read())
|
|
except Exception as e:
|
|
pidnum = e
|
|
pidnum = str(os.popen("systemctl show " + str(botconfig.nick) + " --property=MainPID").read()).split("=")[-1]
|
|
return pidnum
|
|
|
|
|
|
logs = BotLogs()
|