archangelic/pinhook
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add console and file logging

Browse Source
This commit is contained in:
Mallory Hancock 2018-01-05 14:03:32 -08:00
parent 12bd72a689
commit e377527aa6
1 changed files with 27 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pinhook/bot.py
View File

@ -1,4 +1,5 @@
import imp import imp
import logging
import os import os
import ssl import ssl
import time import time
@ -30,6 +31,7 @@ class Message:
class Bot(irc.bot.SingleServerIRCBot): class Bot(irc.bot.SingleServerIRCBot):
def __init__(self, channels, nickname, server, **kwargs): def __init__(self, channels, nickname, server, **kwargs):
self.set_kwargs(**kwargs) self.set_kwargs(**kwargs)
self.start_logging(self.log_level)
if self.ssl_required: if self.ssl_required:
factory = irc.connection.Factory(wrapper=ssl.wrap_socket) factory = irc.connection.Factory(wrapper=ssl.wrap_socket)
irc.bot.SingleServerIRCBot.__init__(self, [(server, self.port)], nickname, nickname, connect_factory=factory) irc.bot.SingleServerIRCBot.__init__(self, [(server, self.port)], nickname, nickname, connect_factory=factory)
@ -47,6 +49,7 @@ class Bot(irc.bot.SingleServerIRCBot):
'ssl_required': False, 'ssl_required': False,
'ns_pass': None, 'ns_pass': None,
'nickserv': 'NickServ', 'nickserv': 'NickServ',
'log_level': 'info',
} }
for k, v in kwargs.items(): for k, v in kwargs.items():
setattr(self, k, v) setattr(self, k, v)
@ -54,6 +57,29 @@ class Bot(irc.bot.SingleServerIRCBot):
if a not in kwargs: if a not in kwargs:
setattr(self, a, kwarguments[a]) setattr(self, a, kwarguments[a])
def start_logging(self, level):
if level == 'error':
level = logging.ERROR
elif level == 'warning':
level = logging.WARNING
elif level == 'info':
level = logging.INFO
elif level == 'debug':
level = logging.DEBUG
self.logger = logging.getLogger('{}'.format(self.bot_nick))
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(module)s - %(message)s')
# Set console logger
ch = logging.StreamHandler()
ch.setLevel(level)
ch.setFormatter(formatter)
# Set file logger
fh = logging.FileHandler('{}.log'.format(self.bot_nick))
fh.setLevel(level)
fh.setFormatter(formatter)
# Add handlers
self.logger.addHandler(ch)
self.logger.addHandler(fh)
def load_plugins(self): def load_plugins(self):
# clear plugin list to ensure no old plugins remain # clear plugin list to ensure no old plugins remain
pinhook.plugin.clear_plugins() pinhook.plugin.clear_plugins()
@ -148,7 +174,7 @@ class Bot(irc.bot.SingleServerIRCBot):
if output: if output:
self.process_output(c, chan, output) self.process_output(c, chan, output)
except Exception as e: except Exception as e:
print(e) self.logger.error(e)
def process_output(self, c, chan, output): def process_output(self, c, chan, output):
for msg in output.msg: for msg in output.msg: