| 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 |