Create youtube_extractor.py

youtube_extractor.py

@@ -0,0 +1,48 @@
+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