[fix] update youtube_noapi paging
This commit is contained in:
		
							parent
							
								
									dd34ac396c
								
							
						
					
					
						commit
						f97b4ff7b6
					
				@ -4,7 +4,7 @@
 | 
				
			|||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from functools import reduce
 | 
					from functools import reduce
 | 
				
			||||||
from json import loads
 | 
					from json import loads, dumps
 | 
				
			||||||
from urllib.parse import quote_plus
 | 
					from urllib.parse import quote_plus
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# about
 | 
					# about
 | 
				
			||||||
@ -20,12 +20,15 @@ about = {
 | 
				
			|||||||
# engine dependent config
 | 
					# engine dependent config
 | 
				
			||||||
categories = ['videos', 'music']
 | 
					categories = ['videos', 'music']
 | 
				
			||||||
paging = True
 | 
					paging = True
 | 
				
			||||||
 | 
					language_support = False
 | 
				
			||||||
time_range_support = True
 | 
					time_range_support = True
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# search-url
 | 
					# search-url
 | 
				
			||||||
base_url = 'https://www.youtube.com/results'
 | 
					base_url = 'https://www.youtube.com/results'
 | 
				
			||||||
search_url = base_url + '?search_query={query}&page={page}'
 | 
					search_url = base_url + '?search_query={query}&page={page}'
 | 
				
			||||||
time_range_url = '&sp=EgII{time_range}%253D%253D'
 | 
					time_range_url = '&sp=EgII{time_range}%253D%253D'
 | 
				
			||||||
 | 
					# the key seems to be constant
 | 
				
			||||||
 | 
					next_page_url = 'https://www.youtube.com/youtubei/v1/search?key=AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8'
 | 
				
			||||||
time_range_dict = {'day': 'Ag',
 | 
					time_range_dict = {'day': 'Ag',
 | 
				
			||||||
                   'week': 'Aw',
 | 
					                   'week': 'Aw',
 | 
				
			||||||
                   'month': 'BA',
 | 
					                   'month': 'BA',
 | 
				
			||||||
@ -40,21 +43,73 @@ base_youtube_url = 'https://www.youtube.com/watch?v='
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
# do search-request
 | 
					# do search-request
 | 
				
			||||||
def request(query, params):
 | 
					def request(query, params):
 | 
				
			||||||
    params['url'] = search_url.format(query=quote_plus(query),
 | 
					    if not params['engine_data'].get('next_page_token'):
 | 
				
			||||||
                                      page=params['pageno'])
 | 
					        params['url'] = search_url.format(query=quote_plus(query), page=params['pageno'])
 | 
				
			||||||
    if params['time_range'] in time_range_dict:
 | 
					        if params['time_range'] in time_range_dict:
 | 
				
			||||||
        params['url'] += time_range_url.format(time_range=time_range_dict[params['time_range']])
 | 
					            params['url'] += time_range_url.format(time_range=time_range_dict[params['time_range']])
 | 
				
			||||||
 | 
					    else:
 | 
				
			||||||
 | 
					        print(params['engine_data']['next_page_token'])
 | 
				
			||||||
 | 
					        params['url'] = next_page_url
 | 
				
			||||||
 | 
					        params['method'] = 'POST'
 | 
				
			||||||
 | 
					        params['data'] = dumps({
 | 
				
			||||||
 | 
					            'context': {"client": {"clientName": "WEB", "clientVersion": "2.20210310.12.01"}},
 | 
				
			||||||
 | 
					            'continuation': params['engine_data']['next_page_token'],
 | 
				
			||||||
 | 
					        })
 | 
				
			||||||
 | 
					        params['headers']['Content-Type'] = 'application/json'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return params
 | 
					    return params
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# get response from search-request
 | 
					# get response from search-request
 | 
				
			||||||
def response(resp):
 | 
					def response(resp):
 | 
				
			||||||
 | 
					    if resp.search_params.get('engine_data'):
 | 
				
			||||||
 | 
					        return parse_next_page_response(resp.text)
 | 
				
			||||||
 | 
					    return parse_first_page_response(resp.text)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def parse_next_page_response(response_text):
 | 
				
			||||||
    results = []
 | 
					    results = []
 | 
				
			||||||
 | 
					    result_json = loads(response_text)
 | 
				
			||||||
 | 
					    with open("/tmp/x", "w") as f:
 | 
				
			||||||
 | 
					        f.write(response_text)
 | 
				
			||||||
 | 
					    for section in (result_json['onResponseReceivedCommands'][0]
 | 
				
			||||||
 | 
					                    .get('appendContinuationItemsAction')['continuationItems'][0]
 | 
				
			||||||
 | 
					                    .get('itemSectionRenderer')['contents']):
 | 
				
			||||||
 | 
					        if 'videoRenderer' not in section:
 | 
				
			||||||
 | 
					            continue
 | 
				
			||||||
 | 
					        section = section['videoRenderer']
 | 
				
			||||||
 | 
					        content = "-"
 | 
				
			||||||
 | 
					        if 'descriptionSnippet' in section:
 | 
				
			||||||
 | 
					            content = ' '.join(x['text'] for x in section['descriptionSnippet']['runs'])
 | 
				
			||||||
 | 
					        results.append({
 | 
				
			||||||
 | 
					            'url': base_youtube_url + section['videoId'],
 | 
				
			||||||
 | 
					            'title': ' '.join(x['text'] for x in section['title']['runs']),
 | 
				
			||||||
 | 
					            'content': content,
 | 
				
			||||||
 | 
					            'author': section['ownerText']['runs'][0]['text'],
 | 
				
			||||||
 | 
					            'length': section['lengthText']['simpleText'],
 | 
				
			||||||
 | 
					            'template': 'videos.html',
 | 
				
			||||||
 | 
					            'embedded': embedded_url.format(videoid=section['videoId']),
 | 
				
			||||||
 | 
					            'thumbnail': section['thumbnail']['thumbnails'][-1]['url'],
 | 
				
			||||||
 | 
					        })
 | 
				
			||||||
 | 
					    try:
 | 
				
			||||||
 | 
					        token = result_json['onResponseReceivedCommands'][0]\
 | 
				
			||||||
 | 
					            .get('appendContinuationItemsAction')['continuationItems'][1]\
 | 
				
			||||||
 | 
					            .get('continuationItemRenderer')['continuationEndpoint']\
 | 
				
			||||||
 | 
					            .get('continuationCommand')['token']
 | 
				
			||||||
 | 
					        results.append({
 | 
				
			||||||
 | 
					            "engine_data": token,
 | 
				
			||||||
 | 
					            "key": "next_page_token",
 | 
				
			||||||
 | 
					        })
 | 
				
			||||||
 | 
					    except:
 | 
				
			||||||
 | 
					        pass
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    results_data = resp.text[resp.text.find('ytInitialData'):]
 | 
					    return results
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def parse_first_page_response(response_text):
 | 
				
			||||||
 | 
					    results = []
 | 
				
			||||||
 | 
					    results_data = response_text[response_text.find('ytInitialData'):]
 | 
				
			||||||
    results_data = results_data[results_data.find('{'):results_data.find(';</script>')]
 | 
					    results_data = results_data[results_data.find('{'):results_data.find(';</script>')]
 | 
				
			||||||
 | 
					 | 
				
			||||||
    results_json = loads(results_data) if results_data else {}
 | 
					    results_json = loads(results_data) if results_data else {}
 | 
				
			||||||
    sections = results_json.get('contents', {})\
 | 
					    sections = results_json.get('contents', {})\
 | 
				
			||||||
                           .get('twoColumnSearchResultsRenderer', {})\
 | 
					                           .get('twoColumnSearchResultsRenderer', {})\
 | 
				
			||||||
@ -63,6 +118,16 @@ def response(resp):
 | 
				
			|||||||
                           .get('contents', [])
 | 
					                           .get('contents', [])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    for section in sections:
 | 
					    for section in sections:
 | 
				
			||||||
 | 
					        if "continuationItemRenderer" in section:
 | 
				
			||||||
 | 
					            next_page_token = section["continuationItemRenderer"]\
 | 
				
			||||||
 | 
					                .get("continuationEndpoint", {})\
 | 
				
			||||||
 | 
					                .get("continuationCommand", {})\
 | 
				
			||||||
 | 
					                .get("token", "")
 | 
				
			||||||
 | 
					            if next_page_token:
 | 
				
			||||||
 | 
					                results.append({
 | 
				
			||||||
 | 
					                    "engine_data": next_page_token,
 | 
				
			||||||
 | 
					                    "key": "next_page_token",
 | 
				
			||||||
 | 
					                })
 | 
				
			||||||
        for video_container in section.get('itemSectionRenderer', {}).get('contents', []):
 | 
					        for video_container in section.get('itemSectionRenderer', {}).get('contents', []):
 | 
				
			||||||
            video = video_container.get('videoRenderer', {})
 | 
					            video = video_container.get('videoRenderer', {})
 | 
				
			||||||
            videoid = video.get('videoId')
 | 
					            videoid = video.get('videoId')
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user