X-Git-Url: https://jfr.im/git/z_archive/twitter.git/blobdiff_plain/f068ff42314f3096f1114b3417285b7fb0d72f3c..0469b7aaf9714edca95c03d5609a67a03d7d7103:/twitter/cmdline.py diff --git a/twitter/cmdline.py b/twitter/cmdline.py index e30272b..c605e21 100644 --- a/twitter/cmdline.py +++ b/twitter/cmdline.py @@ -13,11 +13,35 @@ OPTIONS: -e --email your email to login to twitter -p --password your twitter password + -r --refresh run this command forever, polling every once + in a while (default: every 5 minutes) + -R --refresh-rate set the refresh rate (in seconds) + -f --format specify the output format for status updates + -c --config read username and password from given config + file (default ~/.twitter) + +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? + +CONFIG FILES + + The config file should contain your email and password like so: + +[twitter] +email: +password: """ import sys +import time from getopt import getopt from getpass import getpass +import re +import os.path +from ConfigParser import SafeConfigParser from api import Twitter, TwitterError @@ -25,15 +49,17 @@ options = { 'email': None, 'password': None, 'action': 'friends', - 'forever': False, - 'refresh': 600, - 'verbose': 0, + 'refresh': False, + 'refresh_rate': 600, + 'format': 'default', + 'config_filename': os.environ.get('HOME', '') + os.sep + '.twitter', 'extra_args': [] } def parse_args(args, options): - long_opts = ['email', 'password', 'help', 'verbose'] - short_opts = "e:p:vh?" + long_opts = ['email', 'password', 'help', 'format', 'refresh', + 'refresh-rate', 'config'] + short_opts = "e:p:f:h?rR:c:" opts, extra_args = getopt(args, short_opts, long_opts) for opt, arg in opts: @@ -41,11 +67,17 @@ def parse_args(args, options): options['email'] = arg elif opt in ('-p', '--password'): options['password'] = arg - elif opt in ('-v', '--verbose'): - options['verbose'] = options.get('verbose', 0) + 1 + 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) + elif opt in ('-c', '--config'): + options['config_filename'] = arg if extra_args: options['action'] = extra_args[0] @@ -54,8 +86,7 @@ def parse_args(args, options): class StatusFormatter(object): def __call__(self, status): return (u"%s %s" %( - status['user']['screen_name'], status['text'])).encode( - sys.stdout.encoding, 'replace') + status['user']['screen_name'], status['text'])) class VerboseStatusFormatter(object): def __call__(self, status): @@ -63,54 +94,106 @@ class VerboseStatusFormatter(object): status['user']['screen_name'], status['user']['location'], status['created_at'], - status['text'])).encode( - sys.stdout.encoding, 'replace') + status['text'])) -def get_status_formatter(options): - if options['verbose']: - return VerboseStatusFormatter() - return StatusFormatter() +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 no_action(twitter, options): - print >> sys.stderr, "No such action: ", options['action'] - sys.exit(1) +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()) -def action_friends(twitter, options): - statuses = reversed(twitter.statuses.friends_timeline()) - sf = get_status_formatter(options) - for status in statuses: - print sf(status) - -def action_public(twitter, options): - statuses = reversed(twitter.statuses.public_timeline()) - sf = get_status_formatter(options) - for status in statuses: - print sf(status) - -def action_set_status(twitter, options): - status = (u" ".join(options['extra_args'])).encode( - 'utf8', 'replace') - twitter.statuses.update(status=status) +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': action_friends, - 'public': action_public, - 'set': action_set_status, + 'friends': FriendsAction, + 'public': PublicAction, + 'set': SetStatusAction, } +def loadConfig(filename): + email = None + password = None + if os.path.exists(filename): + cp = SafeConfigParser() + cp.read([filename]) + email = cp.get('twitter', 'email', None) + password = cp.get('twitter', 'password', None) + return email, password def main(): return main_with_args(sys.argv[1:]) def main_with_args(args): parse_args(args, options) + + email, password = loadConfig(options['config_filename']) + if not options['email']: options['email'] = email + if not options['password']: options['password'] = password + + 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'], no_action) + action = actions.get(options['action'], NoSuchAction)() try: - action(twitter, options) + 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, e.args[0] print >> sys.stderr, "Use 'twitter -h' for help." sys.exit(1) + except KeyboardInterrupt: + pass +