youtube_extractor.py

import yt_dlp

class YouTubeExtractor:
    def __init__(self, ydl_opts=None):
        self.ydl_opts = ydl_opts or {
            'quiet': True,
            'no_warnings': True,
            'no_color': True,
            'youtube_include_dash_manifest': False,
            'youtube_include_hls_manifest': False,
            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
        }

    def extract_info(self, youtube_url):
        try:
            with yt_dlp.YoutubeDL(self.ydl_opts) as ydl:
                info = ydl.extract_info(youtube_url, download=False)
                
                metadata = {
                    "제목": info.get('title', 'N/A'),
                    "채널": info.get('channel', 'N/A'),
                    "업로드 날짜": info.get('upload_date', 'N/A'),
                    "조회수": info.get('view_count', 'N/A'),
                    "길이 (초)": info.get('duration', 'N/A'),
                }
                
                formats = info.get('formats', [])
                
                best_video_only = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') == 'none'), None)
                combined = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') != 'none'), None)
                
                return metadata, best_video_only, combined

        except Exception as e:
            return None, None, None

    def format_output(self, metadata, best_video_only, combined):
        if not metadata:
            return "오류 발생: 메타데이터를 추출할 수 없습니다.", "다운로드 링크를 찾을 수 없습니다."

        metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
        
        video_only_link = f'<a href="{best_video_only["url"]}" target="_blank">비디오만 다운로드</a>' if best_video_only else ""
        combined_link = f'<a href="{combined["url"]}" target="_blank">비디오+오디오 다운로드</a>' if combined else ""
        
        download_links = f"{video_only_link}<br>{combined_link}" if video_only_link or combined_link else "다운로드 링크를 찾을 수 없습니다."
        
        return metadata_str, download_links