add e hentai

app.py

@@ -5,7 +5,8 @@ import yaml
 import logging
 import io
 import re
-import img2pdf  # 直接导入库，手动操作
+import img2pdf
+import subprocess  # 新增：用于调用 gallery-dl
 from jmcomic import create_option, JmcomicException
 
 # 定义下载目录
@@ -174,49 +175,158 @@ def run_batch_download(album_ids_str, auth_type, cookies_input, username_input,
     else:
         yield success_files, f"🎉 全部处理完毕！\n成功: {len(success_files)} / {total_count}", "全部完成"
 
-# --- 界面 (保持不变) ---
-with gr.Blocks(title="JMComic 批量下载器") as demo:
-    gr.Markdown("## 🚀 JMComic 批量下载器 (修复版)")
-    gr.Markdown("一行一个 ID，下载完一个才会开始下一个，稳定防崩。")
+# --- 新增：E-Hentai 下载核心逻辑 ---
+def run_eh_download(eh_url, cookies_str):
+    """
+    eh_url: E站画廊链接 (e-hentai.org 或 exhentai.org)
+    cookies_str: 浏览器导出的 cookies (Netscape格式 或 key=value)
+    """
+    # 1. 检查参数
+    if not eh_url:
+        yield None, "❌ 请输入画廊链接", "参数缺失"
+        return
     
-    with gr.Row():
-        inp_ids = gr.Textbox(label="输入本子 ID 列表 (回车分隔)", placeholder="123456\n234567", lines=5)
+    # 2. 目录准备
+    # 定义 EH 下载的临时目录
+    eh_base_dir = os.path.join(BASE_DIR, "eh_temp")
+    if os.path.exists(eh_base_dir):
+        try:
+            shutil.rmtree(eh_base_dir)
+        except:
+            pass
+    os.makedirs(eh_base_dir, exist_ok=True)
     
-    with gr.Row():
-        auth_select = gr.Radio(
-            ["使用账号密码 (推荐)", "使用 Cookies (手动)"], 
-            label="登录方式", 
-            value="使用账号密码 (推荐)"
-        )
+    # PDF 输出目录
+    pdf_final_dir = os.path.join(BASE_DIR, "final_pdfs")
+    os.makedirs(pdf_final_dir, exist_ok=True)
+
+    # 3. 处理 Cookies
+    # gallery-dl 最好使用 cookies.txt 文件
+    cookie_file_path = os.path.join(BASE_DIR, "eh_cookies.txt")
     
-    with gr.Group(visible=True) as group_user:
-        with gr.Row():
-            inp_user = gr.Textbox(label="用户名", placeholder="输入你的账号")
-            inp_pass = gr.Textbox(label="密码", type="password", placeholder="输入你的密码")
-            
-    with gr.Group(visible=False) as group_cookie:
-        inp_cookie = gr.Textbox(label="Cookies", placeholder="key=value...")
+    # 简单的 Cookie 写入逻辑
+    with open(cookie_file_path, "w", encoding="utf-8") as f:
+        # 如果用户输入的是 key=value; 格式，这里可能需要转换，
+        # 但 gallery-dl 对 Netscape 格式支持最好。
+        # 建议让用户直接粘贴 Netscape 格式 (EditThisCookie 插件导出)
+        f.write(cookies_str)
 
-    def toggle_auth(choice):
-        if choice == "使用账号密码 (推荐)":
-            return gr.update(visible=True), gr.update(visible=False)
-        else:
-            return gr.update(visible=False), gr.update(visible=True)
-    auth_select.change(fn=toggle_auth, inputs=auth_select, outputs=[group_user, group_cookie])
+    yield None, "🚀 正在启动 gallery-dl 解析...", "开始解析"
 
-    btn_run = gr.Button("🚀 批量下载", variant="primary")
-    
-    with gr.Row():
-        out_badge = gr.Label(value="等待指令", label="总体进度")
-        out_log = gr.Textbox(label="实时日志", lines=6)
+    # 4. 构建 gallery-dl 命令
+    # 格式: gallery-dl -d "目录" --cookies "cookie文件" "URL"
+    cmd = [
+        "gallery-dl",
+        "--directory", eh_base_dir,
+        "--cookies", cookie_file_path,
+        eh_url
+    ]
+
+    # 5. 执行下载
+    try:
+        process = subprocess.Popen(
+            cmd, 
+            stdout=subprocess.PIPE, 
+            stderr=subprocess.STDOUT,
+            text=True
+        )
+        
+        # 实时捕获日志
+        logs = ""
+        for line in process.stdout:
+            logs += line
+            # 简单的进度反馈
+            if "Example" in line or "http" in line:
+                yield None, f"📥 正在下载: \n{line.strip()}", "下载中"
+        
+        process.wait()
+
+        if process.returncode != 0:
+            yield None, f"❌ 下载失败，请检查 Cookies 是否过期或链接是否有效。\n日志:\n{logs[-500:]}", "下载失败"
+            return
+
+    except Exception as e:
+        yield None, f"❌ 调用 gallery-dl 出错: {e}", "系统错误"
+        return
+
+    # 6. 寻找下载好的文件夹
+    # gallery-dl 会在 eh_base_dir 下自动创建 "E-Hentai/画廊名" 的结构
+    # 我们需要找到包含图片的那个最底层文件夹
+    target_img_dir = None
+    for root, dirs, files in os.walk(eh_base_dir):
+        # 如果当前文件夹里有图片，就是它了
+        has_img = any(f.lower().endswith(('.jpg', '.png', '.jpeg')) for f in files)
+        if has_img:
+            target_img_dir = root
+            break
     
-    out_files = gr.File(label="下载结果", file_count="multiple")
+    if not target_img_dir:
+        yield None, "❌ 未找到下载的图片，可能是权限不足 (Sad Panda) 或下载被中断。", "未找到图片"
+        return
+
+    # 获取画廊标题作为文件名
+    folder_name = os.path.basename(target_img_dir)
+    safe_title = sanitize_filename(folder_name)
+    final_pdf_name = f"[EH] {safe_title}.pdf"
+    final_pdf_path = os.path.join(pdf_final_dir, final_pdf_name)
+
+    # 7. 合并 PDF (复用你现有的函数)
+    yield None, "🔨 图片下载完成，正在合并 PDF...", "正在合并"
+    try:
+        manual_merge_pdf(target_img_dir, final_pdf_path)
+    except Exception as e:
+        yield None, f"❌ 合并 PDF 失败: {e}", "合并失败"
+        return
+
+    yield [final_pdf_path], f"✅ 处理完成！\n文件: {final_pdf_name}", "完成"
+
+
+
+
+
+
+
+# --- 界面部分 (在原有 Blocks 中新增 Tab) ---
+with gr.Blocks(title="超级漫画下载器") as demo:
+    gr.Markdown("## 🚀 超级漫画下载器 (JMComic + E-Hentai)")
+
+    with gr.Tabs():
+        # === Tab 1: JMComic (你原有的代码) ===
+        with gr.TabItem("JMComic (ID下载)"):
+             # 这里放你原来的 JMComic 界面代码...
+             # (inp_ids, auth_select, btn_run 等等...)
+             # 为了节省篇幅，这里省略，保持你原样即可
+             gr.Markdown("这里是你原来的 JM 界面...")
+             
+        # === Tab 2: E-Hentai (新增) ===
+        with gr.TabItem("E-Hentai (链接下载)"):
+            gr.Markdown("输入 E站/EX站 画廊链接。**必须提供 Cookies** 才能下载原始画质或 EX 站内容。")
+            
+            with gr.Row():
+                inp_eh_url = gr.Textbox(label="画廊链接 (URL)", placeholder="https://e-hentai.org/g/12345/xxxx/")
+            
+            with gr.Row():
+                # 提示用户如何获取 Cookies
+                inp_eh_cookies = gr.Textbox(
+                    label="Cookies (Netscape 格式)", 
+                    placeholder="# Netscape HTTP Cookie File\n.e-hentai.org ...",
+                    lines=5,
+                    info="推荐使用 Chrome 插件 'Get cookies.txt LOCALLY' 导出并在粘贴至此。"
+                )
+
+            btn_eh_run = gr.Button("🚀 下载并转 PDF", variant="primary")
+            
+            with gr.Row():
+                out_eh_badge = gr.Label(value="等待指令", label="状态")
+                out_eh_log = gr.Textbox(label="运行日志", lines=3)
+            
+            out_eh_files = gr.File(label="下载结果")
 
-    btn_run.click(
-        fn=run_batch_download, 
-        inputs=[inp_ids, auth_select, inp_cookie, inp_user, inp_pass], 
-        outputs=[out_files, out_log, out_badge]
-    )
+            btn_eh_run.click(
+                fn=run_eh_download,
+                inputs=[inp_eh_url, inp_eh_cookies],
+                outputs=[out_eh_files, out_eh_log, out_eh_badge]
+            )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue().launch() # 建议加上 queue()