This repository has been archived on 2019-12-04. You can view files and clone it, but cannot push or open issues or pull requests.
hrafn/mytwitter.py

225 lines
7.0 KiB
Python
Executable File

#!/usr/bin/python
#
# Custom twitter client... mostly for learning Python
import sys, twitter, wx, ConfigParser, os, tkMessageBox, datetime, dateutil.tz
class TwitWindow(wx.Frame):
""" Display Tweets, post to twitter """
def __init__(self, parent, title):
wx.Frame.__init__(self, parent, title=title, size=(300,600))
config = ConfigParser.ConfigParser()
config.read(os.path.expanduser("~/.mytwitter"))
self.username = config.get('global', 'username')
self.password = config.get('global', 'password')
self.num_entries = int(config.get('global', 'entries'))
self.refresh_time = int(config.get('global', 'refreshtime'))
self.labels = []
self.texts = []
self.init_widgets()
def init_widgets(self):
main_sizer = wx.BoxSizer(wx.VERTICAL)
# Create the scrolled frame that will hold the tweets
tweet_view = wx.ScrolledWindow(self)
tweet_sizer = wx.BoxSizer(wx.VERTICAL)
# Status bar
self.CreateStatusBar()
# Menu bar
filemenu = wx.Menu()
filemenu.Append(wx.ID_ABOUT, "&About", "About MyTwitter")
menu_exit = filemenu.Append(wx.ID_EXIT, "E&xit", "Close MyTwitter")
self.Bind(wx.EVT_MENU, self.on_exit, menu_exit)
menu_bar = wx.MenuBar()
menu_bar.Append(filemenu, "&File")
self.SetMenuBar(menu_bar)
# Create labels and text widgets
for i in range(0, self.num_entries):
self.labels.append(wx.TextCtrl(tweet_view, style=wx.TE_READONLY))
self.texts.append(wx.TextCtrl(tweet_view, style=wx.TE_MULTILINE | wx.TE_READONLY))
self.labels[i].SetDefaultStyle(wx.TextAttr('DARK_BLUE'))
self.texts[i].SetDefaultStyle(wx.TextAttr('LIGHT_BLUE'))
tweet_sizer.Add(self.labels[i])
tweet_sizer.Add(self.texts[i])
# Layout the tweet view widget
width,height=tweet_view.GetSizeTuple()
tweet_view.SetSizerAndFit(tweet_sizer)
tweet_view.SetScrollbars(0, 20, 0, height/20)
tweet_view.SetSize((300,400))
# A button to refresh manually
button_sizer = wx.BoxSizer(wx.HORIZONTAL)
refresh_button = wx.Button(self, label="Refresh")
refresh_button.Bind(wx.EVT_BUTTON, self.update_window)
button_sizer.Add(refresh_button)
# Create an update box at the bottom of the window
update_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.update_entry = wx.TextCtrl(self)
self.update_entry.Bind(wx.EVT_TEXT, self.char_counter)
self.update_entry.Bind(wx.EVT_TEXT_ENTER, self.update_status_callback)
update_button = wx.Button(self, label="Update")
update_button.Bind(wx.EVT_BUTTON, self.update_status)
self.update_count = wx.StaticText(self, label="0/140")
update_sizer.Add(self.update_entry, 1, wx.EXPAND)
update_sizer.Add(update_button)
update_sizer.Add(self.update_count)
### Set up bindings
# Bind scrollwheel to move the tweets, as well as page up/down
# fixme: convert this to wxPython
# self.tkroot.bind_all("<Button-4>", self.scroll_wheel)
# self.tkroot.bind_all("<Button-5>", self.scroll_wheel)
# self.tkroot.bind_all("<Up>", self.line_up)
# self.tkroot.bind_all("<Down>", self.line_down)
# self.tkroot.bind_all("<Prior>", self.page_up)
# self.tkroot.bind_all("<Next>", self.page_down)
# Pack the top level together
main_sizer.Add(tweet_view, 1, wx.EXPAND)
main_sizer.Add(button_sizer)
main_sizer.Add(update_sizer)
self.SetSizerAndFit(main_sizer)
# Init the twitter API
self.api = twitter.Api(username=self.username, password=self.password)
# Updates!
self.timer = wx.Timer(self)
self.timer.Bind(wx.EVT_TIMER, self.update_window_callback)
self.timer.Start(self.refresh_time * 1000)
self.update_window()
# end of init_widgets
def update_window(self):
print "DEBUG: update_window()"
timezone = dateutil.tz.gettz()
time_format = "%Y.%m.%d %H:%M:%S %Z"
statuses = self.api.GetFriendsTimeline(self.username, count=self.num_entries)
for i in range(0, self.num_entries):
if i < len(statuses):
# Update the label with the user's name and screen name
user = statuses[i].user
timestamp = datetime.datetime.strptime(statuses[i].created_at, "%a %b %d %H:%M:%S +0000 %Y")
timestamp = timestamp.replace(tzinfo=dateutil.tz.gettz('UTC'))
timestring = timestamp.astimezone(timezone).strftime(time_format)
labeltext = user.name + " (" + user.screen_name + ") " + timestring
self.labels[i].Clear()
self.labels[i].AppendText(labeltext)
# Display the text of the tweet
self.texts[i].Clear()
self.texts[i].AppendText(statuses[i].text)
def update_window_callback(self, event):
self.update_window()
def update_status(self):
text = self.update_entry.GetValue()
if len(text) > 140:
d = wx.MessageDialog(self, 'Tweet too long', 'Tweets must be <= 140 characters');
d.ShowModal()
d.Destroy()
return
self.update_entry.Clear()
self.api.PostUpdate(text)
self.update_window()
# Just calls update_status, here so that things
# that pass an event can be used
def update_status_callback(self, event):
self.update_status()
# fixme: convert all of the below to wxPython
# def scroll_wheel(self, event):
# if event.num == 4:
# self.tweet_view.yview('scroll', -5, 'units');
# if event.num == 5:
# self.tweet_view.yview('scroll', 5, 'units');
# def page_up(self, event):
# self.tweet_view.yview('scroll', -15, 'units');
# def page_down(self, event):
# self.tweet_view.yview('scroll', 15, 'units');
# def line_up(self, event):
# self.tweet_view.yview('scroll', -5, 'units');
# def line_down(self, event):
# self.tweet_view.yview('scroll', 5, 'units');
def char_counter(self, event):
new_count = str(len(self.update_entry.GetValue())) + "/140"
self.update_count.SetLabel(new_count)
def on_exit(self, event):
self.Close(True)
### end class TwitWindow
def print_lists():
data = api.GetUserLists()
for l in data['lists']:
print l.name
def print_statuses_list():
statuses = api.GetListStatuses(sys.argv[2])
print_formatted(statuses)
def print_statuses():
statuses = api.GetFriendsTimeline(username)
print_formatted(statuses)
def print_formatted(statuses):
for s in statuses:
print s.user.name.encode("utf-8"), "(", s.user.screen_name, ") :"
print s.text.encode("utf-8")
print
# main
app = wx.App(False)
root = TwitWindow(None, "MyTwitter")
root.Show(True)
app.MainLoop()