update

main.py

@@ -124,34 +124,40 @@ def get_all_galleries() -> List[GalleryInfo]:
     return galleries
 
 async def download_single_image(client: httpx.AsyncClient, url: str, file_path: Path, semaphore: asyncio.Semaphore) -> bool:
-    """下载单张图片"""
+    """下载单张图片 - 精简版"""
     async with semaphore:
         try:
-            logger.info(f"开始下载: {url} -> {file_path}")
+            logger.info(f"开始下载: {url}")
             
-            # 如果文件已存在，跳过下载
             if file_path.exists():
-                logger.info(f"文件已存在，跳过: {file_path}")
+                logger.info(f"文件已存在: {file_path}")
                 return True
             
-            # 发送请求下载图片
+            # 第一步：获取中间页面
-            async with client.stream('GET', url, timeout=DOWNLOAD_TIMEOUT) as response:
+            response = await client.get(url, timeout=DOWNLOAD_TIMEOUT)
-                response.raise_for_status()
+            response.raise_for_status()
             
-                # 异步写入文件
+            # 第二步：提取真实图片URL
-                async with aiofiles.open(file_path, 'wb') as f:
+            import re
-                    async for chunk in response.aiter_bytes():
+            match = re.search(r'img id="img" src="(.*?)"', response.text)
-                        await f.write(chunk)
+            if not match:
+                logger.error(f"无法提取图片URL: {url}")
+                return False
+            
+            real_img_url = match.group(1)
+            logger.info(f"真实URL: {real_img_url}")
+            
+            # 第三步：下载图片
+            img_response = await client.get(real_img_url, timeout=DOWNLOAD_TIMEOUT)
+            img_response.raise_for_status()
+            
+            # 保存图片
+            async with aiofiles.open(file_path, 'wb') as f:
+                await f.write(img_response.content)
             
             logger.info(f"下载完成: {file_path}")
             return True
             
-        except httpx.TimeoutException:
-            logger.error(f"下载超时: {url}")
-            return False
-        except httpx.HTTPStatusError as e:
-            logger.error(f"HTTP错误 {e.response.status_code}: {url}")
-            return False
         except Exception as e:
             logger.error(f"下载失败 {url}: {e}")
             return False