feat: 新增 pdf_translate 命令行工具

- 支持命令: translate/list/status/download/config
- 支持用户认证: --user --password 参数
- 翻译进度实时显示，结果可在网页查看
- 新增 /api/config 和 /api/translations 接口
- 修复异步翻译任务的配置获取逻辑

admin.py

@@ -255,6 +255,28 @@ def delete_translation(trans_id):
     return jsonify({'success': True})
 
 
+@admin_bp.route('/translation/<int:trans_id>/toggle-share', methods=['POST'])
+@admin_required
+def toggle_translation_share(trans_id):
+    """切换翻译共享状态"""
+    translation = Translation.query.get_or_404(trans_id)
+    translation.no_share = not translation.no_share
+    db.session.commit()
+    
+    # 记录日志
+    log = OperationLog(
+        user_id=session.get('user_id'),
+        username='admin',
+        action='toggle_translation_share',
+        target=f'翻译#{trans_id}',
+        detail=f'设置共享状态为: {not translation.no_share}'
+    )
+    db.session.add(log)
+    db.session.commit()
+    
+    return jsonify({'success': True, 'no_share': translation.no_share})
+
+
 # ==================== 缓存管理 ====================
 @admin_bp.route('/cache')
 @admin_required
@@ -341,14 +363,14 @@ def save_site_settings():
     """保存网站基础配置"""
     data = request.json
     
-    # 保存每个配置项
+    # 保存每个配置项（key 直接使用，不带 site_ 前缀）
     for key, value in data.items():
         if key in ['max_file_size', 'cache_expire_days']:
-            DynamicConfig.set(f'site_{key}', int(value), category='site', value_type='int', user_id=session.get('user_id'))
+            DynamicConfig.set(key, int(value), category='site', value_type='int', user_id=session.get('user_id'))
         elif key in ['enable_email_notify', 'enable_cache', 'enable_guest']:
-            DynamicConfig.set(f'site_{key}', bool(value), category='site', value_type='bool', user_id=session.get('user_id'))
+            DynamicConfig.set(key, bool(value), category='site', value_type='bool', user_id=session.get('user_id'))
         else:
-            DynamicConfig.set(f'site_{key}', value, category='site', user_id=session.get('user_id'))
+            DynamicConfig.set(key, value, category='site', user_id=session.get('user_id'))
     
     # 记录日志
     log = OperationLog(

app.py

@@ -365,8 +365,11 @@ def upload_pdf():
     cache_path = cache_service.get_cache(file_hash)
     from_cache = False
     
-    if cache_path and ENABLE_CACHE and not instruction:
+    # 检查是否有用户设置了不共享此文件
-        # 有缓存且无特殊翻译要求，直接使用缓存
+    no_share_check = Translation.query.filter_by(file_hash=file_hash, no_share=True).first()
+    
+    if cache_path and ENABLE_CACHE and not instruction and not no_share_check:
+        # 有缓存且无特殊翻译要求且无不共享标记，直接使用缓存
         from_cache = True
         output_path = cache_path
     else:
@@ -442,6 +445,36 @@ def upload_pdf():
     })
 
 
+@app.route('/api/config')
+def api_config():
+    """获取系统配置"""
+    from admin import get_llm_config, get_site_config
+    
+    return jsonify({
+        'site_name': get_site_config().get('site_name'),
+        'max_file_size': get_site_config().get('max_file_size'),
+        'cache_expire_days': get_site_config().get('cache_expire_days'),
+        'llm_config': get_llm_config()
+    })
+
+
+@app.route('/api/translations')
+def api_translations_list():
+    """获取翻译记录列表"""
+    user = get_current_user()
+    
+    if user:
+        translations = Translation.query.filter_by(user_id=user.id)\
+            .order_by(Translation.created_at.desc()).limit(20).all()
+    else:
+        # 访客返回空列表
+        translations = []
+    
+    return jsonify({
+        'translations': [t.to_dict() for t in translations]
+    })
+
+
 @app.route('/api/status/<int:translation_id>')
 def translation_status(translation_id):
     """获取翻译状态"""
@@ -459,8 +492,12 @@ def translation_status(translation_id):
         'id': translation.id,
         'status': translation.status,
         'progress': translation.progress,
+        'filename': translation.original_filename,
+        'pages': translation.page_count,
         'from_cache': translation.from_cache,
-        'error': translation.error_message
+        'error': translation.error_message,
+        'created_at': translation.created_at.isoformat() if translation.created_at else None,
+        'completed_at': translation.completed_at.isoformat() if translation.completed_at else None,
     })
 
 

models.py

@@ -178,6 +178,9 @@ class Translation(db.Model):
     # 是否来自缓存
     from_cache = db.Column(db.Boolean, default=False)
     
+    # 不共享缓存
+    no_share = db.Column(db.Boolean, default=False)  # 不共享此翻译给其他用户
+    
     # 重译信息
     retranslate_request = db.Column(db.Text, nullable=True)  # 重译要求
     parent_id = db.Column(db.Integer, db.ForeignKey('translations.id'), nullable=True)  # 原翻译ID
@@ -190,6 +193,7 @@ class Translation(db.Model):
             'status': self.status,
             'progress': self.progress,
             'from_cache': self.from_cache,
+            'no_share': self.no_share,
             'file_size': self.file_size,
             'created_at': self.created_at.isoformat() if self.created_at else None,
             'completed_at': self.completed_at.isoformat() if self.completed_at else None,

pdf_translate.py

@@ -0,0 +1,307 @@
+#!/usr/bin/env python3
+"""
+PDF翻译命令行工具
+
+使用方法:
+    pdf_translate translate <pdf_file> [--instruction "翻译要求"] [--user <username>] [--password <password>]
+    pdf_translate list [--user <username>] [--password <password>]
+    pdf_translate status <translation_id> [--user <username>] [--password <password>]
+    pdf_translate download <translation_id> <output_file> [--user <username>] [--password <password>]
+    pdf_translate config
+"""
+
+import argparse
+import sys
+import os
+import json
+import requests
+from pathlib import Path
+
+# API 基础地址
+API_BASE = "http://localhost:19000"
+
+def login(username, password, api_base=API_BASE):
+    """登录获取session"""
+    url = f"{api_base}/api/login"
+    data = {"username": username, "password": password}
+    
+    try:
+        response = requests.post(url, json=data, timeout=10)
+        result = response.json()
+        
+        if result.get('success'):
+            print(f"✅ 登录成功: {username}")
+            return response.cookies
+        else:
+            print(f"❌ 登录失败: {result.get('error', '未知错误')}")
+            return None
+    except Exception as e:
+        print(f"❌ 登录请求失败: {e}")
+        return None
+
+def get_user_info(cookies, api_base=API_BASE):
+    """获取用户信息"""
+    url = f"{api_base}/api/user/info"
+    try:
+        response = requests.get(url, cookies=cookies, timeout=10)
+        result = response.json()
+        return result.get('user')
+    except:
+        return None
+
+def translate_pdf(pdf_file, instruction=None, cookies=None, api_base=API_BASE):
+    """上传并翻译PDF"""
+    if not os.path.exists(pdf_file):
+        print(f"❌ 文件不存在: {pdf_file}")
+        return None
+    
+    url = f"{api_base}/api/upload"
+    
+    print(f"📤 上传文件: {pdf_file}")
+    
+    try:
+        with open(pdf_file, 'rb') as f:
+            files = {'file': f}
+            data = {}
+            if instruction:
+                data['instruction'] = instruction
+            
+            response = requests.post(url, files=files, data=data, cookies=cookies, timeout=60)
+            result = response.json()
+        
+        if response.status_code != 200:
+            print(f"❌ 上传失败: {result.get('error', '未知错误')}")
+            return None
+        
+        translation_id = result.get('translation_id')
+        task_id = result.get('task_id')
+        from_cache = result.get('from_cache', False)
+        
+        print(f"✅ 上传成功!")
+        print(f"   翻译ID: {translation_id}")
+        print(f"   任务ID: {task_id}")
+        print(f"   使用缓存: {'是' if from_cache else '否'}")
+        
+        # 等待翻译完成
+        if not from_cache:
+            print(f"\n🔄 等待翻译完成...")
+            poll_translation_status(translation_id, cookies, api_base)
+        
+        return translation_id
+        
+    except Exception as e:
+        print(f"❌ 上传请求失败: {e}")
+        return None
+
+def poll_translation_status(translation_id, cookies=None, api_base=API_BASE):
+    """轮询翻译状态"""
+    url = f"{api_base}/api/status/{translation_id}"
+    
+    import time
+    max_wait = 300  # 最大等待5分钟
+    waited = 0
+    
+    while waited < max_wait:
+        try:
+            response = requests.get(url, cookies=cookies, timeout=10)
+            result = response.json()
+            
+            status = result.get('status', 'unknown')
+            progress = result.get('progress', 0)
+            
+            print(f"   状态: {status}, 进度: {progress}%")
+            
+            if status == 'completed':
+                print(f"✅ 翻译完成!")
+                return True
+            elif status == 'failed':
+                print(f"❌ 翻译失败: {result.get('error', '未知错误')}")
+                return False
+            
+            time.sleep(5)
+            waited += 5
+            
+        except Exception as e:
+            print(f"⚠️ 状态查询失败: {e}")
+            time.sleep(5)
+            waited += 5
+    
+    print(f"⚠️ 等待超时，请在网页查看结果")
+    return False
+
+def list_translations(cookies=None, api_base=API_BASE):
+    """列出翻译记录"""
+    url = f"{api_base}/api/translations"
+    
+    try:
+        response = requests.get(url, cookies=cookies, timeout=10)
+        result = response.json()
+        
+        translations = result.get('translations', [])
+        
+        if not translations:
+            print("暂无翻译记录")
+            return
+        
+        print(f"\n📋 翻译记录 (共{len(translations)}条):\n")
+        print("ID    | 文件名           | 状态      | 进度 | 时间")
+        print("-" * 60)
+        
+        for t in translations:
+            id_str = str(t.get('id', '?'))[:4]
+            filename = t.get('filename', '?')[:15]
+            status = t.get('status', '?')
+            progress = t.get('progress', 0)
+            created = t.get('created_at', '?')
+            if created and created != '?':
+                created = created[5:16]  # 月-日 时:分
+            
+            print(f"{id_str:4} | {filename:15} | {status:8} | {progress:3}% | {created}")
+        
+    except Exception as e:
+        print(f"❌ 获取列表失败: {e}")
+
+def get_translation_status(translation_id, cookies=None, api_base=API_BASE):
+    """获取翻译状态"""
+    url = f"{api_base}/api/status/{translation_id}"
+    
+    try:
+        response = requests.get(url, cookies=cookies, timeout=10)
+        result = response.json()
+        
+        print(f"\n📊 翻译状态 (ID: {translation_id}):\n")
+        print(f"   状态: {result.get('status', '未知')}")
+        print(f"   进度: {result.get('progress', 0)}%")
+        print(f"   文件: {result.get('filename', '未知')}")
+        print(f"   页数: {result.get('pages', '?')}")
+        print(f"   缓存: {'是' if result.get('from_cache') else '否'}")
+        
+        if result.get('error'):
+            print(f"   错误: {result.get('error')}")
+        
+    except Exception as e:
+        print(f"❌ 获取状态失败: {e}")
+
+def download_translation(translation_id, output_file, cookies=None, api_base=API_BASE):
+    """下载翻译结果"""
+    url = f"{api_base}/api/download/{translation_id}"
+    
+    try:
+        response = requests.get(url, cookies=cookies, timeout=30)
+        
+        if response.status_code != 200:
+            print(f"❌ 下载失败: {response.status_code}")
+            return False
+        
+        with open(output_file, 'wb') as f:
+            f.write(response.content)
+        
+        print(f"✅ 下载成功: {output_file}")
+        return True
+        
+    except Exception as e:
+        print(f"❌ 下载失败: {e}")
+        return False
+
+def show_config(api_base=API_BASE):
+    """显示当前大模型配置"""
+    url = f"{api_base}/api/config"
+    
+    try:
+        response = requests.get(url, timeout=10)
+        result = response.json()
+        
+        print(f"\n⚙️ 当前配置:\n")
+        print(f"   网站名称: {result.get('site_name', '未知')}")
+        print(f"   最大文件: {result.get('max_file_size', '?')}MB")
+        print(f"   缓存天数: {result.get('cache_expire_days', '?')}天")
+        
+        llm = result.get('llm_config', {})
+        print(f"   大模型API: {llm.get('api_base', '未设置')}")
+        print(f"   模型: {llm.get('model', '未设置')}")
+        
+    except Exception as e:
+        print(f"❌ 获取配置失败: {e}")
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="PDF翻译命令行工具",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+示例:
+  pdf_translate translate test.pdf --user admin --password admin123
+  pdf_translate translate test.pdf --instruction "保持专业术语原文"
+  pdf_translate list --user admin --password admin123
+  pdf_translate status 1
+  pdf_translate download 1 result.md
+  pdf_translate config
+        """
+    )
+    
+    parser.add_argument('--user', '-u', help='用户名')
+    parser.add_argument('--password', '-p', help='密码')
+    parser.add_argument('--api', default=API_BASE, help=f'API地址 (默认: {API_BASE})')
+    
+    subparsers = parser.add_subparsers(dest='command', help='命令')
+    
+    # translate 命令
+    trans_parser = subparsers.add_parser('translate', help='翻译PDF文件')
+    trans_parser.add_argument('file', help='PDF文件路径')
+    trans_parser.add_argument('--instruction', '-i', help='翻译要求')
+    
+    # list 命令
+    list_parser = subparsers.add_parser('list', help='列出翻译记录')
+    
+    # status 命令
+    status_parser = subparsers.add_parser('status', help='查看翻译状态')
+    status_parser.add_argument('id', type=int, help='翻译ID')
+    
+    # download 命令
+    download_parser = subparsers.add_parser('download', help='下载翻译结果')
+    download_parser.add_argument('id', type=int, help='翻译ID')
+    download_parser.add_argument('output', help='输出文件路径')
+    
+    # config 命令
+    config_parser = subparsers.add_parser('config', help='显示当前配置')
+    
+    args = parser.parse_args()
+    
+    # 更新 API 地址（如果用户指定了）
+    api_base_arg = args.api
+    
+    if not args.command:
+        parser.print_help()
+        return
+    
+    # 登录（如果提供了用户名密码）
+    cookies = None
+    if args.user and args.password:
+        cookies = login(args.user, args.password, api_base_arg)
+        if not cookies:
+            print("请检查用户名和密码")
+            return
+        
+        user = get_user_info(cookies, api_base_arg)
+        if user:
+            print(f"👤 用户: {user.get('username')} ({user.get('user_type')})")
+    
+    # 执行命令
+    if args.command == 'translate':
+        translation_id = translate_pdf(args.file, args.instruction, cookies, api_base_arg)
+        if translation_id:
+            print(f"\n💡 在网页查看结果: {api_base_arg}/history")
+    
+    elif args.command == 'list':
+        list_translations(cookies, api_base_arg)
+    
+    elif args.command == 'status':
+        get_translation_status(args.id, cookies, api_base_arg)
+    
+    elif args.command == 'download':
+        download_translation(args.id, args.output, cookies, api_base_arg)
+    
+    elif args.command == 'config':
+        show_config(api_base_arg)
+
+if __name__ == '__main__':
+    main()

services.py

@@ -279,12 +279,14 @@ class TranslationTask:
                 with app.app_context():
                     from admin import get_llm_config
                     llm_config = get_llm_config()
-                    config['LLM_CONFIG'] = llm_config
+                    config = {'LLM_CONFIG': llm_config}
             
             service = TranslationService(config)
             task['status'] = 'processing'
             task['started_at'] = datetime.now().isoformat()
             
+            print(f"[翻译任务] 开始翻译，使用配置: {config.get('LLM_CONFIG', {}).get('api_base', '未知')}")
+            
             # 更新数据库状态为 processing
             if app and translation_id:
                 with app.app_context():

templates/admin/translations.html

@@ -69,13 +69,14 @@
                             <th>大小</th>
                             <th>状态</th>
                             <th>缓存</th>
+                            <th>共享</th>
                             <th>时间</th>
                             <th>操作</th>
                         </tr>
                     </thead>
                     <tbody>
                         {% for t in translations.items %}
-                        <tr>
+                        <tr id="trans-row-{{ t.id }}">
                             <td>{{ t.id }}</td>
                             <td>{{ t.original_filename[:25] }}{% if t.original_filename|length > 25 %}...{% endif %}</td>
                             <td>{% if t.user_id %}ID:{{ t.user_id }}{% else %}访客{% endif %}</td>
@@ -87,14 +88,26 @@
                                 </span>
                             </td>
                             <td>{% if t.from_cache %}<i class="bi bi-check-circle text-success"></i>{% else %}-{% endif %}</td>
+                            <td>
+                                {% if t.no_share %}
+                                <span class="badge bg-secondary" id="share-badge-{{ t.id }}">不共享</span>
+                                {% else %}
+                                <span class="badge bg-success" id="share-badge-{{ t.id }}">共享</span>
+                                {% endif %}
+                            </td>
                             <td>{{ t.created_at.strftime('%m-%d %H:%M') }}</td>
                             <td>
-                                <a href="{{ url_for('admin.translation_detail', trans_id=t.id) }}" class="btn btn-sm btn-outline-primary"><i class="bi bi-eye"></i></a>
+                                <div class="d-flex gap-1">
-                                <button class="btn btn-sm btn-outline-danger" onclick="deleteTrans({{ t.id }})"><i class="bi bi-trash"></i></button>
+                                    <button class="btn btn-sm btn-outline-{% if t.no_share %}success{% else %}warning{% endif %}" onclick="toggleShare({{ t.id }})" title="切换共享状态">
+                                        <i class="bi bi-share{% if t.no_share %}-fill{% endif %}"></i>
+                                    </button>
+                                    <a href="{{ url_for('admin.translation_detail', trans_id=t.id) }}" class="btn btn-sm btn-outline-primary"><i class="bi bi-eye"></i></a>
+                                    <button class="btn btn-sm btn-outline-danger" onclick="deleteTrans({{ t.id }})"><i class="bi bi-trash"></i></button>
+                                </div>
                             </td>
                         </tr>
                         {% else %}
-                        <tr><td colspan="9" class="text-center text-muted py-4">暂无数据</td></tr>
+                        <tr><td colspan="10" class="text-center text-muted py-4">暂无数据</td></tr>
                         {% endfor %}
                     </tbody>
                 </table>
@@ -122,6 +135,26 @@
                 .then(r => r.json())
                 .then(d => { if (d.success) location.reload(); });
         }
+        
+        function toggleShare(id) {
+            fetch(`/admin/translation/${id}/toggle-share`, { method: 'POST' })
+                .then(r => r.json())
+                .then(d => {
+                    if (d.success) {
+                        // 更新显示状态
+                        const badge = document.getElementById(`share-badge-${id}`);
+                        if (d.no_share) {
+                            badge.className = 'badge bg-secondary';
+                            badge.textContent = '不共享';
+                        } else {
+                            badge.className = 'badge bg-success';
+                            badge.textContent = '共享';
+                        }
+                        // 更新按钮样式
+                        location.reload(); // 简化处理，刷新页面
+                    }
+                });
+        }
     </script>
 </body>
 </html>