94 lines
2.7 KiB
Python
Executable File
94 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
twitter.py - Phenny Twitter Module
|
|
Copyright 2012, Sean B. Palmer, inamidst.com
|
|
Modified by sfan5 2012
|
|
Licensed under the Eiffel Forum License 2.
|
|
|
|
http://inamidst.com/phenny/
|
|
"""
|
|
|
|
import re, time
|
|
import web
|
|
|
|
r_username = re.compile(r'^@[a-zA-Z0-9_]{1,15}$')
|
|
r_link = re.compile(r'^https?://twitter.com/\S+$')
|
|
r_p = re.compile(r'(?ims)(<p class="js-tweet-text.*?</p>)')
|
|
r_tag = re.compile(r'(?ims)<[^>]+>')
|
|
r_anchor = re.compile(r'(?ims)(<a.*?</a>)')
|
|
r_expanded = re.compile(r'(?ims)data-expanded-url=["\'](.*?)["\']')
|
|
r_whiteline = re.compile(r'(?ims)[ \t]+[\r\n]+')
|
|
r_breaks = re.compile(r'(?ims)[\r\n]+')
|
|
|
|
def entity(*args, **kargs):
|
|
return web.entity(*args, **kargs)
|
|
|
|
def decode(html):
|
|
return web.r_entity.sub(entity, html)
|
|
|
|
def expand(tweet):
|
|
def replacement(match):
|
|
anchor = match.group(1)
|
|
for link in r_expanded.findall(anchor):
|
|
return link
|
|
return r_tag.sub('', anchor)
|
|
return r_anchor.sub(replacement, tweet)
|
|
|
|
def read_tweet(url):
|
|
bytes, sc = web.get(url)
|
|
bytes = str(bytes, 'utf-8')
|
|
shim = '<div class="content clearfix">'
|
|
if shim in bytes:
|
|
bytes = bytes.split(shim, 1).pop()
|
|
|
|
for text in r_p.findall(bytes):
|
|
text = expand(text)
|
|
text = r_tag.sub('', text)
|
|
text = text.strip()
|
|
text = r_whiteline.sub(' ', text)
|
|
text = r_breaks.sub(' ', text)
|
|
return decode(text)
|
|
return "Sorry, couldn't get a tweet from %s" % url
|
|
|
|
def format(tweet, username):
|
|
return '%s (@%s)' % (tweet, username)
|
|
|
|
def user_tweet(username):
|
|
tweet = read_tweet('https://twitter.com/' + username + "?" + str(time.time()))
|
|
return format(tweet, username)
|
|
|
|
def id_tweet(tid):
|
|
link = 'https://twitter.com/twitter/status/' + tid
|
|
headers, status = web.head(link)
|
|
if status == 301:
|
|
if not "Location" in headers:
|
|
return "Sorry, couldn't get a tweet from %s" % link
|
|
url = headers["Location"]
|
|
username = url.split('/')[3]
|
|
tweet = read_tweet(url)
|
|
return format(tweet, username)
|
|
return "Sorry, couldn't get a tweet from %s" % link
|
|
|
|
def twitter(phenny, input):
|
|
arg = input.group(2)
|
|
if not arg:
|
|
return phenny.reply("Give me a link, a @username, or a tweet id")
|
|
|
|
arg = arg.strip()
|
|
log.log("event", "%s queried Twitter for '%s'" % (log.fmt_user(input), arg), phenny)
|
|
if arg.isdigit():
|
|
phenny.say(id_tweet(arg))
|
|
elif r_username.match(arg):
|
|
phenny.say(user_tweet(arg[1:]))
|
|
elif r_link.match(arg):
|
|
username = arg.split('/')[3]
|
|
tweet = read_tweet(arg)
|
|
phenny.say(format(tweet, username))
|
|
else: phenny.reply("Give me a link, a @username, or a tweet id")
|
|
|
|
twitter.commands = ['tw', 'twitter']
|
|
twitter.thread = True
|
|
|
|
if __name__ == '__main__':
|
|
print(__doc__)
|