* Fix #74
authorvunhat_minh <vunhat_minh@dwango.co.jp>
Mon, 22 Dec 2014 10:17:49 +0000 (19:17 +0900)
committervunhat_minh <vunhat_minh@dwango.co.jp>
Mon, 22 Dec 2014 10:17:49 +0000 (19:17 +0900)
* Add ability to switch to no-keyword public stream

docs/conf.py
rainbowstream/rainbow.py
setup.py

index e2bf812..1ea3774 100644 (file)
@@ -48,9 +48,9 @@ copyright = u'2014, Vu Nhat Minh'
 # built documents.
 #
 # The short X.Y version.
-version = '1.2.3'
+version = '1.2.4'
 # The full version, including alpha/beta/rc tags.
-release = '1.2.3'
+release = '1.2.4'
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.
index 7fdab84..a2c80e7 100644 (file)
@@ -44,7 +44,7 @@ def parse_arguments():
     parser.add_argument(
         '-s',
         '--stream',
-       default="mine",
+             default="mine",
         help='Default stream after program start. (Default: mine)')
     parser.add_argument(
         '-to',
@@ -960,6 +960,7 @@ def get_slug():
             light_magenta('List name should follow "@owner/list_name" format.'))
         raise Exception('Wrong list name')
 
+
 def check_slug(list_name):
     """
     Check slug
@@ -975,6 +976,7 @@ def check_slug(list_name):
             light_magenta('List name should follow "@owner/list_name" format.'))
         raise Exception('Wrong list name')
 
+
 def show_lists(t):
     """
     List list
@@ -1273,94 +1275,19 @@ def switch():
         except:
             printNicely(red('Sorry, wrong format.'))
             return
-        # Public stream
-        if target == 'public':
-            keyword = g['stuff'].split()[1]
-            if keyword[0] == '#':
-                keyword = keyword[1:]
-            # Kill old thread
-            g['stream_stop'] = True
-            args.track_keywords = keyword
-            # Set the variable to tracked keyword
-            # and reset the listname
-            g['keyword'] = keyword
-            g['listname'] = ''
-            # Reset prefix
-            g['PREFIX'] = u2str(emojize(format_prefix(keyword=g['keyword'])))
-            # Start new thread
-            th = threading.Thread(
-                target=stream,
-                args=(
-                    c['PUBLIC_DOMAIN'],
-                    args))
-            th.daemon = True
-            th.start()
-        # Personal stream
-        elif target == 'mine':
-            # Kill old thread
-            g['stream_stop'] = True
-            # Reset the tracked keyword and listname
-            g['keyword'] = g['listname'] = ''
-            # Reset prefix
-            g['PREFIX'] = u2str(emojize(format_prefix()))
-            # Start new thread
-            th = threading.Thread(
-                target=stream,
-                args=(
-                    c['USER_DOMAIN'],
-                    args,
-                    g['original_name']))
-            th.daemon = True
-            th.start()
-        # Stream base on list
-        elif target == 'list':
-            try:
-                owner, slug = check_slug(g['stuff'].split()[1])
-            except:
-                owner, slug = get_slug()
-            # Force python 2 not redraw readline buffer
-            listname = '/'.join([owner, slug])
-            # Set the listname variable
-            # and reset tracked keyword
-            g['listname'] = listname
-            g['keyword'] = ''
-            g['PREFIX'] = g['cmd'] = u2str(emojize(format_prefix(
-                listname=g['listname']
-            )))
-            printNicely(light_yellow('getting list members ...'))
-            # Get members
-            t = Twitter(auth=authen())
-            members = []
-            next_cursor = -1
-            while next_cursor != 0:
-                m = t.lists.members(
-                    slug=slug,
-                    owner_screen_name=owner,
-                    cursor=next_cursor,
-                    include_entities=False)
-                for u in m['users']:
-                    members.append('@' + u['screen_name'])
-                next_cursor = m['next_cursor']
-            printNicely(light_yellow('... done.'))
-            # Build thread filter array
-            args.filter = members
-            # Kill old thread
-            g['stream_stop'] = True
-            # Start new thread
-            th = threading.Thread(
-                target=stream,
-                args=(
-                    c['USER_DOMAIN'],
-                    args,
-                    slug))
-            th.daemon = True
-            th.start()
-        printNicely('')
-        if args.filter:
-            printNicely(cyan('Include: ' + str(len(args.filter)) + ' people.'))
-        if args.ignore:
-            printNicely(red('Ignore: ' + str(len(args.ignore)) + ' people.'))
-        printNicely('')
+        # Kill old thread
+        g['stream_stop'] = True
+        try: 
+            stuff = g['stuff'].split()[1]
+        except:
+            stuff = None
+        # Spawn new thread
+        spawn_dict = {
+            'public': spawn_public_stream,
+            'list': spawn_list_stream,
+            'mine': spawn_personal_stream,
+        }
+        spawn_dict.get(target)(args, stuff)
     except:
         debug_option()
         printNicely(red('Sorry I can\'t understand.'))
@@ -2042,7 +1969,7 @@ def stream(domain, args, name='Rainbow Stream'):
         c['SITE_DOMAIN']: name,
     }
     if c['ASCII_ART']:
-        ascii_art(art_dict[domain])
+        ascii_art(art_dict.get(domain, name))
     # These arguments are optional:
     stream_args = dict(
         timeout=0.5,  # To check g['stream_stop'] after each 0.5 s
@@ -2145,6 +2072,101 @@ def stream(domain, args, name='Rainbow Stream'):
         sys.exit()
 
 
+def spawn_public_stream(args, keyword=None):
+    """
+    Spawn a new public stream
+    """
+    # Only set keyword if specified
+    if keyword:
+        if keyword[0] == '#':
+            keyword = keyword[1:]
+        args.track_keywords = keyword
+    # Set the variable to tracked keyword
+    g['keyword'] = keyword
+    g['listname'] = ''
+    # Reset prefix
+    g['PREFIX'] = u2str(emojize(format_prefix(keyword=g['keyword'])))
+    # Start new thread
+    th = threading.Thread(
+        target=stream,
+        args=(
+            c['PUBLIC_DOMAIN'],
+            args))
+    th.daemon = True
+    th.start()
+
+
+def spawn_list_stream(args, stuff=None):
+    """
+    Spawn a new list stream
+    """
+    try:
+        owner, slug = check_slug(stuff)
+    except:
+        owner, slug = get_slug()
+    
+    # Force python 2 not redraw readline buffer
+    listname = '/'.join([owner, slug])
+    # Set the listname variable
+    # and reset tracked keyword
+    g['listname'] = listname
+    g['keyword'] = ''
+    g['PREFIX'] = g['cmd'] = u2str(emojize(format_prefix(
+        listname=g['listname']
+    )))
+    printNicely(light_yellow('getting list members ...'))
+    # Get members
+    t = Twitter(auth=authen())
+    members = []
+    next_cursor = -1
+    while next_cursor != 0:
+        m = t.lists.members(
+            slug=slug,
+            owner_screen_name=owner,
+            cursor=next_cursor,
+            include_entities=False)
+        for u in m['users']:
+            members.append('@' + u['screen_name'])
+        next_cursor = m['next_cursor']
+    printNicely(light_yellow('... done.'))
+    # Build thread filter array
+    args.filter = members
+    # Start new thread
+    th = threading.Thread(
+        target=stream,
+        args=(
+            c['USER_DOMAIN'],
+            args,
+            slug))
+    th.daemon = True
+    th.start()
+    printNicely('')
+    if args.filter:
+        printNicely(cyan('Include: ' + str(len(args.filter)) + ' people.'))
+    if args.ignore:
+        printNicely(red('Ignore: ' + str(len(args.ignore)) + ' people.'))
+    printNicely('')
+
+
+def spawn_personal_stream(args, stuff=None):
+    """
+    Spawn a new personal stream
+    """
+    # Reset the tracked keyword and listname
+    g['keyword'] = g['listname'] = ''
+    # Reset prefix
+    g['PREFIX'] = u2str(emojize(format_prefix()))
+    # Start new thread
+    th = threading.Thread(
+        target=stream,
+        args=(
+            c['USER_DOMAIN'],
+            args,
+            g['original_name']))
+    th.daemon = True
+    th.start()
+
+
 def fly():
     """
     Main function
@@ -2173,111 +2195,20 @@ def fly():
 
     # Spawn stream thread
     target = args.stream.split()[0]
-    if target == 'public':
+    if target == 'mine' :
+        spawn_personal_stream(args)
+    else :
         try:
-            keyword = args.stream.split()[1]
-            if keyword[0] == '#':
-                keyword = keyword[1:]
-            args.track_keywords = keyword
-            # Set the variable to tracked keyword
-            g['keyword'] = keyword
-            # Reset prefix
-            g['PREFIX'] = u2str(emojize(format_prefix(keyword=g['keyword'])))
-            # Start new thread
-            th = threading.Thread(
-                target=stream,
-                args=(
-                    c['PUBLIC_DOMAIN'],
-                    args))
-            th.daemon = True
-            th.start()
+            stuff = args.stream.split()[1]
+            spawn_dict = {
+                'public': spawn_public_stream,
+                'list': spawn_list_stream,
+            }
+            spawn_dict.get(target)(args, stuff)
         except:
-             printNicely(red('Public requires a keyword! Loading your personal stream.'))
-             # Start new thread
-             th = threading.Thread(
-                 target=stream,
-                 args=(
-                     c['USER_DOMAIN'],
-                     args,
-                     g['original_name']))
-             th.daemon = True
-             th.start()
-    elif target == "list":
-        try:
-            owner, slug = check_slug(args.stream.split()[1])
-            # Force python 2 not redraw readline buffer
-            listname = '/'.join([owner, slug])
-            # Set the listname variable
-            # and reset tracked keyword
-            g['listname'] = listname
-            g['keyword'] = ''
-            g['PREFIX'] = g['cmd'] = u2str(emojize(format_prefix(
-                listname=g['listname']
-            )))
-            printNicely(light_yellow('getting list members ...'))
-            # Get members
-            t = Twitter(auth=authen())
-            members = []
-            next_cursor = -1
-            while next_cursor != 0:
-                m = t.lists.members(
-                    slug=slug,
-                    owner_screen_name=owner,
-                    cursor=next_cursor,
-                    include_entities=False)
-                for u in m['users']:
-                    members.append('@' + u['screen_name'])
-                next_cursor = m['next_cursor']
-            printNicely(light_yellow('... done.'))
-            # Build thread filter array
-            args.filter = members
-            # Kill old thread
-            g['stream_stop'] = True
-            # Start new thread
-            th = threading.Thread(
-                target=stream,
-                args=(
-                    c['USER_DOMAIN'],
-                    args,
-                    slug))
-            th.daemon = True
-            th.start()
-            printNicely('')
-            if args.filter:
-                printNicely(cyan('Include: ' + str(len(args.filter)) + ' people.'))
-        except:
-             printNicely(red('List requieres a correct name of a list! Loading your personal stream.'))
-             # Start new thread
-             th = threading.Thread(
-                 target=stream,
-                 args=(
-                     c['USER_DOMAIN'],
-                     args,
-                     g['original_name']))
-             th.daemon = True
-             th.start()
-    elif target == "mine": 
-        # Start new thread
-        th = threading.Thread(
-            target=stream,
-            args=(
-                c['USER_DOMAIN'],
-                args,
-                g['original_name']))
-        th.daemon = True
-        th.start()
-    else: 
-        printNicely(red('Wrong -s/--stream argument given. Loading your personal stream.'))
-        # Start new thread
-        th = threading.Thread(
-            target=stream,
-            args=(
-                c['USER_DOMAIN'],
-                args,
-                g['original_name']))
-        th.daemon = True
-        th.start()
-
+            printNicely(red('Wrong -s/--stream argument given. Loading your personal stream.'))
+            spawn_personal_stream(args)
+    
     # Start listen process
     time.sleep(0.5)
     g['reset'] = True
index fc7a1af..4fc7dc5 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@ import os
 import os.path
 
 # Bumped version
-version = '1.2.3'
+version = '1.2.4'
 
 # Require
 install_requires = [