YouTube into Mendable

mendable-youtube.py

import requests
import scrapetube
import json

def get_video_urls_from_channel_or_playlist(id, type):
    if type == 'channel':
        videos = scrapetube.get_channel(id)
    elif type == 'playlist':
        videos = scrapetube.get_playlist(id)
    else:
        raise ValueError('Invalid type. Please choose "channel" or "playlist".')
    video_urls = ['https://www.youtube.com/watch?v=' + video['videoId'] for video in videos]
    return video_urls

def ingest_urls(api_key, urls):
    url = 'https://api.mendable.ai/v1/ingestData'
    headers = {'Content-Type': 'application/json'}
    for video_url in urls:
        data = {
            'api_key': api_key,
            'url': video_url,
            'type': 'youtube'
        }
        response = requests.post(url, headers=headers, data=json.dumps(data))
        if response.status_code != 200:
            print(f'Failed to ingest {video_url}. Status code: {response.status_code}')

# Example usage
urls = get_video_urls_from_channel_or_playlist('UCCezIgC97PvUuR4_gbFUs5g', 'channel')
ingest_urls('YOUR_API_KEY', urls)