Release v0.1.0: GDM Refactor, Rate Limit, Metallic UI

downloader/http_direct.py

@@ -0,0 +1,91 @@
+"""
+HTTP 직접 다운로더
+- 단순 HTTP 파일 다운로드
+- aiohttp 비동기 사용 (고성능)
+"""
+import os
+import traceback
+from typing import Dict, Any, Optional, Callable
+
+from .base import BaseDownloader
+
+try:
+    from ..setup import P
+    logger = P.logger
+except:
+    import logging
+    logger = logging.getLogger(__name__)
+
+
+class HttpDirectDownloader(BaseDownloader):
+    """HTTP 직접 다운로더"""
+    
+    def download(
+        self,
+        url: str,
+        save_path: str,
+        filename: Optional[str] = None,
+        progress_callback: Optional[Callable] = None,
+        **options
+    ) -> Dict[str, Any]:
+        """HTTP로 직접 다운로드"""
+        try:
+            import requests
+            
+            os.makedirs(save_path, exist_ok=True)
+            
+            # 파일명 결정
+            if not filename:
+                filename = url.split('/')[-1].split('?')[0] or f"download_{int(__import__('time').time())}"
+            
+            filepath = os.path.join(save_path, filename)
+            
+            # 헤더 설정
+            headers = options.get('headers', {})
+            if 'User-Agent' not in headers:
+                headers['User-Agent'] = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+            
+            # 스트리밍 다운로드
+            response = requests.get(url, headers=headers, stream=True, timeout=60)
+            response.raise_for_status()
+            
+            total_size = int(response.headers.get('content-length', 0))
+            downloaded = 0
+            chunk_size = 1024 * 1024  # 1MB 청크
+            
+            with open(filepath, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=chunk_size):
+                    if self._cancelled:
+                        return {'success': False, 'error': 'Cancelled'}
+                    
+                    if chunk:
+                        f.write(chunk)
+                        downloaded += len(chunk)
+                        
+                        if total_size > 0 and progress_callback:
+                            progress = int(downloaded / total_size * 100)
+                            speed = ''  # TODO: 속도 계산
+                            progress_callback(progress, speed, '')
+            
+            if progress_callback:
+                progress_callback(100, '', '')
+            
+            return {'success': True, 'filepath': filepath}
+            
+        except Exception as e:
+            logger.error(f'HTTP download error: {e}')
+            logger.error(traceback.format_exc())
+            return {'success': False, 'error': str(e)}
+    
+    def get_info(self, url: str) -> Dict[str, Any]:
+        """URL 정보 추출"""
+        try:
+            import requests
+            
+            response = requests.head(url, timeout=10)
+            return {
+                'content_length': response.headers.get('content-length'),
+                'content_type': response.headers.get('content-type'),
+            }
+        except:
+            return {}