1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#!/usr/bin/env python
"""
twitter.py - Phenny Twitter Module
Copyright 2012, Sean B. Palmer, inamidst.com
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=["\'](.*?)["\']')
def entity(*args, **kargs):
return web.entity(*args, **kargs).encode('utf-8')
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 = web.get(url)
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 = text.replace('\r', '')
text = text.replace('\n', '')
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
data = web.head(link)
message, status = tuple(data)
if status == 301:
url = message.get("Location")
if not url: return "Sorry, couldn't get a tweet from %s" % link
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).strip()
if isinstance(arg, unicode):
arg = arg.encode('utf-8')
if arg.isdigit():
phenny.say(id_tweet(arg))
elif r_username.match(arg):
phenny.say(user_tweet(arg))
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__
|
