-
Notifications
You must be signed in to change notification settings - Fork 1
/
search.py
73 lines (56 loc) · 2.23 KB
/
search.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#!/usr/bin/python
# This sample executes a search request for the specified search term.
# Sample usage:
# python search.py --q=surfing --max-results=10
# NOTE: To use the sample, you must provide a developer key obtained
# in the Google APIs Console. Search for "REPLACE_ME" in this code
# to find the correct place to provide that key..
import argparse
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
# Set DEVELOPER_KEY to the API key value from the APIs & auth > Registered apps
# tab of
# https://cloud.google.com/console
# Please ensure that you have enabled the YouTube Data API for your project.
DEVELOPER_KEY = 'AIzaSyC7OYQ0iXKGMYAjcTLIHEG_jE-1mfpDYog'
YOUTUBE_API_SERVICE_NAME = 'youtube'
YOUTUBE_API_VERSION = 'v3'
def print_videoURL(search_response):
videos = []
for search_result in search_response.get('items', []):
if search_result['id']['kind'] == 'youtube#video':
videos.append('%s' % (search_result['id']['videoId']))
print 'Videos:\nhttps://www.youtube.com/watch?v=', '\nhttps://www.youtube.com/watch?v='.join(videos), '\n'
def youtube_search(options):
youtube = build(YOUTUBE_API_SERVICE_NAME, YOUTUBE_API_VERSION,
developerKey=DEVELOPER_KEY)
search_response = youtube.search().list(
q=options.q,
part='id',
maxResults=options.max_results
).execute()
if 'nextPageToken' in search_response:
nextpageToken=search_response['nextPageToken']
print_videoURL(search_response)#print(nextpageToken)
while(1):
search_response = youtube.search().list(
q=options.q,
part='id,snippet',
maxResults=options.max_results,
pageToken=nextpageToken
).execute()
if 'nextPageToken' in search_response:
nextpageToken = search_response['nextPageToken']
print_videoURL(search_response)#print(nextpageToken)
else:
break
#
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--q', help='Search term', default='Google')
parser.add_argument('--max-results', help='Max results', default=50)
args = parser.parse_args()
try:
youtube_search(args)
except HttpError, e:
print 'An HTTP error %d occurred:\n%s' % (e.resp.status, e.content)