From ed3d8e095eadbf2500ace5967b2b74173affe3d3 Mon Sep 17 00:00:00 2001 From: coder Date: Thu, 16 Apr 2026 21:50:24 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=20pdf=5Ftranslate=20?= =?UTF-8?q?=E5=91=BD=E4=BB=A4=E8=A1=8C=E5=B7=A5=E5=85=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 支持命令: translate/list/status/download/config - 支持用户认证: --user --password 参数 - 翻译进度实时显示，结果可在网页查看 - 新增 /api/config 和 /api/translations 接口 - 修复异步翻译任务的配置获取逻辑 --- app.py | 36 +++++- pdf_translate.py | 307 +++++++++++++++++++++++++++++++++++++++++++++++ services.py | 4 +- 3 files changed, 345 insertions(+), 2 deletions(-) create mode 100755 pdf_translate.py diff --git a/app.py b/app.py index 2230b2b..a29590c 100644 --- a/app.py +++ b/app.py @@ -445,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/') def translation_status(translation_id): """获取翻译状态""" @@ -462,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, }) diff --git a/pdf_translate.py b/pdf_translate.py new file mode 100755 index 0000000..d449514 --- /dev/null +++ b/pdf_translate.py @@ -0,0 +1,307 @@ +#!/usr/bin/env python3 +""" +PDF翻译命令行工具 + +使用方法: + pdf_translate translate [--instruction "翻译要求"] [--user ] [--password ] + pdf_translate list [--user ] [--password ] + pdf_translate status [--user ] [--password ] + pdf_translate download [--user ] [--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() \ No newline at end of file diff --git a/services.py b/services.py index 01b7b9c..1f68ce0 100644 --- a/services.py +++ b/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():