save_shorts.py

import yt_dlp
import csv
import os

def get_short_links(channel_url, output_csv='short_links.csv'):
    ydl_opts = {
        'quiet': True,
        'extract_flat': True,
        'playlist_end': 100,
    }

    # Modifie le format de l'URL si nécessaire
    if '/@' in channel_url:
        channel_username = channel_url.split('/@')[1].split('/')[0]
        channel_url = f'https://www.youtube.com/@{channel_username}/shorts'
    else:
        # Supprime les parties inutiles de l'URL
        suffixes_to_remove = ['/about', '/community', '/playlist', '/playlists', '/streams', '/featured', '/videos']
        for suffix in suffixes_to_remove:
            channel_url = channel_url.split(suffix)[0]
        channel_url += '/shorts'

    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        result = ydl.extract_info(channel_url, download=False)
        if 'entries' in result:
            video_info_list = []

            for entry in result['entries']:
                video_id = entry['id']
                video_title = entry.get('title', 'N/A')
                short_link = f'https://www.youtube.com/shorts/{video_id}'

                # Enregistre le lien et le titre dans la liste
                video_info_list.append((short_link, video_title))

            # Vérifie si le fichier CSV existe
            file_exists = os.path.isfile(output_csv)

            # Enregistre la liste dans un fichier CSV en mode 'a' (append) si le fichier existe, sinon en mode 'w' (write)
            with open(output_csv, mode='a' if file_exists else 'w', newline='', encoding='utf-8') as csv_file:
                csv_writer = csv.writer(csv_file)
                
                # Écrit l'en-tête seulement si le fichier vient d'être créé
                if not file_exists:
                    csv_writer.writerow(['Short Link', 'Video Title'])

                for video_info in video_info_list:
                    csv_writer.writerow(video_info)

            print(f"Les liens ont été ajoutés à {output_csv}")
            return video_info_list
        else:
            print("Aucune vidéo trouvée sur la chaîne.")
            return []