add channel searching

2 files changed, 72 insertions, 1 deletions

diff --git a/youtube/channel.py b/youtube/channel.py
index 5215249..414ea16 100644
--- a/youtube/channel.py
+++ b/youtube/channel.py
@@ -119,9 +119,22 @@ def grid_items_html(items, additional_info={}):
     result += '''\n</nav>'''
     return result
 
+def list_items_html(items, additional_info={}):
+    result = '''                <nav class="item-list">'''
+    for item in items:
+        result += common.renderer_html(item, additional_info)
+    result += '''\n</nav>'''
+    return result
+
 channel_tab_template = Template('''\n<a class="tab page-button"$href_attribute>$tab_name</a>''')
+channel_search_template = Template('''
+                <form class="channel-search" action="$action">
+                    <input type="search" name="query" class="search-box" value="$search_box_value">
+                    <button type="submit" value="Search" class="search-button">Search</button>
+                </form>''')
+
 tabs = ('Videos', 'Playlists', 'About')
-def channel_tabs_html(channel_id, current_tab):
+def channel_tabs_html(channel_id, current_tab, search_box_value=''):
     result = ''
     for tab_name in tabs:
         if tab_name == current_tab:
@@ -134,6 +147,10 @@ def channel_tabs_html(channel_id, current_tab):
                 href_attribute = 'href="' + URL_ORIGIN + "/channel/" + channel_id + "/" + tab_name.lower() + '"',
                 tab_name = tab_name,
             )
+    result += channel_search_template.substitute(
+        action = URL_ORIGIN + "/channel/" + channel_id + "/search",
+        search_box_value = html.escape(search_box_value),
+    )
     return result
             
 
@@ -238,7 +255,44 @@ def channel_about_page(polymer_json):
         stats               = stats,
         channel_tabs        = channel_tabs_html(channel_metadata['channelId'], 'About'),
     )
+
+def channel_search_page(polymer_json, query, current_page=1, number_of_videos = 1000, current_query_string=''):
+    microformat = polymer_json[1]['response']['microformat']['microformatDataRenderer']
+    channel_url = microformat['urlCanonical'].rstrip('/')
+    channel_id = channel_url[channel_url.rfind('/')+1:]
+
+    response = polymer_json[1]['response']
+    try:
+        items = response['contents']['twoColumnBrowseResultsRenderer']['tabs'][6]['expandableTabRenderer']['content']['sectionListRenderer']['contents']
+    except KeyError:
+        items = response['continuationContents']['sectionListContinuation']['contents']
+
+    items_html = list_items_html(items)
+
+    return yt_channel_items_template.substitute(
+        header              = common.get_header(),
+        channel_title       = html.escape(query + ' - Channel search'),
+        channel_tabs        = channel_tabs_html(channel_id, '', query),
+        avatar              = '/' + microformat['thumbnail']['thumbnails'][0]['url'],
+        page_title          = microformat['title'] + ' - Channel',
+        items               = items_html,
+        page_buttons        = common.page_buttons_html(current_page, math.ceil(number_of_videos/29), URL_ORIGIN + "/channel/" + channel_id + "/search", current_query_string),
+        number_of_results   = '',
+    )
+def get_channel_search_json(channel_id, query, page):
+    params = proto.string(2, 'search') + proto.string(15, str(page))
+    params = proto.percent_b64encode(params)
+    ctoken = proto.string(2, channel_id) + proto.string(3, params) + proto.string(11, query)
+    ctoken = base64.urlsafe_b64encode(proto.nested(80226972, ctoken)).decode('ascii')
+
+    polymer_json = common.fetch_url("https://www.youtube.com/browse_ajax?ctoken=" + ctoken, headers_1)
+    '''with open('debug/channel_search_debug', 'wb') as f:
+        f.write(polymer_json)'''
+    polymer_json = json.loads(polymer_json)
+
+    return polymer_json
     
+
 def get_channel_page(url, query_string=''):
     path_components = url.rstrip('/').lstrip('/').split('/')
     channel_id = path_components[0]
@@ -251,6 +305,7 @@ def get_channel_page(url, query_string=''):
     page_number = int(common.default_multi_get(parameters, 'page', 0, default='1'))
     sort = common.default_multi_get(parameters, 'sort', 0, default='3')
     view = common.default_multi_get(parameters, 'view', 0, default='1')
+    query = common.default_multi_get(parameters, 'query', 0, default='')
 
     if tab == 'videos':
         tasks = (
@@ -271,6 +326,15 @@ def get_channel_page(url, query_string=''):
             f.write(polymer_json)'''
         polymer_json = json.loads(polymer_json)
         return channel_playlists_html(polymer_json)
+    elif tab == 'search':
+        tasks = (
+            gevent.spawn(get_number_of_videos, channel_id ), 
+            gevent.spawn(get_channel_search_json, channel_id, query, page_number)
+        )
+        gevent.joinall(tasks)
+        number_of_videos, polymer_json = tasks[0].value, tasks[1].value
+
+        return channel_search_page(polymer_json, query, page_number, number_of_videos, query_string)
     else:
         raise ValueError('Unknown channel tab: ' + tab)
     
@@ -293,5 +357,10 @@ def get_user_page(url, query_string=''):
         polymer_json = common.fetch_url('https://www.youtube.com/user/' + username + '/playlists?pbj=1&view=1', headers_1)
         polymer_json = json.loads(polymer_json)
         return channel_playlists_html(polymer_json)
+    elif page == 'search':
+        raise NotImplementedError()
+        '''polymer_json = common.fetch_url('https://www.youtube.com/user' + username +  '/search?pbj=1&' + query_string, headers_1)
+        polymer_json = json.loads(polymer_json)
+        return channel_search_page('''
     else:
         raise ValueError('Unknown channel page: ' + page)
\ No newline at end of file
diff --git a/youtube/common.py b/youtube/common.py
index eaabdda..ab7ec17 100644
--- a/youtube/common.py
+++ b/youtube/common.py
@@ -593,6 +593,8 @@ did_you_mean = Template('''
 def renderer_html(renderer, additional_info={}, current_query_string=''):
     type = list(renderer.keys())[0]
     renderer = renderer[type]
+    if type == 'itemSectionRenderer':
+        return renderer_html(renderer['contents'][0], additional_info, current_query_string)
 
     if type == 'channelRenderer':
         info = renderer_info(renderer)