Version 1.17.0

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

diff --git a/twitter/cmdline.py b/twitter/cmdline.py
index 0c21a8b8992699094308eb7867896283f6931ec7..545e8fdee1822d42e8ebc89f0bb34ab4acf1fff7 100755 (executable)
--- a/twitter/cmdline.py
+++ b/twitter/cmdline.py
@@ -22,7 +22,8 @@ ACTIONS:
  set            set your twitter status
  shell          login to the twitter shell
  rate           get your current rate limit status (remaining API reqs)
  set            set your twitter status
  shell          login to the twitter shell
  rate           get your current rate limit status (remaining API reqs)

-
+ repl           begin a Read-Eval-Print-Loop with a configured twitter
+                    object

 
 OPTIONS:
 
 
 OPTIONS:
 


@@ -73,13 +74,12 @@ except (AttributeError, KeyError):
 CONSUMER_KEY = 'uS6hO2sV6tDKIOeVjhnFnQ'
 CONSUMER_SECRET = 'MEYTOS97VvlHX7K1rwHPEqVpTSqZ71HtvoK4sVuYk'
 
 CONSUMER_KEY = 'uS6hO2sV6tDKIOeVjhnFnQ'
 CONSUMER_SECRET = 'MEYTOS97VvlHX7K1rwHPEqVpTSqZ71HtvoK4sVuYk'
 

+import code

 from getopt import gnu_getopt as getopt, GetoptError
 from getopt import gnu_getopt as getopt, GetoptError

-from getpass import getpass

 import json
 import locale
 import os.path
 import re
 import json
 import locale
 import os.path
 import re

-import string

 import sys
 import time
 
 import sys
 import time
 


@@ -97,10 +97,8 @@ try:
 except ImportError:
     import html.parser as HTMLParser
 
 except ImportError:
     import html.parser as HTMLParser
 

-import webbrowser
-

 from .api import Twitter, TwitterError
 from .api import Twitter, TwitterError

-from .oauth import OAuth, write_token_file, read_token_file
+from .oauth import OAuth, read_token_file

 from .oauth_dance import oauth_dance
 from . import ansi
 from .util import smrt_input, printNicely, align_text
 from .oauth_dance import oauth_dance
 from . import ansi
 from .util import smrt_input, printNicely, align_text


@@ -131,6 +129,7 @@ hashtagRe = re.compile(r'(?P<hashtag>#\S+)')
 profileRe = re.compile(r'(?P<profile>\@\S+)')
 ansiFormatter = ansi.AnsiCmd(False)
 
 profileRe = re.compile(r'(?P<profile>\@\S+)')
 ansiFormatter = ansi.AnsiCmd(False)
 

+

 def parse_args(args, options):
     long_opts = ['help', 'format=', 'refresh', 'oauth=',
                  'refresh-rate=', 'config=', 'length=', 'timestamp',
 def parse_args(args, options):
     long_opts = ['help', 'format=', 'refresh', 'oauth=',
                  'refresh-rate=', 'config=', 'length=', 'timestamp',


@@ -169,6 +168,7 @@ def parse_args(args, options):
         options['action'] = extra_args[0]
     options['extra_args'] = extra_args[1:]
 
         options['action'] = extra_args[0]
     options['extra_args'] = extra_args[1:]
 

+

 def get_time_string(status, options, format="%a %b %d %H:%M:%S +0000 %Y"):
     timestamp = options["timestamp"]
     datestamp = options["datestamp"]
 def get_time_string(status, options, format="%a %b %d %H:%M:%S +0000 %Y"):
     timestamp = options["timestamp"]
     datestamp = options["datestamp"]


@@ -187,6 +187,7 @@ def get_time_string(status, options, format="%a %b %d %H:%M:%S +0000 %Y"):
         return time.strftime("%Y-%m-%d ", t)
     return ""
 
         return time.strftime("%Y-%m-%d ", t)
     return ""
 

+

 def reRepl(m):
     ansiTypes = {
         'clear':   ansiFormatter.cmdReset(),
 def reRepl(m):
     ansiTypes = {
         'clear':   ansiFormatter.cmdReset(),


@@ -203,11 +204,14 @@ def reRepl(m):
         pass
     return s
 
         pass
     return s
 

+

 def replaceInStatus(status):
     txt = gHtmlParser.unescape(status)
     txt = re.sub(hashtagRe, reRepl, txt)
     txt = re.sub(profileRe, reRepl, txt)
     return txt
 def replaceInStatus(status):
     txt = gHtmlParser.unescape(status)
     txt = re.sub(hashtagRe, reRepl, txt)
     txt = re.sub(profileRe, reRepl, txt)
     return txt

+
+

 def correctRTStatus(status):
     if 'retweeted_status' in status:
         return ("RT @" + status['retweeted_status']['user']['screen_name']
 def correctRTStatus(status):
     if 'retweeted_status' in status:
         return ("RT @" + status['retweeted_status']['user']['screen_name']


@@ -215,6 +219,7 @@ def correctRTStatus(status):
     else:
         return status['text']
 
     else:
         return status['text']
 

+

 class StatusFormatter(object):
     def __call__(self, status, options):
         return ("%s@%s %s" % (
 class StatusFormatter(object):
     def __call__(self, status, options):
         return ("%s@%s %s" % (


@@ -222,6 +227,7 @@ class StatusFormatter(object):
             status['user']['screen_name'],
             gHtmlParser.unescape(correctRTStatus(status))))
 
             status['user']['screen_name'],
             gHtmlParser.unescape(correctRTStatus(status))))
 

+

 class AnsiStatusFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()
 class AnsiStatusFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()


@@ -234,6 +240,7 @@ class AnsiStatusFormatter(object):
             ansiFormatter.cmdReset(),
             align_text(replaceInStatus(correctRTStatus(status)))))
 
             ansiFormatter.cmdReset(),
             align_text(replaceInStatus(correctRTStatus(status)))))
 

+

 class VerboseStatusFormatter(object):
     def __call__(self, status, options):
         return ("-- %s (%s) on %s\n%s\n" % (
 class VerboseStatusFormatter(object):
     def __call__(self, status, options):
         return ("-- %s (%s) on %s\n%s\n" % (


@@ -242,13 +249,16 @@ class VerboseStatusFormatter(object):
             status['created_at'],
             gHtmlParser.unescape(correctRTStatus(status))))
 
             status['created_at'],
             gHtmlParser.unescape(correctRTStatus(status))))
 

+

 class JSONStatusFormatter(object):
     def __call__(self, status, options):
         status['text'] = gHtmlParser.unescape(status['text'])
         return json.dumps(status)
 
 class JSONStatusFormatter(object):
     def __call__(self, status, options):
         status['text'] = gHtmlParser.unescape(status['text'])
         return json.dumps(status)
 

+

 class URLStatusFormatter(object):
     urlmatch = re.compile(r'https?://\S+')
 class URLStatusFormatter(object):
     urlmatch = re.compile(r'https?://\S+')

+

     def __call__(self, status, options):
         urls = self.urlmatch.findall(correctRTStatus(status))
         return '\n'.join(urls) if urls else ""
     def __call__(self, status, options):
         urls = self.urlmatch.findall(correctRTStatus(status))
         return '\n'.join(urls) if urls else ""


@@ -262,6 +272,7 @@ class ListsFormatter(object):
             list_str = "%-30s" % (list['name'])
         return "%s\n" % list_str
 
             list_str = "%-30s" % (list['name'])
         return "%s\n" % list_str
 

+

 class ListsVerboseFormatter(object):
     def __call__(self, list):
         list_str = "%-30s\n description: %s\n members: %s\n mode:%s\n" % (
 class ListsVerboseFormatter(object):
     def __call__(self, list):
         list_str = "%-30s\n description: %s\n members: %s\n mode:%s\n" % (


@@ -269,6 +280,7 @@ class ListsVerboseFormatter(object):
             list['member_count'], list['mode'])
         return list_str
 
             list['member_count'], list['mode'])
         return list_str
 

+

 class AnsiListsFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()
 class AnsiListsFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()


@@ -288,6 +300,7 @@ class AdminFormatter(object):
         else:
             return "You are no longer following %s.\n" % (user_str)
 
         else:
             return "You are no longer following %s.\n" % (user_str)
 

+

 class VerboseAdminFormatter(object):
     def __call__(self, action, user):
         return("-- %s: %s (%s): %s" % (
 class VerboseAdminFormatter(object):
     def __call__(self, action, user):
         return("-- %s: %s (%s): %s" % (


@@ -296,21 +309,26 @@ class VerboseAdminFormatter(object):
             user['name'],
             user['url']))
 
             user['name'],
             user['url']))
 

+

 class SearchFormatter(object):
     def __call__(self, result, options):
         return("%s%s %s" % (
             get_time_string(result, options, "%a, %d %b %Y %H:%M:%S +0000"),
             result['from_user'], result['text']))
 
 class SearchFormatter(object):
     def __call__(self, result, options):
         return("%s%s %s" % (
             get_time_string(result, options, "%a, %d %b %Y %H:%M:%S +0000"),
             result['from_user'], result['text']))
 

+

 class VerboseSearchFormatter(SearchFormatter):
     pass  # Default to the regular one
 
 class VerboseSearchFormatter(SearchFormatter):
     pass  # Default to the regular one
 

+

 class URLSearchFormatter(object):
     urlmatch = re.compile(r'https?://\S+')
 class URLSearchFormatter(object):
     urlmatch = re.compile(r'https?://\S+')

+

     def __call__(self, result, options):
         urls = self.urlmatch.findall(result['text'])
         return '\n'.join(urls) if urls else ""
 
     def __call__(self, result, options):
         urls = self.urlmatch.findall(result['text'])
         return '\n'.join(urls) if urls else ""
 

+

 class AnsiSearchFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()
 class AnsiSearchFormatter(object):
     def __init__(self):
         self._colourMap = ansi.ColourMap()


@@ -323,6 +341,8 @@ class AnsiSearchFormatter(object):
             ansiFormatter.cmdReset(), result['text']))
 
 _term_encoding = None
             ansiFormatter.cmdReset(), result['text']))
 
 _term_encoding = None

+
+

 def get_term_encoding():
     global _term_encoding
     if not _term_encoding:
 def get_term_encoding():
     global _term_encoding
     if not _term_encoding:


@@ -367,6 +387,7 @@ lists_formatters = {
 }
 formatters['lists'] = lists_formatters
 
 }
 formatters['lists'] = lists_formatters
 

+

 def get_formatter(action_type, options):
     formatters_dict = formatters.get(action_type)
     if not formatters_dict:
 def get_formatter(action_type, options):
     formatters_dict = formatters.get(action_type)
     if not formatters_dict:


@@ -379,6 +400,7 @@ def get_formatter(action_type, options):
             "Unknown formatter '%s' for status actions" % (options['format']))
     return f()
 
             "Unknown formatter '%s' for status actions" % (options['format']))
     return f()
 

+

 class Action(object):
 
     def ask(self, subject='perform this action', careful=False):
 class Action(object):
 
     def ask(self, subject='perform this action', careful=False):


@@ -401,9 +423,9 @@ class Action(object):
         except EOFError:
             print(file=sys.stderr)  # Put Newline since Enter was never pressed
             # TODO:
         except EOFError:
             print(file=sys.stderr)  # Put Newline since Enter was never pressed
             # TODO:

-                #   Figure out why on OS X the raw_input keeps raising
-                #   EOFError and is never able to reset and get more input
-                #   Hint: Look at how IPython implements their console
+            #   Figure out why on OS X the raw_input keeps raising
+            #   EOFError and is never able to reset and get more input
+            #   Hint: Look at how IPython implements their console

             default = True
             if careful:
                 default = False
             default = True
             if careful:
                 default = False


@@ -424,13 +446,16 @@ class Action(object):
             print('\n[Keyboard Interrupt]', file=sys.stderr)
             pass
 
             print('\n[Keyboard Interrupt]', file=sys.stderr)
             pass
 

+

 class NoSuchActionError(Exception):
     pass
 
 class NoSuchActionError(Exception):
     pass
 

+

 class NoSuchAction(Action):
     def __call__(self, twitter, options):
         raise NoSuchActionError("No such action: %s" % (options['action']))
 
 class NoSuchAction(Action):
     def __call__(self, twitter, options):
         raise NoSuchActionError("No such action: %s" % (options['action']))
 

+

 class StatusAction(Action):
     def __call__(self, twitter, options):
         statuses = self.getStatuses(twitter, options)
 class StatusAction(Action):
     def __call__(self, twitter, options):
         statuses = self.getStatuses(twitter, options)


@@ -440,6 +465,7 @@ class StatusAction(Action):
             if statusStr.strip():
                 printNicely(statusStr)
 
             if statusStr.strip():
                 printNicely(statusStr)
 

+

 class SearchAction(Action):
     def __call__(self, twitter, options):
         # We need to be pointing at search.twitter.com to work, and it is less
 class SearchAction(Action):
     def __call__(self, twitter, options):
         # We need to be pointing at search.twitter.com to work, and it is less


@@ -459,6 +485,7 @@ class SearchAction(Action):
             if resultStr.strip():
                 printNicely(resultStr)
 
             if resultStr.strip():
                 printNicely(resultStr)
 

+

 class AdminAction(Action):
     def __call__(self, twitter, options):
         if not (options['extra_args'] and options['extra_args'][0]):
 class AdminAction(Action):
     def __call__(self, twitter, options):
         if not (options['extra_args'] and options['extra_args'][0]):


@@ -477,6 +504,7 @@ class AdminAction(Action):
         else:
             printNicely(af(options['action'], user))
 
         else:
             printNicely(af(options['action'], user))
 

+

 class ListsAction(StatusAction):
     def getStatuses(self, twitter, options):
         if not options['extra_args']:
 class ListsAction(StatusAction):
     def getStatuses(self, twitter, options):
         if not options['extra_args']:


@@ -494,6 +522,7 @@ class ListsAction(StatusAction):
             return []
         else:
             return list(reversed(twitter.lists.statuses(
             return []
         else:
             return list(reversed(twitter.lists.statuses(

+                count=options['length'],

                 owner_screen_name=screen_name,
                 slug=options['extra_args'][1])))
 
                 owner_screen_name=screen_name,
                 slug=options['extra_args'][1])))
 


@@ -510,19 +539,23 @@ class FriendsAction(StatusAction):
         return list(reversed(
             twitter.statuses.home_timeline(count=options["length"])))
 
         return list(reversed(
             twitter.statuses.home_timeline(count=options["length"])))
 

+

 class RepliesAction(StatusAction):
     def getStatuses(self, twitter, options):
         return list(reversed(
             twitter.statuses.mentions_timeline(count=options["length"])))
 
 class RepliesAction(StatusAction):
     def getStatuses(self, twitter, options):
         return list(reversed(
             twitter.statuses.mentions_timeline(count=options["length"])))
 

+

 class FollowAction(AdminAction):
     def getUser(self, twitter, user):
         return twitter.friendships.create(screen_name=user)
 
 class FollowAction(AdminAction):
     def getUser(self, twitter, user):
         return twitter.friendships.create(screen_name=user)
 

+

 class LeaveAction(AdminAction):
     def getUser(self, twitter, user):
         return twitter.friendships.destroy(screen_name=user)
 
 class LeaveAction(AdminAction):
     def getUser(self, twitter, user):
         return twitter.friendships.destroy(screen_name=user)
 

+

 class SetStatusAction(Action):
     def __call__(self, twitter, options):
         statusTxt = (" ".join(options['extra_args'])
 class SetStatusAction(Action):
     def __call__(self, twitter, options):
         statusTxt = (" ".join(options['extra_args'])


@@ -558,6 +591,7 @@ class SetStatusAction(Action):
         for status in splitted:
             twitter.statuses.update(status=status)
 
         for status in splitted:
             twitter.statuses.update(status=status)
 

+

 class TwitterShell(Action):
 
     def render_prompt(self, prompt):
 class TwitterShell(Action):
 
     def render_prompt(self, prompt):


@@ -586,11 +620,12 @@ class TwitterShell(Action):
                     continue
                 elif options['action'] == 'help':
                     print('''\ntwitter> `action`\n
                     continue
                 elif options['action'] == 'help':
                     print('''\ntwitter> `action`\n

-                          The Shell Accepts all the command line actions along with:
+                          The Shell accepts all the command line actions along with:

 
                           exit    Leave the twitter shell (^D may also be used)
 
 
                           exit    Leave the twitter shell (^D may also be used)
 

-                          Full CMD Line help is appended below for your convinience.''', file=sys.stderr)
+                          Full CMD Line help is appended below for your convenience.''',
+                          file=sys.stderr)

                 Action()(twitter, options)
                 options['action'] = ''
             except NoSuchActionError as e:
                 Action()(twitter, options)
                 options['action'] = ''
             except NoSuchActionError as e:


@@ -605,6 +640,7 @@ class TwitterShell(Action):
                 else:
                     raise SystemExit(0)
 
                 else:
                     raise SystemExit(0)
 

+

 class PythonPromptAction(Action):
     def __call__(self, twitter, options):
         try:
 class PythonPromptAction(Action):
     def __call__(self, twitter, options):
         try:


@@ -613,22 +649,50 @@ class PythonPromptAction(Action):
         except EOFError:
             pass
 
         except EOFError:
             pass
 

+

 class HelpAction(Action):
     def __call__(self, twitter, options):
         print(__doc__)
 
 class HelpAction(Action):
     def __call__(self, twitter, options):
         print(__doc__)
 

+

 class DoNothingAction(Action):
     def __call__(self, twitter, options):
         pass
 
 class DoNothingAction(Action):
     def __call__(self, twitter, options):
         pass
 

+

 class RateLimitStatus(Action):
     def __call__(self, twitter, options):
         rate = twitter.application.rate_limit_status()
 class RateLimitStatus(Action):
     def __call__(self, twitter, options):
         rate = twitter.application.rate_limit_status()

-        print("Remaining API requests: %s / %s (hourly limit)" % (
-            rate['remaining_hits'], rate['hourly_limit']))
-        print("Next reset in %ss (%s)" % (
-            int(rate['reset_time_in_seconds'] - time.time()),
-            time.asctime(time.localtime(rate['reset_time_in_seconds']))))
+        resources = rate['resources']
+        for resource in resources:
+            for method in resources[resource]:
+                limit = resources[resource][method]['limit']
+                remaining = resources[resource][method]['remaining']
+                reset = resources[resource][method]['reset']
+
+                print("Remaining API requests for %s: %s / %s" %
+                      (method, remaining, limit))
+                print("Next reset in %ss (%s)\n" % (int(reset - time.time()),
+                      time.asctime(time.localtime(reset))))
+
+
+class ReplAction(Action):
+    def __call__(self, twitter, options):
+        upload = Twitter(
+            auth=twitter.auth,
+            domain="upload.twitter.com")
+        printNicely(
+            "\nUse the 'twitter' object to interact with"
+            " the Twitter REST API.\n"
+            "Use twitter_upload to interact with "
+            "upload.twitter.com\n\n")
+        code.interact(local={
+            "twitter": twitter,
+            "t": twitter,
+            "twitter_upload": upload,
+            "u": upload
+            })
+

 
 actions = {
     'authorize' : DoNothingAction,
 
 actions = {
     'authorize' : DoNothingAction,


@@ -644,8 +708,10 @@ actions = {
     'set'       : SetStatusAction,
     'shell'     : TwitterShell,
     'rate'      : RateLimitStatus,
     'set'       : SetStatusAction,
     'shell'     : TwitterShell,
     'rate'      : RateLimitStatus,

+    'repl'      : ReplAction,

 }
 
 }
 

+

 def loadConfig(filename):
     options = dict(OPTIONS)
     if os.path.exists(filename):
 def loadConfig(filename):
     options = dict(OPTIONS)
     if os.path.exists(filename):


@@ -660,6 +726,7 @@ def loadConfig(filename):
                 options[option] = cp.getboolean('twitter', option)
     return options
 
                 options[option] = cp.getboolean('twitter', option)
     return options
 

+

 def main(args=sys.argv[1:]):
     arg_options = {}
     try:
 def main(args=sys.argv[1:]):
     arg_options = {}
     try:


@@ -679,7 +746,8 @@ def main(args=sys.argv[1:]):
     options = dict(OPTIONS)
     for d in config_options, arg_options:
         for k, v in list(d.items()):
     options = dict(OPTIONS)
     for d in config_options, arg_options:
         for k, v in list(d.items()):

-            if v: options[k] = v
+            if v:
+                options[k] = v

 
     if options['refresh'] and options['action'] not in ('friends', 'replies'):
         print("You can only refresh the friends or replies actions.",
 
     if options['refresh'] and options['action'] not in ('friends', 'replies'):
         print("You can only refresh the friends or replies actions.",