# NEEDS
# adding a bold character for '<user> MEANT so say'
import sys
from logger import Logger
from event import Event
if sys.version_info > (3, 0, 0):
try:
from .basemodule import BaseModule
except (ImportError, SystemError):
from modules.basemodule import BaseModule
else:
try:
from basemodule import BaseModule
except (ImportError, SystemError):
from modules.basemodule import BaseModule
[docs]class Replace(BaseModule):
[docs] def post_init(self):
if not 'replace' in self.bot.mem_store:
self.bot.mem_store['replace'] = {}
replace = Event("__.r__")
replace.define(msg_definition=".*")
replace.subscribe(self)
self.bot.register_event(replace, self)
self.help = ".r <search string> | <replacement text> OR s/<search string>/<replacement string>"
self.MAX_BUFFER_SIZE = 300
self.MAX_HISTORY_SIZE = 10
[docs] def add_buffer(self, event=None, debug=False):
"""Takes a channel name and line passed to it and stores them in the bot's mem_store dict
for future access. The dict will have channel as key. The value to that key will be a list
of formatted lines of activity.
If the buffer size is not yet exceeded, lines are just added. If the buffer
is maxed out, the oldest line is removed and newest one inserted at the beginning.
"""
if debug:
print("Line: " + event.line)
print("Verb: " + event.verb)
print("Channel: " + event.channel)
print("")
if not event:
return
# there are certain things we want to record in history, like nick changes and quits
# these often add to the humor of a quote. however, these are not specific to a channel
# in IRC and our bot does not maintain a userlist per channel. Therefore, when nick
# changes and quits occur, we will add them to every buffer. This is not technically
# correct behavior and could very well lead to quits/nick changes that are not visible
# showing up in a quote, but it's the best we can do at the moment
if not event.channel:
# discard events with unwanted verbs
if event.verb not in ["QUIT", "NICK"]:
return
try:
for chan in list(self.bot.mem_store['replace'].keys()):
if len(self.bot.mem_store['replace']
[chan]) >= self.MAX_BUFFER_SIZE:
self.bot.mem_store['replace'][chan].pop()
line = self.format_line(event)
if line:
self.bot.mem_store['replace'][chan].insert(0, line)
except (KeyError, IndexError):
self.bot.logger.write(
Logger.WARNING,
line="couldn't find channel in modules.replace")
self.bot.logger.write(
Logger.WARNING,
line="Replace add_buffer() error when no event channel")
# now we continue with normal, per channel line addition
# create a dictionary associating the channel with an empty list if it doesn't exist yet
# END if not event.channel:
else:
if event.channel not in self.bot.mem_store['replace']:
self.bot.mem_store['replace'][event.channel] = []
try:
# check for the length of the buffer. if it's too long, pop the
# last item
if len(self.bot.mem_store['replace']
[event.channel]) >= self.MAX_BUFFER_SIZE:
self.bot.mem_store['replace'][event.channel].pop()
# get a line by passing event to format_line
# insert the line into the first position in the list
line = self.format_line(event)
if line:
self.bot.mem_store['replace'][event.channel].insert(
0, line)
except IndexError:
self.bot.logger.write(
Logger.WARNING,
line="Replace add_buffer() error. Couldn't access the list index.")
[docs] def get_replacement_message(self, channel=None, find_msg=''):
"""Looks through the mem_store to find the most recent message containing find_msg"""
if not channel:
self.bot.logger.write(Logger.WARNING, line="couldn't find channel")
return None
# must have at least one msg to search for and channel to look it up in
if len(find_msg) == 0 or not channel:
self.bot.logger.write(Logger.WARNING, line="find_msg is empty")
return None
# search for a matching string and saves the index of that entry.
# Searches from most recent to oldest.
found_index = -1
for index, line in enumerate(self.bot.mem_store['replace'][channel]):
message = line
msg_index = message.find(">")
message = message[msg_index:]
# if the current entry of mem_store contains our string, we set the
# index and then BREAK to stop looking
if sys.version_info < (3, 0, 0):
if find_msg.decode('utf-8', 'ignore') in message:
found_index = index
break
else:
if find_msg in message:
found_index = index
break
# check to see if index values are positive. if not, string was not
# found and we're done
if found_index == -1:
return None
# returns the entire line
submission = self.bot.mem_store['replace'][channel][found_index]
return submission
[docs] def handle(self, event):
# first we see if we're going to try a replace or just add a line to
# the mem_store
if event.msg.startswith(".r "):
# split the msg with '.r ' stripped off beginning and divide into a
# search string and replace string
string_token = event.msg[3:].split('|', 1)
find_msg = string_token[0].rstrip()
try:
# if there's nothing after the pipe, then this resolves to ''
# which is fine
replace_msg = string_token[1].lstrip()
except IndexError as e:
self.bot.logger.write(Logger.WARNING, line=e)
return
# looking for a message containing our search string
newString = self.get_replacement_message(event.channel, find_msg)
# because the mem_store line shows "<user> message", we have to split up the username and their message
# this actually works to our advantage so we dont have to do
# additional calls to find out who sent what
msg_index = newString.find(">")
message = newString[msg_index + 2:]
message = message.replace(find_msg, replace_msg)
user = newString[1:msg_index]
# pybot sends the new replacement message to the chat
self.say(event.channel, user + " MEANT to say: " + message)
# because both .r and s// are valid formats now
if event.msg.startswith("s/"):
self.bot.logger.write(
Logger.WARNING,
line="i am a log entry from within s/ in modules.replace")
# alternative notation: s/<substring to replace>/<substring to
# replace with>
string_token = event.msg[2:].split('/', 1)
find_msg = string_token[0]
try:
replace_msg = string_token[1]
except IndexError:
return
# looking for a message containing our search string
newString = self.get_replacement_message(event.channel, find_msg)
# because the mem_store line shows "<user> message", we have to split up the username and their message
# this actually works to our advantage so we dont have to do
# additional calls to find out who sent what
msg_index = newString.find(">")
message = newString[msg_index + 2:]
message = message.replace(find_msg, replace_msg)
user = newString[1:msg_index]
# pybot sends the new replacement message to the chat
self.say(event.channel, user + " MEANT to say: " + message)
if event.user != self.bot.NICK:
self.add_buffer(event)