class crowbar:
"""Class for viewing and posting status updates.
"""
def get_username(self):
"""Get the user login name"""
cachedir = self.get_cachedir()
names = os.listdir(cachedir)
names = [a[:-5] for a in names if a[-5:]=='.auth']
self.myname = dialog.select('Twitter username?','Login:'******'t42Eq9aoZWuOCQEvhrTe9A'
self.consumer_secret = 'xVrGA1o0njIwtTtuw0n4GAhmfvPpndMcnl6fabbDPI'
me = self.get_username()
# load cached authorization file
cachedir = self.get_cachedir()
cachefile = cachedir+me+'.auth'
if os.path.exists(cachefile):
with open(cachefile) as f:
data=f.read()
access_token = twitter.simplejson.loads(data)
return access_token
request_token_url = 'https://api.twitter.com/oauth/request_token'
access_token_url = 'https://api.twitter.com/oauth/access_token'
authorize_url = 'https://api.twitter.com/oauth/authorize'
consumer = oauth.Consumer(self.consumer_key, self.consumer_secret)
client = oauth.Client(consumer)
# Step 1: Get a request token. This is a temporary token that is used for
# having the user authorize an access token and to sign the request to obtain
# said access token.
resp, content = client.request(request_token_url, "GET")
if resp['status'] != '200':
raise Exception("Invalid response %s." % resp['status'])
request_token = dict(urlparse.parse_qsl(content))
# Step 2: Redirect to the provider. Since this is a CLI script we do not
url="%s?oauth_token=%s" % (authorize_url, request_token['oauth_token'])
webbrowser.open(url)
# After the user has granted access to you, the consumer, the provider will
# redirect you to whatever URL you have told them to redirect to. You can
# usually define this in the oauth_callback argument as well.
oauth_verifier = dialog.getText('What is the pin?',gtk.MESSAGE_QUESTION,'pin:',
'Fill out the form in the web page that pops up.\n'
'You will be given a PIN number.\n'
'Come back and enter that number here.').strip()
# Step 3: Once the consumer has redirected the user back to the oauth_callback
# URL you can request the access token the user has approved. You use the
# request token to sign this request. After this is done you throw away the
# request token and use the access token returned. You should store this
# access token somewhere safe, like a database, for future use.
token = oauth.Token(request_token['oauth_token'],
request_token['oauth_token_secret'])
token.set_verifier(oauth_verifier)
client = oauth.Client(consumer, token)
resp, content = client.request(access_token_url, "POST")
access_token = dict(urlparse.parse_qsl(content))
data=twitter.simplejson.dumps(access_token)
if '<error' not in data:
with open(cachefile,'w') as f:
f.write(data)
return access_token
def update_followers_count(self):
"""Updates the count of followers"""
self.me = api.GetUser(screen_name=self.myname)
fo=str(self.me.followers_count)
fr=str(self.me.friends_count)
self.window2.set_title(fr+" friends "+fo+" followers")
def get_cachedir(self):
"""Returns the application's Cache dir:"""
cachedir = os.path.expanduser('~/.crowbar_cache/')
if not os.path.exists(cachedir):
os.mkdir(cachedir)
return cachedir
def getshouts(self,s):
"""get names of people mentioned in s[].text"""
names = []
for a in s:
try:
atext = re.match("[^@]*(.*)",a.text).groups()[0]
for b in atext.split("@")[1:]:
al = re.match("(\w+)",b).groups()[0].lower()
if al not in names:
names.append(al)
except:
pass
return names
def get_screen_names(self,s):
"""get s[].user.screen_name from search results"""
names = []
for a in s:
al = a.user.screen_name.lower()
if al not in names:
names.append(al)
return names
def subem(self,sub):
"""sub everybody in sub"""
global friends
for a in sub:
al = a.lower()
if al not in friends:
try:
api.CreateFriendship(al)
friends.append(al)
except:
pass
def gtk_main_quit(self,widget,data = None):
gtk.main_quit()
return False # return False destroys window.
def delete_event(self,widget,data = None):
widget.hide() # hide the window
return True # return True does not destroy window.
def link_clicked(self,view,url,type_):
"""follow a url or reply link"""
url = url.replace('%26','&')
url = url.replace('&','&')
print("url-clicked %s" % url)
if 'http://' not in url:
if '@' in url:
self.entry1.set_text(url)
elif '#' == url[0]:
self.comboboxentry1.child.set_text(url)
self.search_clicked(None,self.comboboxentry1.child.get_text)
elif '☆' in url:
api.CreateFavorite(api.GetStatus(int(str(url).replace('☆',''))))
elif '☞' in url:
api.PostRetweet(int(str(url).replace('☞','')))
elif '☠' in url:
api.DestroyDirectMessage(int(str(url).replace('☠','')))
elif '✗' in url:
api.DestroyStatus(int(str(url).replace('✗','')))
else:
webbrowser.open(url)
def post_update(self,widget,data = None):
if data():
api.PostUpdate(data())
self.entry1.set_text('')
return True
def image_from_url(self,url,fname):
"""return a gtk.Image from url"""
#~ cache img to speed up loading
cachedir = self.get_cachedir()
if url:
ext = '.'+url[url.rindex('.')+1:].lower()
if ext not in ['.jpg','.png']:
ext = '.jpg'
fname = str(fname)+ext
cachefile = cachedir+fname
if os.path.exists(cachefile):
#~ cache hit
pb = gtk.gdk.pixbuf_new_from_file(cachefile)
pbs = pb.scale_simple(48,48,0)
return gtk.image_new_from_pixbuf(pbs)
fp = urllib.urlopen(url)
pbl = gtk.gdk.PixbufLoader()
data = fp.read()
#~ read url into pixbuf
if pbl.write(data):
pb = pbl.get_pixbuf()
try:
pbs = pb.scale_simple(48,48,0)
except:
try:
fp.close()
pbl.close()
except:
pass
print('could not scale image for %s '% fname)
return gtk.image_new_from_file('blank.jpg')
#~ pbs = pb
#~ create image from pixbuf
image = gtk.Image()
image.set_from_pixbuf(pbs)
#~ save cached copy
if ext != '.png':
pb.save(cachefile,"jpeg")
else:
pb.save(cachefile,"png")
try:
pbl.close()
except:
print('%s truncated or incomplete.' % url)
else:
#~ todo: make this a broken image
print('broken image for %s'%fname)
image = gtk.image_new_from_file('blank.jpg')
fp.close()
else:
print('no url for image%'%fname)
image = gtk.image_new_from_file('blank.jpg')
return image
def escape(self,s):
s = s.replace('&','&')
s = s.replace('<','<')
s = s.replace('>','>')
#~ s = s.replace('"','"')
return s
def unescape(self,s):
p = htmllib.HTMLParser(None)
p.save_bgn()
p.feed(str(s))
s = p.save_end()
#~ s = s.replace('<','<')
#~ s = s.replace('>','>')
s = s.replace(' rel="nofollow"','')
#~ s = s.replace('"e;','"')
s = s.replace('&','%26')
s = s.replace('[1]','')
return s
def process_urls(self,s):
s = self.escape(s)
s = re.sub("(http://[^ ]+)",
lambda m: "<a href='%s'>%s</a>" %
(m.group(1).replace('&','&'),
m.group(1).replace('&','&')),s)
s = re.sub("(@)(\w+)",
lambda m: "%s<a href='%s'>%s</a>" %
(m.group(1),'http://twitter.com/'+m.group(2),m.group(2)),s)
s = re.sub("(#)(\w+)",
lambda m: "%s<a href='%s'>%s</a>" %
(m.group(1),'#'+m.group(2),m.group(2)),s)
return s
def html_view(self,s,dm=False):
"""show search results and timelines in htmlview"""
# clear, add viewport & scrolling table
if self.sw1.get_child() is self.html:
self.sw1.remove(self.html)
self.sw1.set_property("hscrollbar-policy",gtk.POLICY_NEVER)
self.viewport1 = gtk.Viewport()
self.sw1.add(self.viewport1)
else:
self.viewport1.remove(self.table1)
self.table1 = gtk.Table(23,2,False)
self.table1.set_row_spacings(2)
self.table1.set_col_spacings(2)
self.viewport1.add(self.table1)
self.pic = []
self.msg = []
t = ['']
rows = 0
#display each user pic and instant message
for x in s:
if not dm:
user = x.GetUser()
else:
user = api.GetUser(x.sender_screen_name)
img = user.GetProfileImageUrl()
usn = str(user.GetScreenName())
t[0] = x
reply = usn
shouts = self.getshouts(t)
star = '☆'
if not dm:
if x.favorited:
star = '★'
if shouts:
reply+=' @'+string.join(self.getshouts(t),' @')
self.pic.append(self.image_from_url(img,usn))
text = self.process_urls(str(x.text))
#~ (re)construct html message
self.msg.append(HtmlTextView())
self.msg[rows].connect("url-clicked",self.link_clicked)
h = '<span>'
if not dm:
h+='<a title="Favorite" href="'+star+str(x.id)+'">'+star +'</a>'
h+='<span style="font-weight: bold">'+'<a href="http://twitter.com/'+usn+'">' +usn+'</a></span>: '+ text + '<br /><span style="font-size:small">'
if not dm:
h+=x.relative_created_at+' via '+self.unescape(x.source)+' | '
if dm:
h+='<a href="@'+reply+'">reply</a> | <a href="☠'+str(x.id)+'" title="Delete this.">Delete</a>'
elif (usn==self.me.screen_name):
h+='<a href="✗'+str(x.id)+'" title="Delete this tweet.">Delete</a>'
else:
h+='<a href="@'+reply+'">reply</a> | <a href="☞' +str(x.id)+'">retweet</a>'
h+='</span></span>'
try:
self.msg[rows].display_html(str(h))
except:
print('Error displaying:')
print(h+'\n')
self.table1.attach(self.pic[rows],0,1,rows,rows+1)
self.table1.attach(self.msg[rows],1,2,rows,rows+1)
rows+=1
#~ self.table1.attach(self.html,0,2,rows,rows+2)
self.blank = gtk.Label()
self.blank.set_property('height-request',100)
self.table1.attach(self.blank,0,2,rows,rows+2)
#~ self.table1.set_property("border-width", 5)
self.sw1.show_all()
def initialize_window(self):
"""Show the intro page"""
self.html = HtmlTextView()
self.html.connect("url-clicked",self.link_clicked)
self.html.display_html('<div><span style="color: red; text-decoration:underline">Welcome to</span><br/>\n'
' <img src="http://comptune.com/images/penguin5.gif" alt="penguin" height="48" width="48" /><br/>\n'
' <span style="font-size: 500%; font-family: serif">crowbar</span>\n'
'</div>\n')
self.sw1.add(self.html)
#~ add a search comboboxentry1 and 3 buttons
self.liststore1 = gtk.ListStore(str)
self.comboboxentry1 = gtk.ComboBoxEntry(self.liststore1, 0)
popular = ['#teamfollowback','#tfb','#f4f']
trending=api.GetTrendsCurrent()
trends = [x.name for x in trending]
for row in popular:
self.liststore1.append([row])
for row in trends:
self.liststore1.append([row])
# insert a horizontal box with combo and buttons
self.hbox2 = gtk.HBox(homogeneous = False)
self.hbox2.pack_start(self.comboboxentry1)
button = gtk.Button()
button.set_label('search')
button.set_property('width-request',55)
button.connect('clicked',self.search_clicked,self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button,expand = False,fill = True)
button = gtk.Button()
button.set_label('friends')
button.set_property('width-request',55)
button.connect('clicked',self.timeline_clicked,self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button,expand = False,fill = True)
button = gtk.Button()
button.set_label('user')
button.set_property('width-request',40)
button.connect('clicked',self.user_timeline_clicked,self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button,expand = False,fill = True)
button = gtk.Button()
button.set_label('dm')
button.set_property('width-request',35)
button.connect('clicked',self.dm_clicked,self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button,expand = False,fill = True)
button = gtk.Button()
button.set_label('mention')
button.set_property('width-request',70)
button.connect('clicked',self.mentions_clicked,self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button,expand = False,fill = True)
#~ testify.test(self)
self.vbox1.pack_start(self.hbox2,expand = False)
self.html.show_all()
self.hbox2.show_all()
def display_results(self,s,getnames,dm=False):
"""display results of searches and timelines"""
lnames = ''
if getnames:
snames = self.get_screen_names(s)
lnames = "@"+string.join(snames," @")
shouts = self.getshouts(s)
self.textview1.set_wrap_mode(gtk.WRAP_WORD)
buf = self.textview1.get_buffer()
buf.set_text(lnames+"@"+string.join(shouts," @"))
self.html_view(s,dm)
self.window2.show_all()
def get_list(self,widget):
it = widget.get_iter('0')
data = []
while 1:
try:
data.append(widget.get_value(it,0))
it = widget.iter_next(it)
except:
return data
def update_search(self,text):
l = self.get_list(self.liststore1)
if text and (text not in l):
self.liststore1.append([text])
def search_clicked(self,widget,method = None):
"""get names and shouts from api search and display them"""
text = method()
s = api.GetSearch(text)
self.display_results(s,True)
self.last_action = self.search_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def timeline_clicked(self,widget,method = None):
"""get friends' timeline"""
text = method()
getnames = False
if text:
getnames = True
if '#' in text:
getmames = False
text = None
s = api.GetUserTimeline(text)
else:
s = api.GetHomeTimeline()
self.display_results(s,getnames)
self.last_action = self.timeline_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def user_timeline_clicked(self,widget,method = None):
"""get user timeline"""
text = method()
getnames = False
s = api.GetUserTimeline(text)
self.display_results(s,getnames)
self.last_action = self.user_timeline_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def mentions_clicked(self,widget,method = None):
"""get mentions"""
s = api.GetMentions()
self.display_results(s,True)
self.last_action = self.mentions_clicked
self.last_method = method
self.update_followers_count()
def dm_clicked(self,widget,method = None):
"""get mentions"""
s = api.GetDirectMessages()
self.display_results(s,False,True)
self.last_action = self.mentions_clicked
self.last_method = method
self.update_followers_count()
def refresh_clicked(self,widget,method = None):
self.liststore1.clear()
popular = ['#teamfollowback','#tfb','#f4f']
trending=api.GetTrendsCurrent()
trends = [x.name for x in trending]
for row in popular:
self.liststore1.append([row])
for row in trends:
self.liststore1.append([row])
self.last_action(None,self.last_method)
def get_friendIDs(self):
"""Get friend and follower IDs"""
global friendIDs,followerIDs
if not globals().has_key('friendIDs'):
print('getting friends list...')
try:
friendIDs=api.GetFriendIDs()['ids']
except:
print("User not properly authenticated. Will not be able to post updates.")
return False
if not globals().has_key('followerIDs'):
print('getting list of followers...')
followerIDs=api.GetFollowerIDs()['ids']
return True
def follow_clicked(self,widget,data = None):
"""follow all poeple in textview1 by name"""
#~ iterate through the text buffer
buf = self.textview1.get_buffer()
(iter_first, iter_last) = buf.get_bounds()
text = buf.get_text(iter_first, iter_last)
#~ remove spaces and build a list to follow
text = text.replace(' ','')
fol = []
#~ create fol list
for x in text.split('@')[1:]:
if x not in fol:
fol.append(x)
#~ ask
if self.warn and (not dialog.ok("Follow %s people.\n"
"Are you sure?" % len(fol))):
return
else:
self.warn=False
# load nofollow list if it exists
me = self.me.screen_name
nofol = []
cachedir = self.get_cachedir()
nofollow = cachedir+me+'.nofollow'
if os.path.exists(nofollow):
with open(nofollow) as f:
nofol = [x.strip() for x in f.readlines()]
nofollen=len(nofol)
#~ friend everybody unless nofol, add to nofol
for a in fol:
if a not in nofol:
try:
api.CreateFriendship(a)
print('followed %s' % a)
except TwitterError as err:
print(err)
if 'follow-limits' in str(err):
dialog.alert("You have reached Twitter's follow limit!")
break
nofol.append(a)
#~ Write nofol to disk. These could be invalid names,
#~ existing friends, or blocked accounts.
#~ We don't care. We just don't follow them.
if nofollen < len(nofol):
with open(nofollow,'a') as f:
f.writelines([x+'\n' for x in nofol[nofollen:]])
def follow_followers(self,widget,data = None):
"""follow followers by id not by name"""
global friendIDs,followerIDs
if not self.get_friendIDs():
return
#~ safety feature in case twitter f's up
if len(friendIDs) < 100:
print('not enough friends')
return
#~ load protected and pending list
me = self.me.screen_name
cachedir = self.get_cachedir()
protfile = cachedir+me+'.prot'
newprot = []; prot=[]
if os.path.exists(protfile):
with open(protfile) as f:
prot = [int(x.strip()) for x in f.readlines()]
#~ follow everyone in followerIDs
for a in followerIDs:
if (a not in friendIDs) and (a not in prot):
try:
ret = api.CreateFriendship(a)
friendIDs.append(a)
print('followed %s' % ret.screen_name)
except TwitterError as err:
print(err)
if not 'follow-limits' in str(err):
newprot.append(a)
else:
dialog.alert("You have reached Twitter's follow limit!")
break
#~ those we can not follow are probably protected
with open(protfile,'a') as f:
f.writelines([str(x)+'\n' for x in newprot])
def unfollow_non_followers(self,widget,data = None):
"""unfollow everybody who is not following me
todo: add safelist"""
if not self.get_friendIDs():
return
me = self.me.screen_name
# load safelist
cachedir = self.get_cachedir()
safefile = cachedir+me+'.safe'
safe = []; unsub = []
if os.path.exists(safefile):
with open(safefile) as f:
safe = [x.strip() for x in f.readlines()]
#~ safety feature in case twitter f*s up
if len(followerIDs) < 100:
print('Need at least 100 followers to use this feature.')
return
for a in friendIDs:
if (a not in followerIDs) and (a not in safe):
unsub.append(a)
#~ unsub.remove(self.me.id)
#~ save nofollow list so we do not re-follow
#~ (unless they follow us, of course)
unsub_names=[]
if dialog.ok('Unfriend all '+str(len(unsub))+' non-followers?'):
#UNSUB EVERYBODY IN unsub
for a in unsub:
try:
u=api.DestroyFriendship(a)
print('unfriended %s' % u.screen_name)
unsub_names.append(u.screen_name)
except TwitterError as err:
print(err)
nofollow = cachedir+me+'.nofollow'
with open(nofollow,'a') as f:
f.writelines([x+'\n' for x in unsub_names])
def __init__(self):
"""main initialization routine"""
global api,friendIDs,friends
friends=[]
tok=self.authorize()
self.warn=True
api = twitter.Api(self.consumer_key,self.consumer_secret,
tok['oauth_token'],tok['oauth_token_secret'])
builder = gtk.Builder()
builder.add_from_file("crowbar.glade")
self.window1 = builder.get_object("window1")
self.vbox1 = builder.get_object("vbox1")
self.sw1 = builder.get_object("sw1")
self.window2 = builder.get_object("window2")
self.textview1 = builder.get_object("textview1")
self.entry1 = builder.get_object("entry1")
self.button1 = builder.get_object("button1")
builder.connect_signals(self)
self.window2.connect("delete-event",self.delete_event)
self.button1.connect("clicked",self.post_update,self.entry1.get_text)
self.update_followers_count()
self.window1.set_title("Crowbar @"+self.me.screen_name)
self.initialize_window()
#~ some variables to remember what we did last
self.last_action = None
self.last_method = None
class crowbar:
"""Class for viewing and posting status updates.
"""
def get_username(self):
"""Get the user login name"""
cachedir = self.get_cachedir()
names = os.listdir(cachedir)
names = [a[:-5] for a in names if a[-5:] == '.auth']
self.myname = dialog.select('Twitter username?', 'Login:'******'t42Eq9aoZWuOCQEvhrTe9A'
self.consumer_secret = 'xVrGA1o0njIwtTtuw0n4GAhmfvPpndMcnl6fabbDPI'
me = self.get_username()
# load cached authorization file
cachedir = self.get_cachedir()
cachefile = cachedir + me + '.auth'
if os.path.exists(cachefile):
with open(cachefile) as f:
data = f.read()
access_token = twitter.simplejson.loads(data)
return access_token
request_token_url = 'https://api.twitter.com/oauth/request_token'
access_token_url = 'https://api.twitter.com/oauth/access_token'
authorize_url = 'https://api.twitter.com/oauth/authorize'
consumer = oauth.Consumer(self.consumer_key, self.consumer_secret)
client = oauth.Client(consumer)
# Step 1: Get a request token. This is a temporary token that is used for
# having the user authorize an access token and to sign the request to obtain
# said access token.
resp, content = client.request(request_token_url, "GET")
if resp['status'] != '200':
raise Exception("Invalid response %s." % resp['status'])
request_token = dict(urlparse.parse_qsl(content))
# Step 2: Redirect to the provider. Since this is a CLI script we do not
url = "%s?oauth_token=%s" % (authorize_url,
request_token['oauth_token'])
webbrowser.open(url)
# After the user has granted access to you, the consumer, the provider will
# redirect you to whatever URL you have told them to redirect to. You can
# usually define this in the oauth_callback argument as well.
oauth_verifier = dialog.getText(
'What is the pin?', gtk.MESSAGE_QUESTION, 'pin:',
'Fill out the form in the web page that pops up.\n'
'You will be given a PIN number.\n'
'Come back and enter that number here.').strip()
# Step 3: Once the consumer has redirected the user back to the oauth_callback
# URL you can request the access token the user has approved. You use the
# request token to sign this request. After this is done you throw away the
# request token and use the access token returned. You should store this
# access token somewhere safe, like a database, for future use.
token = oauth.Token(request_token['oauth_token'],
request_token['oauth_token_secret'])
token.set_verifier(oauth_verifier)
client = oauth.Client(consumer, token)
resp, content = client.request(access_token_url, "POST")
access_token = dict(urlparse.parse_qsl(content))
data = twitter.simplejson.dumps(access_token)
if '<error' not in data:
with open(cachefile, 'w') as f:
f.write(data)
return access_token
def update_followers_count(self):
"""Updates the count of followers"""
self.me = api.GetUser(screen_name=self.myname)
fo = str(self.me.followers_count)
fr = str(self.me.friends_count)
self.window2.set_title(fr + " friends " + fo + " followers")
def get_cachedir(self):
"""Returns the application's Cache dir:"""
cachedir = os.path.expanduser('~/.crowbar_cache/')
if not os.path.exists(cachedir):
os.mkdir(cachedir)
return cachedir
def getshouts(self, s):
"""get names of people mentioned in s[].text"""
names = []
for a in s:
try:
atext = re.match("[^@]*(.*)", a.text).groups()[0]
for b in atext.split("@")[1:]:
al = re.match("(\w+)", b).groups()[0].lower()
if al not in names:
names.append(al)
except:
pass
return names
def get_screen_names(self, s):
"""get s[].user.screen_name from search results"""
names = []
for a in s:
al = a.user.screen_name.lower()
if al not in names:
names.append(al)
return names
def subem(self, sub):
"""sub everybody in sub"""
global friends
for a in sub:
al = a.lower()
if al not in friends:
try:
api.CreateFriendship(al)
friends.append(al)
except:
pass
def gtk_main_quit(self, widget, data=None):
gtk.main_quit()
return False # return False destroys window.
def delete_event(self, widget, data=None):
widget.hide() # hide the window
return True # return True does not destroy window.
def link_clicked(self, view, url, type_):
"""follow a url or reply link"""
url = url.replace('%26', '&')
url = url.replace('&', '&')
print("url-clicked %s" % url)
if 'http://' not in url:
if '@' in url:
self.entry1.set_text(url)
elif '#' == url[0]:
self.comboboxentry1.child.set_text(url)
self.search_clicked(None, self.comboboxentry1.child.get_text)
elif '☆' in url:
api.CreateFavorite(
api.GetStatus(int(str(url).replace('☆', ''))))
elif '☞' in url:
api.PostRetweet(int(str(url).replace('☞', '')))
elif '☠' in url:
api.DestroyDirectMessage(int(str(url).replace('☠', '')))
elif '✗' in url:
api.DestroyStatus(int(str(url).replace('✗', '')))
else:
webbrowser.open(url)
def post_update(self, widget, data=None):
if data():
api.PostUpdate(data())
self.entry1.set_text('')
return True
def image_from_url(self, url, fname):
"""return a gtk.Image from url"""
#~ cache img to speed up loading
cachedir = self.get_cachedir()
if url:
ext = '.' + url[url.rindex('.') + 1:].lower()
if ext not in ['.jpg', '.png']:
ext = '.jpg'
fname = str(fname) + ext
cachefile = cachedir + fname
if os.path.exists(cachefile):
#~ cache hit
pb = gtk.gdk.pixbuf_new_from_file(cachefile)
pbs = pb.scale_simple(48, 48, 0)
return gtk.image_new_from_pixbuf(pbs)
fp = urllib.urlopen(url)
pbl = gtk.gdk.PixbufLoader()
data = fp.read()
#~ read url into pixbuf
if pbl.write(data):
pb = pbl.get_pixbuf()
try:
pbs = pb.scale_simple(48, 48, 0)
except:
try:
fp.close()
pbl.close()
except:
pass
print('could not scale image for %s ' % fname)
return gtk.image_new_from_file('blank.jpg')
#~ pbs = pb
#~ create image from pixbuf
image = gtk.Image()
image.set_from_pixbuf(pbs)
#~ save cached copy
if ext != '.png':
pb.save(cachefile, "jpeg")
else:
pb.save(cachefile, "png")
try:
pbl.close()
except:
print('%s truncated or incomplete.' % url)
else:
#~ todo: make this a broken image
print('broken image for %s' % fname)
image = gtk.image_new_from_file('blank.jpg')
fp.close()
else:
print('no url for image%' % fname)
image = gtk.image_new_from_file('blank.jpg')
return image
def escape(self, s):
s = s.replace('&', '&')
s = s.replace('<', '<')
s = s.replace('>', '>')
#~ s = s.replace('"','"')
return s
def unescape(self, s):
p = htmllib.HTMLParser(None)
p.save_bgn()
p.feed(str(s))
s = p.save_end()
#~ s = s.replace('<','<')
#~ s = s.replace('>','>')
s = s.replace(' rel="nofollow"', '')
#~ s = s.replace('"e;','"')
s = s.replace('&', '%26')
s = s.replace('[1]', '')
return s
def process_urls(self, s):
s = self.escape(s)
s = re.sub(
"(http://[^ ]+)", lambda m: "<a href='%s'>%s</a>" %
(m.group(1).replace('&', '&'), m.group(1).replace(
'&', '&')), s)
s = re.sub(
"(@)(\w+)", lambda m: "%s<a href='%s'>%s</a>" %
(m.group(1), 'http://twitter.com/' + m.group(2), m.group(2)), s)
s = re.sub(
"(#)(\w+)", lambda m: "%s<a href='%s'>%s</a>" %
(m.group(1), '#' + m.group(2), m.group(2)), s)
return s
def html_view(self, s, dm=False):
"""show search results and timelines in htmlview"""
# clear, add viewport & scrolling table
if self.sw1.get_child() is self.html:
self.sw1.remove(self.html)
self.sw1.set_property("hscrollbar-policy", gtk.POLICY_NEVER)
self.viewport1 = gtk.Viewport()
self.sw1.add(self.viewport1)
else:
self.viewport1.remove(self.table1)
self.table1 = gtk.Table(23, 2, False)
self.table1.set_row_spacings(2)
self.table1.set_col_spacings(2)
self.viewport1.add(self.table1)
self.pic = []
self.msg = []
t = ['']
rows = 0
#display each user pic and instant message
for x in s:
if not dm:
user = x.GetUser()
else:
user = api.GetUser(x.sender_screen_name)
img = user.GetProfileImageUrl()
usn = str(user.GetScreenName())
t[0] = x
reply = usn
shouts = self.getshouts(t)
star = '☆'
if not dm:
if x.favorited:
star = '★'
if shouts:
reply += ' @' + string.join(self.getshouts(t), ' @')
self.pic.append(self.image_from_url(img, usn))
text = self.process_urls(str(x.text))
#~ (re)construct html message
self.msg.append(HtmlTextView())
self.msg[rows].connect("url-clicked", self.link_clicked)
h = '<span>'
if not dm:
h += '<a title="Favorite" href="' + star + str(
x.id) + '">' + star + '</a>'
h += '<span style="font-weight: bold">' + '<a href="http://twitter.com/' + usn + '">' + usn + '</a></span>: ' + text + '<br /><span style="font-size:small">'
if not dm:
h += x.relative_created_at + ' via ' + self.unescape(
x.source) + ' | '
if dm:
h += '<a href="@' + reply + '">reply</a> | <a href="☠' + str(
x.id) + '" title="Delete this.">Delete</a>'
elif (usn == self.me.screen_name):
h += '<a href="✗' + str(
x.id) + '" title="Delete this tweet.">Delete</a>'
else:
h += '<a href="@' + reply + '">reply</a> | <a href="☞' + str(
x.id) + '">retweet</a>'
h += '</span></span>'
try:
self.msg[rows].display_html(str(h))
except:
print('Error displaying:')
print(h + '\n')
self.table1.attach(self.pic[rows], 0, 1, rows, rows + 1)
self.table1.attach(self.msg[rows], 1, 2, rows, rows + 1)
rows += 1
#~ self.table1.attach(self.html,0,2,rows,rows+2)
self.blank = gtk.Label()
self.blank.set_property('height-request', 100)
self.table1.attach(self.blank, 0, 2, rows, rows + 2)
#~ self.table1.set_property("border-width", 5)
self.sw1.show_all()
def initialize_window(self):
"""Show the intro page"""
self.html = HtmlTextView()
self.html.connect("url-clicked", self.link_clicked)
self.html.display_html(
'<div><span style="color: red; text-decoration:underline">Welcome to</span><br/>\n'
' <img src="http://comptune.com/images/penguin5.gif" alt="penguin" height="48" width="48" /><br/>\n'
' <span style="font-size: 500%; font-family: serif">crowbar</span>\n'
'</div>\n')
self.sw1.add(self.html)
#~ add a search comboboxentry1 and 3 buttons
self.liststore1 = gtk.ListStore(str)
self.comboboxentry1 = gtk.ComboBoxEntry(self.liststore1, 0)
popular = ['#teamfollowback', '#tfb', '#f4f']
trending = api.GetTrendsCurrent()
trends = [x.name for x in trending]
for row in popular:
self.liststore1.append([row])
for row in trends:
self.liststore1.append([row])
# insert a horizontal box with combo and buttons
self.hbox2 = gtk.HBox(homogeneous=False)
self.hbox2.pack_start(self.comboboxentry1)
button = gtk.Button()
button.set_label('search')
button.set_property('width-request', 55)
button.connect('clicked', self.search_clicked,
self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button, expand=False, fill=True)
button = gtk.Button()
button.set_label('friends')
button.set_property('width-request', 55)
button.connect('clicked', self.timeline_clicked,
self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button, expand=False, fill=True)
button = gtk.Button()
button.set_label('user')
button.set_property('width-request', 40)
button.connect('clicked', self.user_timeline_clicked,
self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button, expand=False, fill=True)
button = gtk.Button()
button.set_label('dm')
button.set_property('width-request', 35)
button.connect('clicked', self.dm_clicked,
self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button, expand=False, fill=True)
button = gtk.Button()
button.set_label('mention')
button.set_property('width-request', 70)
button.connect('clicked', self.mentions_clicked,
self.comboboxentry1.child.get_text)
self.hbox2.pack_start(button, expand=False, fill=True)
#~ testify.test(self)
self.vbox1.pack_start(self.hbox2, expand=False)
self.html.show_all()
self.hbox2.show_all()
def display_results(self, s, getnames, dm=False):
"""display results of searches and timelines"""
lnames = ''
if getnames:
snames = self.get_screen_names(s)
lnames = "@" + string.join(snames, " @")
shouts = self.getshouts(s)
self.textview1.set_wrap_mode(gtk.WRAP_WORD)
buf = self.textview1.get_buffer()
buf.set_text(lnames + "@" + string.join(shouts, " @"))
self.html_view(s, dm)
self.window2.show_all()
def get_list(self, widget):
it = widget.get_iter('0')
data = []
while 1:
try:
data.append(widget.get_value(it, 0))
it = widget.iter_next(it)
except:
return data
def update_search(self, text):
l = self.get_list(self.liststore1)
if text and (text not in l):
self.liststore1.append([text])
def search_clicked(self, widget, method=None):
"""get names and shouts from api search and display them"""
text = method()
s = api.GetSearch(text)
self.display_results(s, True)
self.last_action = self.search_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def timeline_clicked(self, widget, method=None):
"""get friends' timeline"""
text = method()
getnames = False
if text:
getnames = True
if '#' in text:
getmames = False
text = None
s = api.GetUserTimeline(text)
else:
s = api.GetHomeTimeline()
self.display_results(s, getnames)
self.last_action = self.timeline_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def user_timeline_clicked(self, widget, method=None):
"""get user timeline"""
text = method()
getnames = False
s = api.GetUserTimeline(text)
self.display_results(s, getnames)
self.last_action = self.user_timeline_clicked
self.last_method = method
self.update_search(text)
self.update_followers_count()
def mentions_clicked(self, widget, method=None):
"""get mentions"""
s = api.GetMentions()
self.display_results(s, True)
self.last_action = self.mentions_clicked
self.last_method = method
self.update_followers_count()
def dm_clicked(self, widget, method=None):
"""get mentions"""
s = api.GetDirectMessages()
self.display_results(s, False, True)
self.last_action = self.mentions_clicked
self.last_method = method
self.update_followers_count()
def refresh_clicked(self, widget, method=None):
self.liststore1.clear()
popular = ['#teamfollowback', '#tfb', '#f4f']
trending = api.GetTrendsCurrent()
trends = [x.name for x in trending]
for row in popular:
self.liststore1.append([row])
for row in trends:
self.liststore1.append([row])
self.last_action(None, self.last_method)
def get_friendIDs(self):
"""Get friend and follower IDs"""
global friendIDs, followerIDs
if not globals().has_key('friendIDs'):
print('getting friends list...')
try:
friendIDs = api.GetFriendIDs()['ids']
except:
print(
"User not properly authenticated. Will not be able to post updates."
)
return False
if not globals().has_key('followerIDs'):
print('getting list of followers...')
followerIDs = api.GetFollowerIDs()['ids']
return True
def follow_clicked(self, widget, data=None):
"""follow all poeple in textview1 by name"""
#~ iterate through the text buffer
buf = self.textview1.get_buffer()
(iter_first, iter_last) = buf.get_bounds()
text = buf.get_text(iter_first, iter_last)
#~ remove spaces and build a list to follow
text = text.replace(' ', '')
fol = []
#~ create fol list
for x in text.split('@')[1:]:
if x not in fol:
fol.append(x)
#~ ask
if self.warn and (not dialog.ok("Follow %s people.\n"
"Are you sure?" % len(fol))):
return
else:
self.warn = False
# load nofollow list if it exists
me = self.me.screen_name
nofol = []
cachedir = self.get_cachedir()
nofollow = cachedir + me + '.nofollow'
if os.path.exists(nofollow):
with open(nofollow) as f:
nofol = [x.strip() for x in f.readlines()]
nofollen = len(nofol)
#~ friend everybody unless nofol, add to nofol
for a in fol:
if a not in nofol:
try:
api.CreateFriendship(a)
print('followed %s' % a)
except TwitterError as err:
print(err)
if 'follow-limits' in str(err):
dialog.alert(
"You have reached Twitter's follow limit!")
break
nofol.append(a)
#~ Write nofol to disk. These could be invalid names,
#~ existing friends, or blocked accounts.
#~ We don't care. We just don't follow them.
if nofollen < len(nofol):
with open(nofollow, 'a') as f:
f.writelines([x + '\n' for x in nofol[nofollen:]])
def follow_followers(self, widget, data=None):
"""follow followers by id not by name"""
global friendIDs, followerIDs
if not self.get_friendIDs():
return
#~ safety feature in case twitter f's up
if len(friendIDs) < 100:
print('not enough friends')
return
#~ load protected and pending list
me = self.me.screen_name
cachedir = self.get_cachedir()
protfile = cachedir + me + '.prot'
newprot = []
prot = []
if os.path.exists(protfile):
with open(protfile) as f:
prot = [int(x.strip()) for x in f.readlines()]
#~ follow everyone in followerIDs
for a in followerIDs:
if (a not in friendIDs) and (a not in prot):
try:
ret = api.CreateFriendship(a)
friendIDs.append(a)
print('followed %s' % ret.screen_name)
except TwitterError as err:
print(err)
if not 'follow-limits' in str(err):
newprot.append(a)
else:
dialog.alert(
"You have reached Twitter's follow limit!")
break
#~ those we can not follow are probably protected
with open(protfile, 'a') as f:
f.writelines([str(x) + '\n' for x in newprot])
def unfollow_non_followers(self, widget, data=None):
"""unfollow everybody who is not following me
todo: add safelist"""
if not self.get_friendIDs():
return
me = self.me.screen_name
# load safelist
cachedir = self.get_cachedir()
safefile = cachedir + me + '.safe'
safe = []
unsub = []
if os.path.exists(safefile):
with open(safefile) as f:
safe = [x.strip() for x in f.readlines()]
#~ safety feature in case twitter f*s up
if len(followerIDs) < 100:
print('Need at least 100 followers to use this feature.')
return
for a in friendIDs:
if (a not in followerIDs) and (a not in safe):
unsub.append(a)
#~ unsub.remove(self.me.id)
#~ save nofollow list so we do not re-follow
#~ (unless they follow us, of course)
unsub_names = []
if dialog.ok('Unfriend all ' + str(len(unsub)) + ' non-followers?'):
#UNSUB EVERYBODY IN unsub
for a in unsub:
try:
u = api.DestroyFriendship(a)
print('unfriended %s' % u.screen_name)
unsub_names.append(u.screen_name)
except TwitterError as err:
print(err)
nofollow = cachedir + me + '.nofollow'
with open(nofollow, 'a') as f:
f.writelines([x + '\n' for x in unsub_names])
def __init__(self):
"""main initialization routine"""
global api, friendIDs, friends
friends = []
tok = self.authorize()
self.warn = True
api = twitter.Api(self.consumer_key, self.consumer_secret,
tok['oauth_token'], tok['oauth_token_secret'])
builder = gtk.Builder()
builder.add_from_file("crowbar.glade")
self.window1 = builder.get_object("window1")
self.vbox1 = builder.get_object("vbox1")
self.sw1 = builder.get_object("sw1")
self.window2 = builder.get_object("window2")
self.textview1 = builder.get_object("textview1")
self.entry1 = builder.get_object("entry1")
self.button1 = builder.get_object("button1")
builder.connect_signals(self)
self.window2.connect("delete-event", self.delete_event)
self.button1.connect("clicked", self.post_update, self.entry1.get_text)
self.update_followers_count()
self.window1.set_title("Crowbar @" + self.me.screen_name)
self.initialize_window()
#~ some variables to remember what we did last
self.last_action = None
self.last_method = None
