Spaces:
Running
Running
File size: 3,650 Bytes
cf4944b a413623 |
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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
def playlists_mismatch_func(playlistlink1, playlistlink2, output = 'link'):
"""Return the ids of videos that are only one of the playlists, and not in the other"""
def get_yt_id(url, ignore_playlist=False):
# Examples:
# - http://youtu.be/SA2iWivDJiE
# - http://www.youtube.com/watch?v=_oPAwA_Udwc&feature=feedu
# - http://www.youtube.com/embed/SA2iWivDJiE
# - http://www.youtube.com/v/SA2iWivDJiE?version=3&hl=en_US
query = urlparse(url)
if query.hostname == 'youtu.be': return query.path[1:]
if query.hostname in {'www.youtube.com', 'youtube.com', 'music.youtube.com'}:
if not ignore_playlist:
# use case: get playlist id not current video in playlist
with suppress(KeyError):
return parse_qs(query.query)['list'][0]
if query.path == '/watch': return parse_qs(query.query)['v'][0]
if query.path[:7] == '/watch/': return query.path.split('/')[1]
if query.path[:7] == '/embed/': return query.path.split('/')[2]
if query.path[:3] == '/v/': return query.path.split('/')[2]
playlist1id = get_yt_id(playlistlink1)
playlist2id = get_yt_id(playlistlink2)
print("Playlist IDs obtained")
assert playlist1id!= None, "Playlist 1 link is invalid"
assert playlist2id!= None, "Playlist 2 link is invalid"
vid1_ids = []
vid2_ids = []
def get_video_ids(playlistid):
vid_ids = []
next_page_token = None
while True:
pl_request = youtube.playlistItems().list(
part="contentDetails, snippet",
playlistId=playlistid,
maxResults=50,
pageToken = next_page_token
)
pl_response = pl_request.execute()
for item in pl_response['items']:
vid_ids.append(item['contentDetails']['videoId'])
next_page_token = pl_response.get('nextPageToken')
if next_page_token is None:
break
return vid_ids
vid1_ids = get_video_ids(playlist1id)
print("Playlist 1 video IDs obtained, no. of videos:", len(vid1_ids))
vid2_ids = get_video_ids(playlist2id)
print("Playlist 2 video IDs obtained, no. of videos:", len(vid2_ids))
print("Video IDs obtained")
def get_playlist_name(playlistid):
pl_request = youtube.playlists().list(
part="snippet",
id=playlistid,
maxResults=1
)
pl_response = pl_request.execute()
return pl_response['items'][0]['snippet']['title']
playlist1name = get_playlist_name(playlist1id)
playlist2name = get_playlist_name(playlist2id)
def get_video_name(videoid):
vid_request = youtube.videos().list(
part="snippet",
id=videoid,
maxResults=1
)
vid_response = vid_request.execute()
return vid_response['items'][0]['snippet']['title']
firstnotsecond = list(set(vid1_ids) - set(vid2_ids))
secondnotfirst = list(set(vid2_ids) - set(vid1_ids))
if output == 'id':
return firstnotsecond, secondnotfirst
elif output == 'link':
firstnotsecond = [f"https://youtu.be/{i}" for i in firstnotsecond]
secondnotfirst = [f"https://youtu.be/{i}" for i in secondnotfirst]
return firstnotsecond, secondnotfirst
elif output == 'name':
firstnotsecond = [get_video_name(i) for i in firstnotsecond]
secondnotfirst = [get_video_name(i) for i in secondnotfirst]
return firstnotsecond, secondnotfirst |