bss/dr.botzo
1
0
Fork 0
Code Issues 3 Pull Requests Releases 2 Wiki Activity

regex search in FactFile

Browse Source
This commit is contained in:
Brian S. Stephan 2010-07-30 07:22:29 -05:00
parent 6e86096f2b
commit cb70310616
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TODO
View File

@ -19,4 +19,3 @@ dr.botzo --- TODO
* obligatory info command * obligatory info command
* similar to FactFile module, a generic Trigger module * similar to FactFile module, a generic Trigger module
* do splitting in DrBotServerConnection.privmsg * do splitting in DrBotServerConnection.privmsg
* regex search in FactFile

18
modules/FactFile.py
View File

@ -17,6 +17,7 @@
from ConfigParser import NoOptionError from ConfigParser import NoOptionError
import os import os
import random import random
import re
from irclib import irclib from irclib import irclib
@ -47,11 +48,20 @@ class FactFile(Module):
if os.path.isfile(filename): if os.path.isfile(filename):
# http://www.regexprn.com/2008/11/read-random-line-in-large-file-in.html # http://www.regexprn.com/2008/11/read-random-line-in-large-file-in.html
with open(filename, 'r') as file: with open(filename, 'r') as file:
facts = file.readlines() facts = []
to_print = facts[random.randint(1, len(facts))-1] if len(whats) == 1:
facts = file.readlines()
else:
lines = file.readlines()
for line in lines:
# check if line matches provided regex
if re.search(' '.join(whats[1:]), line, re.IGNORECASE) is not None:
facts.append(line)
if len(facts) > 0:
to_print = facts[random.randint(1, len(facts))-1]
# return text # return text
return self.reply(connection, replypath, to_print.rstrip()) return self.reply(connection, replypath, to_print.rstrip())
else: else:
print('filename in config file for \'' + whats[0] + '\' is wrong') print('filename in config file for \'' + whats[0] + '\' is wrong')