snegov/robocyp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Stop using plitem object in add/remove playlist items functions

Browse Source
This commit is contained in:
Maks Snegov 2024-02-03 12:52:40 -08:00
parent 739338e2d8
commit e3c51daa10
1 changed files with 30 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
main.py
View File

@ -86,7 +86,7 @@ def get_playlist_name(playlist_id: str) -> str:
if plid == playlist_id), playlist_id) if plid == playlist_id), playlist_id)
def list_playlist(yt_api, playlist_id): def list_playlist(yt_api, playlist_id: str):
playlist_name = get_playlist_name(playlist_id) playlist_name = get_playlist_name(playlist_id)
videos = [] videos = []
fetched = 0 fetched = 0
@ -114,14 +114,12 @@ def list_playlist(yt_api, playlist_id):
return videos return videos
def add_video_to_playlist(yt_api, video, playlist_id, def add_video_to_playlist(yt_api, video_id: str, playlist_id: str,
dry_run: bool = False) -> bool: dry_run: bool = False) -> bool:
playlist_name = get_playlist_name(playlist_id) playlist_name = get_playlist_name(playlist_id)
video_id = video['snippet']['resourceId']['videoId']
video_title = _truncate_title(video['snippet']['title'])
try: try:
if dry_run: if dry_run:
print(f"Would add video '{video_title}' [{video_id}] to playlist {playlist_name}") print(f"Would add video {video_id} to playlist {playlist_name}")
return True return True
yt_api.playlistItems().insert( yt_api.playlistItems().insert(
part='snippet', part='snippet',
@ -135,33 +133,29 @@ def add_video_to_playlist(yt_api, video, playlist_id,
} }
} }
).execute() ).execute()
print(f"Added video '{video_title}' [{video_id}] to playlist {playlist_name}") print(f"Added video {video_id} to playlist {playlist_name}")
return True return True
except HttpError as e: except HttpError as e:
exit_on_exceeded_quota(e) exit_on_exceeded_quota(e)
print(f"Error adding video '{video_title}' [{video_id}] to playlist {playlist_name}: {e}") print(f"Error adding video {video_id} to playlist {playlist_name}: {e}")
return False return False
def remove_video_from_playlist(yt_api, video, playlist_id, def remove_video_from_playlist(yt_api, plitem_id: str, playlist_id: str,
dry_run: bool = False) -> bool: dry_run: bool = False) -> bool:
playlist_name = get_playlist_name(playlist_id) playlist_name = get_playlist_name(playlist_id)
video_id = video['snippet']['resourceId']['videoId']
video_title = _truncate_title(video['snippet']['title'])
try: try:
if dry_run: if dry_run:
print(f"Would remove video '{video_title}' [{video_id}]" print(f"Would remove playlist item {plitem_id}]"
f" from playlist {playlist_name}") f" from playlist {playlist_name}")
return True return True
yt_api.playlistItems().delete( yt_api.playlistItems().delete(id=plitem_id).execute()
id=video['id'] print(f"Removed playlist item {plitem_id}"
).execute()
print(f"Removed video '{video_title}' [{video_id}]"
f" from playlist {playlist_name}") f" from playlist {playlist_name}")
return True return True
except HttpError as e: except HttpError as e:
exit_on_exceeded_quota(e) exit_on_exceeded_quota(e)
print(f"Error removing video '{video_title}' [{video_id}]" print(f"Error removing playlist item {plitem_id}"
f" from playlist {playlist_name}: {e}") f" from playlist {playlist_name}: {e}")
return False return False
@ -172,21 +166,21 @@ def copy_playlist_items(yt_api,
delete_from_src: bool = False, delete_from_src: bool = False,
limit: int = -1, limit: int = -1,
dry_run: bool = False): dry_run: bool = False):
src_videos = list_playlist(yt_api, src_playlist) src_playlist_items = list_playlist(yt_api, src_playlist)
dst_videos = list_playlist(yt_api, dst_playlist) dst_playlist_items = list_playlist(yt_api, dst_playlist)
if limit < 0: if limit < 0:
limit = len(src_videos) limit = len(src_playlist_items)
dst_video_map = {video['snippet']['resourceId']['videoId']: video['id'] dst_videos = {pl_item['snippet']['resourceId']['videoId']
for video in dst_videos} for pl_item in dst_playlist_items}
for src_video in src_videos[:limit]: for src_pl_item in src_playlist_items[:limit]:
video_id = src_video['snippet']['resourceId']['videoId'] video_id = src_pl_item['snippet']['resourceId']['videoId']
if video_id not in dst_video_map: if video_id not in dst_videos:
add_video_to_playlist(yt_api, src_video, dst_playlist, dry_run) add_video_to_playlist(yt_api, video_id, dst_playlist, dry_run)
if delete_from_src: if delete_from_src:
remove_video_from_playlist(yt_api, src_video, src_playlist, dry_run) remove_video_from_playlist(yt_api, src_pl_item["id"], src_playlist, dry_run)
def get_video_info(youtube, video_id: str): def get_video_info(youtube, video_id: str):
@ -265,21 +259,18 @@ def main():
if video_id in pl_videos: if video_id in pl_videos:
print(f'Video {video_id} already in playlist {args.playlist}') print(f'Video {video_id} already in playlist {args.playlist}')
continue continue
video = get_video_info(youtube, video_id) add_video_to_playlist(youtube, video_id, playlist, args.dry_run)
if video is None:
continue
add_video_to_playlist(youtube, video, playlist, args.dry_run)
elif args.command == 'dups': elif args.command == "dups":
playlist = get_playlist_id(args.playlist) playlist_id = get_playlist_id(args.playlist)
videos = list_playlist(youtube, playlist) plitems = list_playlist(youtube, playlist_id)
video_map = {} plitems_processed = set()
for video in videos[:args.limit]: for plitem in plitems[:args.limit]:
video_id = video['snippet']['resourceId']['videoId'] video_id = plitem["snippet"]["resourceId"]["videoId"]
if video_id in video_map: if video_id in plitems_processed:
remove_video_from_playlist(youtube, video, playlist, args.dry_run) remove_video_from_playlist(youtube, plitem["id"], playlist_id, args.dry_run)
else: else:
video_map[video_id] = video plitems_processed.add(video_id)
if __name__ == '__main__': if __name__ == '__main__':