get_video_id_from_youtube

get_video_id_from_youtube.py

import csv
import pandas as pd
import googleapiclient
from tqdm import tqdm

DEVELOPER_KEY_GCP = "key_here"

def get_video_id_from_youtube(df):
    # Disable OAuthlib's HTTPS verification when running locally.
    # *DO NOT* leave this option enabled in production.
    os.environ["OAUTHLIB_INSECURE_TRANSPORT"] = "1"
    api_service_name = "youtube"
    api_version = "v3"
    youtube = googleapiclient.discovery.build(
        api_service_name, api_version, developerKey = DEVELOPER_KEY_GCP)
    def get_info_from_youtube(row):
        print(str(row['artist'] +' '+row['title_music']))
        if row['videoID_youtube'] == "missing":
            print('tic')
            query = str(row['artist'] +' '+row['title_music'])
            request = youtube.search().list(
            part="snippet",
            maxResults=1,
            q= query
            )
            try:
                response = request.execute()
                time.sleep(1)
            except :
                return 'missing'
            videoID = response['items'][0]['id']['videoId']
            print(videoID)
            row['videoID_youtube'] = videoID
            with open('data/videoid.csv', 'a',newline='') as writeFile:
                writer = csv.writer(writeFile)
                writer.writerow([row['artist'], row['title_music'], videoID])
        else:
            print('tac')
        return row
    tqdm.pandas()
    df = df.progress_apply(get_info_from_youtube,axis=1)
    return df
  
  output_file = 'data/trackes_list.csv'
  df = pd.read_csv(output_file,sep=";", header=0)
  df = get_video_id_from_youtube(df)