irc bot (not yet working)

[z_archive/twitter.git] / twitter / cmdline.py

diff --git a/twitter/cmdline.py b/twitter/cmdline.py
index af2a51dab886e9455c99b61ef158cd6fde2c4f0d..d36e873d695d9e731f009cd2da9cda283973d189 100644 (file)
--- a/twitter/cmdline.py
+++ b/twitter/cmdline.py
@@ -1,7 +1,170 @@
 """
-Twitter command line tool.
+USAGE:
 
+ twitter [action] [options]
+
+ACTIONS:
+
+ friends        get latest tweets from your friends (default action)
+ public         get latest public tweets
+ set            set your twitter status
+
+OPTIONS:
+
+ -e --email <email>         your email to login to twitter
+ -p --password <password>   your twitter password
+ -r --refresh               run this command forever, polling every once
+                            in a while (default: every 5 minutes)
+ -R --refresh-rate <rate>   set the refresh rate (in seconds)
+ -f --format <format>       specify the output format for status updates
+
+FORMATS for the --format option
+
+ default         one line per status
+ verbose         multiple lines per status, more verbose status info
+ urls            nothing but URLs. Dare you click them?
 """
 
-def main():
+import sys
+import time
+from getopt import getopt
+from getpass import getpass
+import re
+
+from api import Twitter, TwitterError
+
+options = {
+    'email': None,
+    'password': None,
+    'action': 'friends',
+    'refresh': False,
+    'refresh_rate': 600,
+    'format': 'default',
+    'extra_args': []
+}
+
+def parse_args(args, options):
+    long_opts = ['email', 'password', 'help', 'format', 'refresh',
+                 'refresh-rate']
+    short_opts = "e:p:f:h?rR:"
+    opts, extra_args = getopt(args, short_opts, long_opts)
+    
+    for opt, arg in opts:
+        if opt in ('-e', '--email'):
+            options['email'] = arg
+        elif opt in ('-p', '--password'):
+            options['password'] = arg
+        elif opt in ('-f', '--format'):
+            options['format'] = arg
+        elif opt in ('-r', '--refresh'):
+            options['refresh'] = True
+        elif opt in ('-R', '--refresh-rate'):
+            options['refresh_rate'] = int(arg)
+        elif opt in ('-?', '-h', '--help'):
+            print __doc__
+            sys.exit(0)
+    
+    if extra_args:
+        options['action'] = extra_args[0]
+    options['extra_args'] = extra_args[1:]
+
+class StatusFormatter(object):
+    def __call__(self, status):
+        return (u"%s %s" %(
+            status['user']['screen_name'], status['text']))
+
+class VerboseStatusFormatter(object):
+    def __call__(self, status):
+        return (u"-- %s (%s) on %s\n%s\n" %(
+            status['user']['screen_name'],
+            status['user']['location'],
+            status['created_at'],
+            status['text']))
+
+class URLStatusFormatter(object):
+    urlmatch = re.compile(r'https?://\S+')
+    def __call__(self, status):
+        urls = self.urlmatch.findall(status['text'])
+        return u'\n'.join(urls) if urls else ""
+
+formatters = {
+    'default': StatusFormatter,
+    'verbose': VerboseStatusFormatter,
+    'urls': URLStatusFormatter
+}    
+    
+def get_status_formatter(options):
+    sf = formatters.get(options['format'])
+    if (not sf):
+        raise TwitterError(
+            "Unknown formatter '%s'" %(options['format']))
+    return sf()
+
+class Action(object):
     pass
+
+class NoSuchAction(Action):
+    def __call__(self, twitter, options):
+        print >> sys.stderr, "No such action: ", options['action']
+        sys.exit(1)
+
+class StatusAction(Action):
+    def __call__(self, twitter, options):
+        statuses = self.getStatuses(twitter)
+        sf = get_status_formatter(options)
+        for status in statuses:
+            statusStr = sf(status)
+            if statusStr.strip():
+                print statusStr.encode(sys.stdout.encoding, 'replace')
+        
+class FriendsAction(StatusAction):
+    def getStatuses(self, twitter):
+        return reversed(twitter.statuses.friends_timeline())
+    
+class PublicAction(StatusAction):
+    def getStatuses(self, twitter):
+        return reversed(twitter.statuses.public_timeline())
+
+class SetStatusAction(Action):
+    def __call__(self, twitter, options):
+        statusTxt = (u" ".join(options['extra_args']) 
+                     if options['extra_args'] 
+                     else unicode(raw_input("message: ")))
+        status = (statusTxt.encode('utf8', 'replace'))
+        twitter.statuses.update(status=status)
+
+actions = {
+    'friends': FriendsAction,
+    'public': PublicAction,
+    'set': SetStatusAction,
+}
+
+
+def main():
+    return main_with_args(sys.argv[1:])
+    
+def main_with_args(args):
+    parse_args(args, options)
+    if options['refresh'] and options['action'] == 'set':
+        print >> sys.stderr, "You can't repeatedly set your status, silly"
+        print >> sys.stderr, "Use 'twitter -h' for help."
+        sys.exit(1)
+    if options['email'] and not options['password']:
+        options['password'] = getpass("Twitter password: ")
+    twitter = Twitter(options['email'], options['password'])
+    action = actions.get(options['action'], NoSuchAction)()
+    try:
+        doAction = lambda : action(twitter, options)
+        if (options['refresh']):
+            while True:
+                doAction()
+                time.sleep(options['refresh_rate'])
+        else:
+            doAction()
+    except TwitterError, e:
+        print >> sys.stderr, e.message
+        print >> sys.stderr, "Use 'twitter -h' for help."
+        sys.exit(1)
+    except KeyboardInterrupt:
+        pass
+