import string, cPickle, random, urllib, sys, time, re, os, twitter, subprocess, datetime, urlparse, hashlib
from collections import defaultdict
from irclib import irc_lower, nm_to_n
+import json
try:
from blame_filter import bfd
(bot.revision.split()[1], bot.channel, conn.get_nickname(),
bot.owner, len(karma.keys())))
+class FishPond:
+ def __init__(fishpond):
+ fishpond.last=[]
+ fishpond.DoS=0
+ fishpond.quotatime=0
+
+ def note_last(fishpond, msg, cfg):
+ fishpond.last.insert(0,(msg,cfg))
+ fishpond.last = fishpond.last[0:10]
+
# Check on fish stocks
def fish_quota(pond):
if pond.DoS:
return
me = bot.connection.get_nickname()
trout_msg = random.choice(fishlist)
- fishpond.last=trout_msg
+ fishpond.note_last(trout_msg,cfg)
# The bot won't trout or flirt with itself;
if irc_lower(me) == irc_lower(target) or irc_lower(target) in synonyms:
target = nick
return
me = bot.connection.get_nickname()
slash_msg = random.choice(fishlist)
- fishpond.last=slash_msg
+ fishpond.note_last(slash_msg,cfg)
# The bot won't slash people with themselves
if irc_lower(who[0]) == irc_lower(who[1]):
conn.notice(nick, "oooooh no missus!")
sans=__getcommits(sdb,sdbk,what)
return tans+fans+sans
-def blameq(bot,cmd,nick,conn,public,fish,tdb,tdbk,fdb,fdbk,sdb,sdbk):
+def blameq(bot,cmd,nick,conn,public,fishpond,cfgs):
+ tdb,tdbk,x = cfgs[0][7] # urgh, magic, to support magic knowledge below
+ fdb,fdbk,x = cfgs[1][7]
+ sdb,sdbk,x = cfgs[2][7]
clist=cmd.split()
if len(clist) < 2:
bot.automsg(public,nick,"Who or what do you want to blame?")
return
cwhat=' '.join(clist[2:])
+ kindsfile = "fish?"
if clist[1]=="#last":
- ans=__getall(tdb,tdbk,fdb,fdbk,sdb,sdbk,fish.last)
+ try:
+ n = abs(int(clist[2]))-1
+ if n < 0: raise ValueError
+ except IndexError: n = 0
+ except ValueError:
+ bot.automsg(public,nick,"Huh?")
+ return
+ try: lmsg, lcfg = fishpond.last[n]
+ except IndexError:
+ bot.automsg(public,nick,"Nothing")
+ return
+ xdb,xdbk,kindsfile = lcfg[7]
+ ans=__getcommits(xdb,xdbk,lmsg)
elif clist[1]=="#trouts" or clist[1]=="#trout":
ans=__getcommits(tdb,tdbk,cwhat)
elif clist[1]=="#flirts" or clist[1]=="#flirt":
if len(ans[0])==1:
bot.automsg(public,nick,ans[0])
else:
- bot.automsg(public,nick,"Modified %s: %s" % (ans[0][2].isoformat(),ans[0][1]))
+ bot.automsg(public,nick,"Modified %s %s: %s" % (kindsfile, ans[0][2].isoformat(),ans[0][1]))
elif len(ans)>4:
bot.automsg(public,nick,"I found %d matches, which is too many. Please be more specific!" % (len(ans)) )
else:
if len(a)==1:
bot.automsg(public,nick,a)
else:
- bot.automsg(public,nick,"'%s' modified on %s: %s" % (a[0],a[2].isoformat(),a[1]))
+ bot.automsg(public,nick,"%s '%s' modified on %s: %s" % (kindsfile, a[0],a[2].isoformat(),a[1]))
### say to msg/channel
def sayq(bot, cmd, nick, conn, public):
n=time.localtime(time.time())
s="%02d:%02d" % (self.localfirst.tm_hour,self.localfirst.tm_min)
if n.tm_yday != self.localfirst.tm_yday:
- s+=time.strftime(" on %d %B", n)
+ s+=time.strftime(" on %d %B", self.localfirst)
return s
def urltype(self):
z=min(len(urlinfos)-1, self.count-1)
unobfuscate_urls=True
expand_included_tweets=True
stringsout=[]
-
- parts = string.split(urlstring,'/')
- tweetID = parts[-1]
+
+ path = urlparse.urlparse(urlstring).path
+ tweetID = path.split('/')[-1]
try:
status = twitapi.GetStatus(tweetID)
if status == {}:
tweeter_name = tweeter_name + " RTing " + status.user.name #.encode('UTF-8', 'replace')
tweetText = status.full_text
if status.media:
- replacements = defaultdict( list )
+ replacements = defaultdict(list)
+
for medium in status.media:
replacements[medium.url].append(medium.media_url_https)
- for k,v in replacements.items():
+ # The twitter-api 'conveniently' parses this for you and
+ # throws away the actual video URLs, so we have to take the
+ # JSON and reparse it :sadpanda:
+ # This is particularly annoying because we don't know
+ # for sure that status.media and the JSON 'media' entry
+ # have the same elements in the same order. Probably they
+ # do but maybe twitter-api randomly reorganised things or
+ # filtered the list or something. So instead we go through
+ # the JSON and handle the media urls, discarding whatever
+ # unfortunate thing we have put in replacements already.
+ parsed_tweet = json.loads(status.AsJsonString())
+ for medium in parsed_tweet.get('media', []):
+ if medium['type'] == 'video':
+ best = { 'bitrate': -1 }
+ for vt in medium['video_info']['variants']:
+ if (vt.get('content_type') == 'video/mp4' and
+ vt.get('bitrate', -1) > best['bitrate']):
+ best = vt
+ if 'url' in best:
+ video_url = best['url'].split('?',1)[0]
+ duration = medium['video_info']['duration_millis']
+ # ^ duration_millis is a string
+ duration = "%.1f" % (float(duration)/1000.)
+ video_desc = "%s (%ss)" % (video_url, duration)
+ replacements[medium['url']] = [video_desc]
- v = [re.sub(r"/tweet_video_thumb/([\w\-]+).jpg", r"/tweet_video/\1.mp4", link) for link in v]
+ for k,v in replacements.items():
if len(v) > 1:
replacementstring = "[" + " ; ".join(v) +"]"
else:
~matthewv / irc.git / blobdiff