daviddoji/YouTube_API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add regex to split names properly

Browse Source
This commit is contained in:
David Doblas Jiménez
2024-01-21 21:16:25 +01:00
parent aaba1a6e6e
commit e8d5aced72
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
youtube_api.py
View File

@@ -30,6 +30,7 @@ def get_playlist_length(youtube, playlist_id):
return playlist_info['items'][0]['contentDetails']['itemCount'] return playlist_info['items'][0]['contentDetails']['itemCount']
def main(): def main():
import re
parser = argparse.ArgumentParser(description="Add a song to a YouTube playlist.") parser = argparse.ArgumentParser(description="Add a song to a YouTube playlist.")
parser.add_argument("channel_name", help="Name of the YouTube channel") parser.add_argument("channel_name", help="Name of the YouTube channel")
parser.add_argument("song", help="Name of the song to be added") parser.add_argument("song", help="Name of the song to be added")
@@ -38,9 +39,10 @@ def main():
channel_name = args.channel_name channel_name = args.channel_name
song = args.song song = args.song
# this assumes a song is passed as # this assumes a song is passed as in either of the following ways
# https://youtu.be/eEnyi9L6KP4 # https://youtu.be/eEnyi9L6KP4 --> split by "/"
song = str(song).split("/")[-1] # https://youtube.com/watch?v=Ez-gizOF0Wo --> split by "="
song = re.split(r"/|=", song)[-1]
if channel_name not in channels: if channel_name not in channels:
print(f"Error: Channel '{channel_name}' not found.") print(f"Error: Channel '{channel_name}' not found.")