Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Move formatted lines to loggable format to format_activity.py

Browse source
This commit is contained in:
jesopo 2019-03-05 09:19:06 +00:00
parent b43a5a7c74
commit db9744a0b5
2 changed files with 180 additions and 146 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

162
modules/format_activity.py Normal file
View file

@ -0,0 +1,162 @@
from src import EventManager, ModuleManager, utils
class Module(ModuleManager.BaseModule):
def _event(self, type, server, line, context):
self.events.on("formatted").on(type).call(server=server, line=line,
context=context)
def _mode_symbols(self, user, channel, server):
modes = channel.get_user_status(user)
symbols = []
modes = list(channel.get_user_status(user))
modes.sort(key=lambda x: list(server.prefix_modes.keys()).index(x))
for mode in modes:
symbols.append(server.prefix_modes[mode])
return "".join(symbols)
def _privmsg(self, event, channel, nickname):
symbols = ""
if channel:
symbols = self._mode_symbols(user, channel, event["server"])
if event["action"]:
return "* %s%s %s" % (symbols, nickname, event["message"])
else:
return "<%s%s> %s" % (symbols, nickname, event["message"])
@utils.hook("send.message.channel")
@utils.hook("received.message.channel")
def channel_message(self, event):
nickname = event["server"].nickname
if "user" in event:
nickname = event["user"].nickname
line = self._privmsg(event, event["channel"], nickname)
self._event("message.channel", event["server"], line,
event["channel"].name)
def _on_notice(self, event, sender, target):
line = "(notice->%s) <%s> %s" % (
target, sender, event["message"])
self._event("notice.channel", event["server"], line, None)
@utils.hook("received.notice.channel", priority=EventManager.PRIORITY_HIGH)
def channel_notice(self, event):
self._on_notice(event, event["user"].nickname, event["channel"].name)
@utils.hook("send.notice.channel")
def self_notice_channel(self, event):
self._on_notice(event, event["server"].nickname, event["channel"].name)
@utils.hook("received.notice.private", priority=EventManager.PRIORITY_HIGH)
def private_notice(self, event):
self._on_notice(event, event["user"].nickname, event["server"].nickname)
@utils.hook("send.notice.private")
def self_private_notice(self, event):
self._on_notice(event, event["server"].nickname, event["user"].nickname)
def _on_join(self, event, nickname):
line = "%s joined %s" % (nickname, event["channel"].name)
self._event("join", event["server"], line, event["channel"].name)
@utils.hook("received.join")
def join(self, event):
self._on_join(event, event["user"].nickname)
@utils.hook("self.join")
def self_join(self, event):
self._on_join(event, event["server"].nickname)
def _on_part(self, event, nickname):
reason = ""
if event["reason"]:
reason = " (%s)" % event["reason"]
line = "%s left %s%s" % (nickname, event["channel"].name, reason)
self._event("part", event["server"], line, event["channel"].name)
@utils.hook("received.part")
def part(self, event):
self._on_part(event, event["user"].nickname)
@utils.hook("self.part")
def self_part(self, event):
self._on_part(event, event["server"].nickname)
@utils.hook("received.nick")
@utils.hook("self.nick")
def on_nick(self, event):
line = "%s changed nickname to %s" % (
event["old_nickname"], event["new_nickname"])
self._event("nick", event["server"], line, None)
@utils.hook("received.server-notice", priority=EventManager.PRIORITY_HIGH)
def server_notice(self, event):
line = "(server notice) %s" % event["message"]
self._event("server-notice", event["server"], line, None)
@utils.hook("received.invite")
def invite(self, event):
line = "%s invited %s to %s" % (
event["user"].nickname, event["target_user"].nickname,
event["target_channel"])
self._event("invite", event["server"], line, event["target_channel"])
@utils.hook("received.mode.channel")
def mode(self, event):
args = " ".join(event["mode_args"])
if args:
args = " %s" % args
line = "%s set mode %s%s" % (
event["user"].nickname, "".join(event["modes"]), args)
self._event("mode.channel", event["server"], line,
event["channel"].name)
def _on_topic(self, event, setter, action, topic):
line = "topic %s by %s: %s" % (action, setter, topic)
self._event("topic", event["server"], line, event["channel"].name)
@utils.hook("received.topic")
def on_topic(self, event):
self._on_topic(event, event["user"].nickname, "changed",
event["topic"])
@utils.hook("received.333")
def on_333(self, event):
self._on_topic(event, event["setter"], "set",
event["channel"].topic, event["channel"])
unix_dt = datetime.datetime.utcfromtimestamp(event["set_at"])
dt = datetime.datetime.strftime(unix_dt, DATETIME_FORMAT)
line = "topic set at %s" % dt
self._event("topic-timestamp", event["server"], line,
event["channel"].name)
def _on_kick(self, event, nickname):
reason = ""
if event["reason"]:
reason = "(%s)" % event["reason"]
line = "%s kicked %s from %s%s" % (
event["user"].nickname, nickname, event["channel"].name, reason)
self._event("kick", event["server"], line, event["channel"].name)
@utils.hook("received.kick")
def kick(self, event):
self._on_kick(event, event["target_user"].nickname)
@utils.hook("self.kick")
def self_kick(self, event):
self._on_kick(event, event["server"].nickname)
def _quit(self, event, nickname, reason):
reason = ""
if reason:
reason = " (%s)" % reason
line = "%s quit%s" % (nickname, reason)
self._event("quit", event["server"], line, None)
@utils.hook("received.quit")
def on_quit(self, event):
self._quit(event, event["user"].nickname, event["reason"])
@utils.hook("send.quit")
def send_quit(self, event):
self._quit(event, event["server"].nickname, event["reason"])
@utils.hook("received.rename")
def rename(self, event):
line = "%s was renamed to %s" % (event["old_name"], event["new_name"])
self._event("rename", event["server"], line, event["old_name"])
@utils.hook("received.376")
def motd_end(self, event):
for line in event["server"].motd_lines:
line = "[MOTD] %s" % line
self._event("motd", event["server"], line, None)

164
modules/print_activity.py
View file

@ -4,151 +4,23 @@ from src import EventManager, ModuleManager, utils
DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ"
class Module(ModuleManager.BaseModule):
def print_line(self, event, line, channel=None):
timestamp = datetime.datetime.now().isoformat()
target = str(event["server"])
if not channel == None:
target += channel
def _print_line(self, target, context, line):
formatted_line = utils.irc.parse_format(line)
self.bot.log.info("%s | %s", [target, formatted_line])
self.bot.log.info("%s%s | %s", [target, context or "", formatted_line])
def _mode_symbols(self, user, channel, server):
modes = channel.get_user_status(user)
symbols = []
modes = list(channel.get_user_status(user))
modes.sort(key=lambda x: list(server.prefix_modes.keys()).index(x))
for mode in modes:
symbols.append(server.prefix_modes[mode])
return "".join(symbols)
def _on_message(self, event, user):
symbols = self._mode_symbols(user, event["channel"],
event["server"])
if event["action"]:
self.print_line(event, "* %s%s %s" % (symbols, user.nickname,
event["message"]), channel=event["channel"].name)
else:
self.print_line(event, "<%s%s> %s" % (symbols, user.nickname,
event["message"]), channel=event["channel"].name)
@utils.hook("received.message.channel",
priority=EventManager.PRIORITY_HIGH)
def channel_message(self, event):
self._on_message(event, event["user"])
@utils.hook("send.message.channel")
def self_channel_message(self, event):
self._on_message(event, event["server"].get_user(
event["server"].nickname))
def _on_notice(self, event, sender, target):
self.print_line(event, "(notice->%s) <%s> %s" % (
target, sender, event["message"]))
@utils.hook("received.notice.channel",
priority=EventManager.PRIORITY_HIGH)
def channel_notice(self, event):
self._on_notice(event, event["user"].nickname, event["channel"].name)
@utils.hook("send.notice.channel")
def self_notice_channel(self, event):
self._on_notice(event, event["server"].nickname, event["channel"].name)
@utils.hook("received.notice.private", priority=EventManager.PRIORITY_HIGH)
def private_notice(self, event):
self._on_notice(event, event["user"].nickname, event["server"].nickname)
@utils.hook("send.notice.private")
def self_private_notice(self, event):
self._on_notice(event, event["server"].nickname, event["user"].nickname)
@utils.hook("received.server-notice", priority=EventManager.PRIORITY_HIGH)
def server_notice(self, event):
self.print_line(event, "(server notice) %s" % event["message"])
def _on_join(self, event, nickname):
self.print_line(event, "%s joined %s" % (nickname,
event["channel"].name))
@utils.hook("received.join")
def join(self, event):
self._on_join(event, event["user"].nickname)
@utils.hook("self.join")
def self_join(self, event):
self._on_join(event, event["server"].nickname)
def _on_part(self, event, nickname):
self.print_line(event, "%s left %s%s" % (
nickname,
event["channel"].name,
"" if not event["reason"] else " (%s)" % event["reason"]))
@utils.hook("received.part")
def part(self, event):
self._on_part(event, event["user"].nickname)
@utils.hook("self.part")
def self_part(self, event):
self._on_part(event, event["server"].nickname)
@utils.hook("received.nick")
@utils.hook("self.nick")
def on_nick(self, event):
self.print_line(event, "%s changed nickname to %s" % (
event["old_nickname"], event["new_nickname"]))
def _quit(self, event, nickname, reason):
self.print_line(event, "%s quit%s" % (nickname,
"" if not reason else " (%s)" % reason))
@utils.hook("received.quit")
def on_quit(self, event):
self._quit(event, event["user"].nickname, event["reason"])
@utils.hook("send.quit")
def send_quit(self, event):
self._quit(event, event["server"].nickname, event["reason"])
def _on_kick(self, event, nickname):
self.print_line(event, "%s kicked %s from %s%s" % (
event["user"].nickname, nickname, event["channel"].name,
"" if not event["reason"] else " (%s)" % event["reason"]))
@utils.hook("received.kick")
def kick(self, event):
self._on_kick(event, event["target_user"].nickname)
@utils.hook("self.kick")
def self_kick(self, event):
self._on_kick(event, event["server"].nickname)
def _on_topic(self, event, setter, action, topic, channel):
self.print_line(event, "topic %s by %s: %s" % (action, setter,
topic), channel=channel.name)
@utils.hook("received.topic")
def on_topic(self, event):
self._on_topic(event, event["user"].nickname, "changed",
event["topic"], event["channel"])
@utils.hook("received.333")
def on_333(self, event):
self._on_topic(event, event["setter"], "set",
event["channel"].topic, event["channel"])
unix_dt = datetime.datetime.utcfromtimestamp(event["set_at"])
dt = datetime.datetime.strftime(unix_dt, DATETIME_FORMAT)
self.print_line(event, "topic set at %s" % dt,
channel=event["channel"].name)
@utils.hook("received.mode.channel")
def mode(self, event):
args = " ".join(event["mode_args"])
if args:
args = " %s" % args
self.print_line(event, "%s set mode %s%s" % (
event["user"].nickname, "".join(event["modes"]),
args), channel=event["channel"].name)
@utils.hook("received.rename")
def rename(self, event):
self.print_line(event, "%s was renamed to %s" % (
event["old_name"], event["new_name"]))
@utils.hook("received.376")
def motd_end(self, event):
for line in event["server"].motd_lines:
self.print_line(event, "[MOTD] %s" % line)
@utils.hook("received.invite")
def invite(self, event):
self.print_line(event, "%s invited %s to %s" % (
event["user"].nickname, event["target_user"].nickname,
event["target_channel"]))
@utils.hook("formatted.message.channel")
@utils.hook("formatted.notice.channel")
@utils.hook("formatted.join")
@utils.hook("formatted.part")
@utils.hook("formatted.nick")
@utils.hook("formatted.server-notice")
@utils.hook("formatted.invite")
@utils.hook("formatted.mode.channel")
@utils.hook("formatted.topic")
@utils.hook("formatted.topic-timestamp")
@utils.hook("formatted.kick")
@utils.hook("formatted.quit")
@utils.hook("formatted.rename")
@utils.hook("formatted.motd")
def formatted(self, event):
self._print_line(str(event["server"]), event["context"], event["line"])