2 import string, cPickle, random, urllib, sys, time, re, os, twitter, subprocess, datetime
3 from irclib import irc_lower, nm_to_n
6 def karmaq(bot, cmd, nick, conn, public, karma):
8 item=cmd.split()[1].lower()
12 bot.automsg(public,nick,"I have karma on %s items." %
14 elif karma.has_key(item):
15 bot.automsg(public,nick,"%s has karma %s."
18 bot.automsg(public,nick, "%s has no karma set." % item)
21 def karmadelq(bot, cmd, nick, conn, public, karma):
23 item=cmd.split()[1].lower()
25 conn.notice(nick, "What should I delete?")
28 conn.notice(nick, "You are not my owner.")
30 if karma.has_key(item):
32 conn.notice(nick, "Item %s deleted."%item)
34 conn.notice(nick, "There is no karma stored for %s."%item)
36 # help - provides the URL of the help file
37 def helpq(bot, cmd, nick, conn, public):
38 bot.automsg(public,nick,
39 "For help see http://www.chiark.greenend.org.uk/~matthewv/irc/servus.html")
43 def infoq(bot, cmd, nick, conn, public, karma):
44 bot.automsg(public,nick,
45 ("I am Acrobat %s, on %s, as nick %s. "+
46 "My owner is %s; I have karma on %s items.") %
47 (bot.revision.split()[1], bot.channel, conn.get_nickname(),
48 bot.owner, len(karma.keys())))
50 # Check on fish stocks
53 if time.time()>=pond.quotatime:
57 if (time.time()-pond.quotatime)>pond.fish_time_inc:
58 pond.cur_fish+=(((time.time()-pond.quotatime)
59 /pond.fish_time_inc)*pond.fish_inc)
60 if pond.cur_fish>pond.max_fish:
61 pond.cur_fish=pond.max_fish
62 pond.quotatime=time.time()
64 # List of things the bot might be called to work round the self-trouting code
65 synonyms=["itself","the bot","themself"]
67 # trout someone, or flirt with them
68 def troutq(bot, cmd, nick, conn, public, cfg):
75 selftroutchance=cfg[6]
79 conn.notice(nick, quietmsg%fishpond.Boring_Git)
81 if fishpond.cur_fish<=0:
82 conn.notice(nick, nofishmsg)
84 target = string.join(cmd.split()[1:])
86 conn.notice(nick, notargetmsg)
88 me = bot.connection.get_nickname()
89 trout_msg = random.choice(fishlist)
90 fishpond.last=trout_msg
91 # The bot won't trout or flirt with itself;
92 if irc_lower(me) == irc_lower(target) or irc_lower(target) in synonyms:
94 # There's a chance the game may be given away if the request was not
97 if random.random()<=selftroutchance:
98 trout_msg=trout_msg+(selftrout%nick)
100 conn.action(bot.channel, trout_msg % target)
104 def slashq(bot, cmd, nick, conn, public, cfg):
111 selfslashchance=cfg[6]
115 conn.notice(nick, quietmsg%fishpond.Boring_Git)
117 if fishpond.cur_fish<=0:
118 conn.notice(nick, nofishmsg)
120 target = string.join(cmd.split()[1:])
121 #who = cmd.split()[1:]
122 who = ' '.join(cmd.split()[1:]).split(' / ')
124 conn.notice(nick, "it takes two to tango!")
127 conn.notice(nick, "we'll have none of that round here")
129 me = bot.connection.get_nickname()
130 slash_msg = random.choice(fishlist)
131 fishpond.last=slash_msg
132 # The bot won't slash people with themselves
133 if irc_lower(who[0]) == irc_lower(who[1]):
134 conn.notice(nick, "oooooh no missus!")
136 # The bot won't slash with itself, instead slashing the requester
138 if irc_lower(me) == irc_lower(who[n]) or irc_lower(who[n]) in synonyms:
140 # Perhaps someone asked to slash themselves with the bot then we get
141 if irc_lower(who[0]) == irc_lower(who[1]):
142 conn.notice(nick, "you wish!")
144 # There's a chance the game may be given away if the request was not
147 if random.random()<=selfslashchance:
148 slash_msg=slash_msg+(selfslash%nick)
150 conn.action(bot.channel, slash_msg % (who[0], who[1]))
154 def unitq(bot, cmd, nick, conn, public):
155 args = ' '.join(cmd.split()[1:]).split(' as ')
157 args = ' '.join(cmd.split()[1:]).split(' / ')
159 conn.notice(nick, "syntax: units arg1 as arg2")
162 sin,sout=os.popen4(["units","--verbose","--",args[0]],"r")
164 sin,sout=os.popen4(["units","--verbose","--",args[0],args[1]],"r")
167 #popen2 doesn't clean up the child properly. Do this by hand
169 if os.WEXITSTATUS(child[1])==0:
170 bot.automsg(public,nick,res[0].strip())
172 conn.notice(nick,'; '.join(map(lambda x: x.strip(),res)))
174 # Shut up trouting for a minute
175 def nofishq(bot, cmd, nick, conn, public, fish):
179 fish.quotatime=time.time()
180 fish.quotatime+=fish.nofish_time
181 conn.notice(nick, "Fish stocks depleted, as you wish.")
184 def reloadq(bot, cmd, nick, conn, public):
185 if not public and irc_lower(nick) == irc_lower(bot.owner):
188 conn.notice(nick, "Config reloaded.")
190 conn.notice(nick, "Config reloading failed!")
192 bot.automsg(public,nick,
193 "Configuration can only be reloaded by my owner, by /msg.")
196 def quitq(bot, cmd, nick, conn, public):
197 if irc_lower(nick) == irc_lower(bot.owner):
198 bot.die(msg = "I have been chosen!")
200 conn.notice(nick, "Such aggression in public!")
202 conn.notice(nick, "You're not my owner.")
204 # google for something
205 def googleq(bot, cmd, nick, conn, public):
206 cmdrest = string.join(cmd.split()[1:])
207 # "I'm Feeling Lucky" rather than try and parse the html
208 targ = ("http://www.google.com/search?q=%s&btnI=I'm+Feeling+Lucky"
209 % urllib.quote_plus(cmdrest))
211 # get redirected and grab the resulting url for returning
212 gsearch = urllib.urlopen(targ).geturl()
213 if gsearch != targ: # we've found something
214 bot.automsg(public,nick,str(gsearch))
215 else: # we haven't found anything.
216 bot.automsg(public,nick,"No pages found.")
217 except IOError: # if the connection times out. This blocks. :(
218 bot.automsg(public,nick,"The web's broken. Waah!")
220 # Look up the definition of something using google
221 def defineq(bot, cmd, nick, conn, public):
222 #this doesn't work any more
223 bot.automsg(public,nick,"'define' is broken because google are bastards :(")
225 cmdrest = string.join(cmd.split()[1:])
226 targ = ("http://www.google.co.uk/search?q=define%%3A%s&ie=utf-8&oe=utf-8"
227 % urllib.quote_plus(cmdrest))
229 # Just slurp everything into a string
230 defnpage = urllib.urlopen(targ).read()
231 # For definitions we really do have to parse the HTML, sadly.
232 # This is of course going to be a bit fragile. We first look for
233 # 'Definitions of %s on the Web' -- if this isn't present we
234 # assume we have the 'no definitions found page'.
235 # The first defn starts after the following <p> tag, but as the
236 # first <li> in a <ul type="disc" class=std>
237 # Following that we assume that each definition is all the non-markup
238 # before a <br> tag. Currently we just dump out the first definition.
239 match = re.search(r"Definitions of <b>.*?</b> on the Web.*?<li>\s*([^>]*)((<br>)|(<li>))",defnpage,re.MULTILINE)
241 bot.automsg(public,nick,"Some things defy definition.")
243 # We assume google has truncated the definition for us so this
244 # won't flood the channel with text...
245 defn = " ".join(match.group(1).split("\n"))
246 bot.automsg(public,nick,defn)
247 except IOError: # if the connection times out. This blocks. :(
248 bot.automsg(public,nick,"The web's broken. Waah!")
250 # Look up a currency conversion via xe.com
251 def currencyq(bot, cmd, nick, conn, public):
252 args = ' '.join(cmd.split()[1:]).split(' as ')
253 if len(args) != 2 or len(args[0]) != 3 or len(args[1]) != 3:
254 conn.notice(nick, "syntax: currency arg1 as arg2")
256 targ = ("http://www.xe.com/ucc/convert.cgi?From=%s&To=%s" % (args[0], args[1]))
258 currencypage = urllib.urlopen(targ).read()
259 match = re.search(r"(1 %s = [\d\.]+ %s)" % (args[0],args[1]),currencypage,re.MULTILINE)
261 bot.automsg(public,nick,"Dear Chief Secretary, there is no money.")
263 conversion = match.group(1);
264 conversion = conversion.replace(' ',' ');
265 bot.automsg(public,nick,conversion + " (from xe.com)")
266 except IOError: # if the connection times out. This blocks. :(
267 bot.automsg(public,nick,"The web's broken. Waah!")
270 ### extract the commit message and timestamp for commit
271 def __getcommitinfo(commit):
272 cmd=["git","log","-n","1","--pretty=format:%ct|%s",commit]
273 x=subprocess.Popen(cmd,
274 stdout=subprocess.PIPE,stderr=subprocess.PIPE)
275 out,err=x.communicate()
280 ts,mes=out.split('|')
281 when=datetime.date.fromtimestamp(float(ts))
282 return mes.strip(), when
284 ###Return an array of commit messages and timestamps for lines in db that match what
285 def __getcommits(db,keys,what):
289 ret=__getcommitinfo(db[k])
290 if len(ret)==1: #error message
291 return ["Error message from git blame: %s" % ret]
293 ans.append( (k,ret[0],ret[1]) )
296 ###search all three databases for what
297 def __getall(tdb,tdbk,fdb,fdbk,sdb,sdbk,what):
300 tans=__getcommits(tdb,tdbk,what)
301 fans=__getcommits(fdb,fdbk,what)
302 sans=__getcommits(sdb,sdbk,what)
303 return tans+fans+sans
305 def blameq(bot,cmd,nick,conn,public,fish,tdb,tdbk,fdb,fdbk,sdb,sdbk):
308 bot.automsg(public,nick,"Who or what do you want to blame?")
310 cwhat=' '.join(clist[2:])
311 if clist[1]=="#last":
312 ans=__getall(tdb,tdbk,fdb,fdbk,sdb,sdbk,fish.last)
313 elif clist[1]=="#trouts" or clist[1]=="#trout":
314 ans=__getcommits(tdb,tdbk,cwhat)
315 elif clist[1]=="#flirts" or clist[1]=="#flirt":
316 ans=__getcommits(fdb,fdbk,cwhat)
317 elif clist[1]=="#slashes" or clist[1]=="#slash":
318 ans=__getcommits(sdb,sdbk,cwhat)
320 cwhat=' '.join(clist[1:])
321 ans=__getall(tdb,tdbk,fdb,fdbk,sdb,sdbk,cwhat)
323 bot.automsg(public,nick,"No match found")
326 bot.automsg(public,nick,ans[0])
328 bot.automsg(public,nick,"Modified %s: %s" % (ans[0][2].isoformat(),ans[0][1]))
330 bot.automsg(public,nick,"I found %d matches, which is too many. Please be more specific!" % (len(ans)) )
334 bot.automsg(public,nick,a)
336 bot.automsg(public,nick,"'%s' modified on %s: %s" % (a[0],a[2].isoformat(),a[1]))
338 ### say to msg/channel
339 def sayq(bot, cmd, nick, conn, public):
340 if irc_lower(nick) == irc_lower(bot.owner):
341 conn.privmsg(bot.channel, string.join(cmd.split()[1:]))
344 conn.notice(nick, "You're not my owner!")
346 ### action to msg/channel
347 def doq(bot, cmd, nick, conn, public):
348 sys.stderr.write(irc_lower(bot.owner))
349 sys.stderr.write(irc_lower(nick))
351 if irc_lower(nick) == irc_lower(bot.owner):
352 conn.action(bot.channel, string.join(cmd.split()[1:]))
354 conn.notice(nick, "You're not my owner!")
357 def disconnq(bot, cmd, nick, conn, public):
358 if cmd == "disconnect": # hop off for 60s
359 bot.disconnect(msg="Be right back.")
361 ### list keys of a dictionary
362 def listkeysq(bot, cmd, nick, conn, public, dict, sort=False):
366 bot.automsg(public,nick,string.join(d))
368 ### rot13 text (yes, I could have typed out the letters....)
369 ### also "foo".encode('rot13') would have worked
370 def rot13q(bot, cmd, nick, conn, public):
371 a=''.join(map(chr,range((ord('a')),(ord('z')+1))))
373 trans=string.maketrans(a+a.upper(),b+b.upper())
374 conn.notice(nick, string.join(cmd.split()[1:]).translate(trans))
376 ### URL-tracking stuff
378 ### return a easy-to-read approximation of a time period
379 def nicetime(tempus):
381 tm="%d seconds ago"%int(tempus)
383 tm="%d minutes ago"%int(tempus/60)
385 tm="%d hours ago"%int(tempus/3600)
388 ### class to store URL data
390 "contains meta-data about a URL seen on-channel"
391 def __init__(self,url,nick):
394 self.first=time.time()
396 self.lastseen=time.time()
397 self.lastasked=time.time()
398 def recenttime(self):
399 return max(self.lastseen,self.lastasked)
401 return nicetime(time.time()-self.first)
403 z=min(len(urlcomplaints)-1, self.count-1)
404 return urlcomplaints[z]
406 #(?:) is a regexp that doesn't group
407 urlre = re.compile(r"((?:(?:http)|(?:nsfw))s?://[^ ]+)( |$)")
408 hturlre= re.compile(r"(http)(s?://[^ ]+)( |$)")
409 #matches \bre\:?\s+ before a regexp; (?i)==case insensitive match
410 shibboleth = re.compile(r"(?i)\bre\:?\s+((?:(?:http)|(?:nsfw))s?://[^ ]+)( |$)")
411 urlcomplaints = ["a contemporary","an interesting","a fascinating","an overused","a vastly overused"]
413 ### Deal with /msg bot url or ~url in channel
414 def urlq(bot, cmd, nick, conn, public,urldb):
415 if (not urlre.search(cmd)):
416 bot.automsg(False,nick,"Please use 'url' only with http, https, nsfw, or nsfws URLs")
419 urlstring=urlre.search(cmd).group(1)
420 url=canonical_url(urlstring)
423 complaint="That's %s URL that was first mentioned %s by %s" % \
424 (T.urltype(),T.firstmen(),T.nick)
426 complaint=complaint+". Furthermore it defeats the point of this command to use it other than via /msg."
428 bot.automsg(False,nick,complaint)
429 T.lastasked=time.time()
430 #URL suppressed, so mention in #urls
431 if urlstring != cmd.split()[1]: #first argument to URL was not the url
432 conn.privmsg("#urls","%s remarks: %s" % (nick," ".join(cmd.split()[1:])))
434 conn.privmsg("#urls","(via %s) %s"%(nick," ".join(cmd.split()[1:])))
437 bot.automsg(False,nick,"That URL was unique. There is little point in using !url out loud; please use it via /msg")
439 if urlstring != cmd.split()[1]: #first argument to URL was not the url
440 conn.privmsg(bot.channel,"%s remarks: %s" % (nick," ".join(cmd.split()[1:])))
442 conn.privmsg(bot.channel,"(via %s) %s"%(nick," ".join(cmd.split()[1:])))
443 bot.automsg(False,nick,"That URL was unique; I have posted it into IRC")
444 urldb[url]=UrlLog(url,nick)
446 ### Deal with URLs spotted in channel
447 def dourl(bot,conn,nick,command,urldb):
448 urlstring=urlre.search(command).group(1)
449 urlstring=canonical_url(urlstring)
451 if urlstring in urldb:
453 message="observes %s URL, first mentioned %s by %s" % \
454 (T.urltype(),T.firstmen(),T.nick)
455 if shibboleth.search(command)==None:
456 conn.action(bot.channel, message)
457 T.lastseen=time.time()
460 urldb[urlstring]=UrlLog(urlstring,nick)
463 def urlexpire(urldb,expire):
466 if time.time() - urldb[u].recenttime() > expire:
469 # canonicalise BBC URLs (internal use only)
470 def canonical_url(urlstring):
471 if "nsfw://" in urlstring or "nsfws://" in urlstring:
472 urlstring=urlstring.replace("nsfw","http",1)
473 if (urlstring.find("news.bbc.co.uk") != -1):
474 for middle in ("/low/","/mobile/"):
475 x = urlstring.find(middle)
477 urlstring.replace(middle,"/hi/")
480 # automatically make nsfw urls for you and pass them on to url
481 def nsfwq(bot,cmd,nick,conn,public,urldb):
482 if (not hturlre.search(cmd)):
483 bot.automsg(False,nick,"Please use 'nsfw' only with http or https URLs")
485 newcmd=hturlre.sub(nsfwify,cmd)
486 urlq(bot,newcmd,nick,conn,public,urldb)
493 def twitterq(bot,cmd,nick,conn,public,twitapi):
495 if (not urlre.search(cmd)):
496 bot.automsg(False,nick,"Please use 'twit' only with http URLs")
499 urlstring = urlre.search(cmd).group(1)
500 if (urlstring.find("twitter.com") !=-1):
501 stringout = getTweet(urlstring,twitapi)
502 bot.automsg(public, nick, stringout)
504 def getTweet(urlstring,twitapi):
505 parts = string.split(urlstring,'/')
508 status = twitapi.GetStatus(tweetID)
509 #print status, type(status), status=={}
511 return "twitapi.GetStatus returned nothing :-("
512 if status.user == None and status.text == None:
513 return "Empty status object returned :("
514 if status.user is not None:
515 tweeter_screen = status.user.screen_name.encode('UTF-8', 'replace')
516 tweeter_name = status.user.name.encode('UTF-8', 'replace')
518 tweeter_screen = "[not returned]" ; tweeter_name = "[not returned]"
519 tweetText = status.text.encode('UTF-8', 'replace')
520 tweetText = tweetText.replace('\n',' ')
521 stringout = "tweet by %s (%s): %s" %(tweeter_screen,tweeter_name,tweetText)
522 except twitter.TwitterError:
523 terror = sys.exc_info()
524 stringout = "Twitter error: %s" % terror[1].__str__()