Add source

bot/modules/media_loader/direct.py

@@ -0,0 +1,68 @@
+"""
+Direct link downloads
+"""
+import aiohttp
+import aiofiles
+from pathlib import Path
+from typing import Optional
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def download_file(url: str, output_path: str, chunk_size: int = 8192) -> bool:
+    """
+    Download file from direct link
+    
+    Args:
+        url: File URL
+        output_path: Path to save
+        chunk_size: Chunk size for download
+    
+    Returns:
+        True if successful, False otherwise
+    """
+    try:
+        # Create directory if it doesn't exist
+        Path(output_path).parent.mkdir(parents=True, exist_ok=True)
+        
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as response:
+                if response.status != 200:
+                    logger.error(f"Download error: HTTP {response.status}")
+                    return False
+                
+                async with aiofiles.open(output_path, 'wb') as f:
+                    async for chunk in response.content.iter_chunked(chunk_size):
+                        await f.write(chunk)
+        
+        logger.info(f"File downloaded: {output_path}")
+        return True
+    
+    except Exception as e:
+        logger.error(f"Error downloading file: {e}")
+        return False
+
+
+async def get_file_size(url: str) -> Optional[int]:
+    """
+    Get file size from URL
+    
+    Args:
+        url: File URL
+    
+    Returns:
+        File size in bytes or None
+    """
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.head(url) as response:
+                if response.status == 200:
+                    content_length = response.headers.get('Content-Length')
+                    if content_length:
+                        return int(content_length)
+    except Exception as e:
+        logger.error(f"Error getting file size: {e}")
+    
+    return None
+