Merge pull request #76 from TakuyaMK/patch-1
[rainbowstream.git] / rainbowstream / rainbow.py
index fe144015df00367a566d8e0a1b5cc0abf4cd23b2..a5824a6382650984abeb11fc8baf0ed269b8bba3 100644 (file)
@@ -99,7 +99,7 @@ def proxy_connect(args):
                 int(args.proxy_port))
         else:
             printNicely(
-                magenta("Sorry, wrong proxy type specified! Aborting..."))
+                magenta('Sorry, wrong proxy type specified! Aborting...'))
             sys.exit()
         socket.socket = socks.socksocket
 
@@ -115,7 +115,7 @@ def authen():
             'USERPROFILE',
             '')) + os.sep + '.rainbow_oauth'
     if not os.path.exists(twitter_credential):
-        oauth_dance("Rainbow Stream",
+        oauth_dance('Rainbow Stream',
                     CONSUMER_KEY,
                     CONSUMER_SECRET,
                     twitter_credential)
@@ -167,10 +167,10 @@ def upgrade_center():
     Check latest and notify to upgrade
     """
     try:
-        current = pkg_resources.get_distribution("rainbowstream").version
+        current = pkg_resources.get_distribution('rainbowstream').version
         url = 'https://raw.githubusercontent.com/DTVD/rainbowstream/master/setup.py'
         readme = requests.get(url).text
-        latest = readme.split("version = \'")[1].split("\'")[0]
+        latest = readme.split('version = \'')[1].split('\'')[0]
         if current != latest:
             notice = light_magenta('RainbowStream latest version is ')
             notice += light_green(latest)
@@ -202,8 +202,6 @@ def init(args):
     credential = t.account.verify_credentials()
     screen_name = '@' + credential['screen_name']
     name = credential['name']
-    if not get_config('PREFIX'):
-        set_config('PREFIX', screen_name)
     c['original_name'] = g['original_name'] = screen_name[1:]
     g['listname'] = g['keyword'] = ''
     g['PREFIX'] = u2str(emojize(format_prefix()))
@@ -503,6 +501,27 @@ def reply():
     t.statuses.update(status=status, in_reply_to_status_id=tid)
 
 
+def reply_all():
+    """
+    Reply to all
+    """
+    t = Twitter(auth=authen())
+    try:
+        id = int(g['stuff'].split()[0])
+    except:
+        printNicely(red('Sorry I can\'t understand.'))
+        return
+    tid = c['tweet_dict'][id]
+    original_tweet = t.statuses.show(id=tid)
+    text = original_tweet['text']
+    owner = '@' + original_tweet['user']['screen_name']
+    nick_ary = ['@' + re.sub('[\W_]', '', w)
+                for w in text.split() if w.startswith('@')] + [owner]
+    status = ' '.join(g['stuff'].split()[1:])
+    status = ' '.join(nick_ary) + ' ' + str2u(status)
+    t.statuses.update(status=status, in_reply_to_status_id=tid)
+
+
 def favorite():
     """
     Favorite
@@ -1248,7 +1267,7 @@ def switch():
             g['keyword'] = keyword
             g['listname'] = ''
             # Reset prefix
-            g['PREFIX'] = u2str(emojize(format_prefix(keyword = g['keyword'])))
+            g['PREFIX'] = u2str(emojize(format_prefix(keyword=g['keyword'])))
             # Start new thread
             th = threading.Thread(
                 target=stream,
@@ -1279,11 +1298,13 @@ def switch():
             owner, slug = get_slug()
             # Force python 2 not redraw readline buffer
             listname = '/'.join([owner, slug])
-            # Set the listname variable 
+            # Set the listname variable
             # and reset tracked keyword
             g['listname'] = listname
             g['keyword'] = ''
-            g['PREFIX'] = g['cmd'] = u2str(emojize(format_prefix(listname = g['listname'])))
+            g['PREFIX'] = g['cmd'] = u2str(emojize(format_prefix(
+                listname=g['listname']
+            )))
             printNicely(light_yellow('getting list members ...'))
             # Get members
             t = Twitter(auth=authen())
@@ -1420,7 +1441,10 @@ def config():
                 g['decorated_name'] = lambda x: color_func(
                     c['DECORATED_NAME'])('[' + x + ']: ')
             elif key == 'PREFIX':
-                g['PREFIX'] = u2str(emojize(format_prefix(listname = g['listname'], keyword = g['keyword'])))
+                g['PREFIX'] = u2str(emojize(format_prefix(
+                    listname=g['listname'],
+                    keyword=g['keyword']
+                )))
             reload_config()
             printNicely(green('Updated successfully.'))
         except:
@@ -1480,7 +1504,10 @@ def help_tweets():
     usage += s * 2 + light_green('conversation 12') + ' will show the chain of ' + \
         'replies prior to the tweet with ' + light_yellow('[id=12]') + '.\n'
     usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
-        light_yellow('oops') + '" to tweet with ' + \
+        light_yellow('oops') + '" to the owner of the tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
+    usage += s * 2 + light_green('repall 12 oops') + ' will reply "' + \
+        light_yellow('oops') + '" to all people in the tweet with ' + \
         light_yellow('[id=12]') + '.\n'
     usage += s * 2 + \
         light_green('fav 12 ') + ' will favorite the tweet with ' + \
@@ -1768,6 +1795,7 @@ cmdset = [
     'conversation',
     'fav',
     'rep',
+    'repall',
     'del',
     'ufav',
     'share',
@@ -1815,6 +1843,7 @@ funcset = [
     conversation,
     favorite,
     reply,
+    reply_all,
     delete,
     unfavorite,
     share,
@@ -1875,6 +1904,7 @@ def listen():
             [],  # conversation
             [],  # favorite
             [],  # reply
+            [],  # reply_all
             [],  # delete
             [],  # unfavorite
             [],  # url
@@ -1968,12 +1998,12 @@ def reconn_notice():
     """
     Notice when Hangup or Timeout
     """
-    guide = light_magenta("You can use ") + \
-        light_green("switch") + \
-        light_magenta(" command to return to your stream.\n")
-    guide += light_magenta("Type ") + \
-        light_green("h stream") + \
-        light_magenta(" for more details.")
+    guide = light_magenta('You can use ') + \
+        light_green('switch') + \
+        light_magenta(' command to return to your stream.\n')
+    guide += light_magenta('Type ') + \
+        light_green('h stream') + \
+        light_magenta(' for more details.')
     printNicely(guide)
     sys.stdout.write(g['decorated_name'](g['PREFIX']))
     sys.stdout.flush()
@@ -2021,7 +2051,7 @@ def stream(domain, args, name='Rainbow Stream'):
         last_tweet_time = time.time()
         for tweet in tweet_iter:
             if tweet is None:
-                printNicely("-- None --")
+                printNicely('-- None --')
             elif tweet is Timeout:
                 # Because the stream check for each 0.3s
                 # so we shouldn't output anything here
@@ -2029,12 +2059,12 @@ def stream(domain, args, name='Rainbow Stream'):
                     StreamLock.release()
                     break
             elif tweet is HeartbeatTimeout:
-                printNicely("-- Heartbeat Timeout --")
+                printNicely('-- Heartbeat Timeout --')
                 reconn_notice()
                 StreamLock.release()
                 break
             elif tweet is Hangup:
-                printNicely("-- Hangup --")
+                printNicely('-- Hangup --')
                 reconn_notice()
                 StreamLock.release()
                 break
@@ -2082,10 +2112,15 @@ def stream(domain, args, name='Rainbow Stream'):
     except TwitterHTTPError as e:
         printNicely('')
         printNicely(
-            magenta("We have connection problem with twitter'stream API right now :("))
+            magenta('We have connection problem with twitter stream API right now :('))
         detail_twitter_error(e)
         sys.stdout.write(g['decorated_name'](g['PREFIX']))
         sys.stdout.flush()
+    except (URLError, ConnectionResetError):
+        printNicely(
+            magenta('There seems to be a connection problem.'))
+        save_history()
+        sys.exit()
 
 
 def fly():
@@ -2101,16 +2136,16 @@ def fly():
     except TwitterHTTPError as e:
         printNicely('')
         printNicely(
-            magenta("We have connection problem with twitter'REST API right now :("))
+            magenta('We have connection problem with twitter REST API right now :('))
         detail_twitter_error(e)
         save_history()
         sys.exit()
     # Proxy connection problem
     except (socks.ProxyConnectionError, URLError):
         printNicely(
-            magenta("There seems to be a connection problem."))
+            magenta('There seems to be a connection problem.'))
         printNicely(
-            magenta("You might want to check your proxy settings (host, port and type)!"))
+            magenta('You might want to check your proxy settings (host, port and type)!'))
         save_history()
         sys.exit()