diff --git a/服务器启动/server.py b/服务器启动/server.py
new file mode 100644
index 0000000..b977e8b
--- /dev/null
+++ b/服务器启动/server.py
@@ -0,0 +1,567 @@
+from flask import Flask, request, jsonify, render_template, send_from_directory
+from flask_cors import CORS
+import os
+import traceback
+import logging
+import ctypes
+import shutil
+import threading
+import time
+import uuid
+import json
+from concurrent.futures import ThreadPoolExecutor
+from werkzeug.utils import secure_filename
+
+# 首先尝试导入兼容层模块
+try:
+ import pdfkit_patch as pdfkit
+ print("已加载pdfkit补丁模块")
+except ImportError:
+ print("未找到pdfkit补丁模块,尝试创建...")
+ # 创建一个简单的补丁模块
+ with open("pdfkit_patch.py", "w", encoding="utf-8") as f:
+ f.write("""
+import logging
+logger = logging.getLogger("pdfkit_patch")
+
+def from_string(html_content, output_path, options=None):
+ html_output = output_path.replace('.pdf', '.html')
+ with open(html_output, 'w', encoding='utf-8') as f:
+ f.write(html_content)
+ return True
+
+def from_file(input_file, output_file, options=None):
+ return True
+ """)
+ import pdfkit_patch as pdfkit
+ print("已创建并加载pdfkit补丁模块")
+
+# 有条件地导入毕设.py中的函数
+try:
+ from 毕设 import main_process
+ print("成功导入毕设.py处理函数")
+except Exception as e:
+ print(f"警告: 导入毕设.py失败: {str(e)}")
+ print("将使用简化的处理流程...")
+
+ # 定义一个简化的替代函数
+ def main_process(video_path, output_dir=None, progress_callback=None):
+ """简化的视频处理函数,用于在无法导入毕设.py时提供基本功能"""
+ print(f"使用简化处理函数处理视频: {video_path}")
+
+ if not os.path.exists(video_path):
+ print(f"错误: 视频文件不存在: {video_path}")
+ return False
+
+ if output_dir and not os.path.exists(output_dir):
+ os.makedirs(output_dir, exist_ok=True)
+
+ # 模拟进度
+ for progress in range(0, 101, 10):
+ if progress_callback:
+ progress_callback(progress, f"处理中 {progress}%")
+ time.sleep(0.5) # 模拟处理时间
+
+ # 创建一个简单的HTML报告
+ if output_dir:
+ html_path = os.path.join(output_dir, "summary.html")
+ with open(html_path, "w", encoding="utf-8") as f:
+ f.write(f"""
+
+
+
+
+ 简化处理报告
+
+
+
+ 简化视频处理报告
+ 视频文件: {os.path.basename(video_path)}
+ 注意: 这是一个简化的处理报告,完整功能未启用。
+
+
+ """)
+
+ return True
+
+# 配置日志
+log_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'server.log')
+logger = logging.getLogger('server')
+logger.setLevel(logging.DEBUG)
+
+# 创建文件处理器,指定编码为utf-8
+file_handler = logging.FileHandler(log_file, encoding='utf-8', mode='w')
+file_handler.setLevel(logging.DEBUG)
+
+# 创建控制台处理器
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.DEBUG)
+
+# 创建格式化器
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+file_handler.setFormatter(formatter)
+console_handler.setFormatter(formatter)
+
+# 添加处理器到logger
+logger.addHandler(file_handler)
+logger.addHandler(console_handler)
+
+# 确保日志文件存在
+if not os.path.exists(log_file):
+ with open(log_file, 'w', encoding='utf-8') as f:
+ f.write('')
+
+app = Flask(__name__)
+CORS(app) # 启用CORS支持
+
+# 配置上传文件夹
+UPLOAD_FOLDER = os.path.abspath('uploads')
+OUTPUT_FOLDER = os.path.abspath('output')
+TASK_STATUS_FILE = os.path.join(OUTPUT_FOLDER, 'task_status.json')
+
+# 创建必要的目录
+for folder in [UPLOAD_FOLDER, OUTPUT_FOLDER]:
+ if not os.path.exists(folder):
+ os.makedirs(folder)
+ logger.info(f"创建目录: {folder}")
+
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['OUTPUT_FOLDER'] = OUTPUT_FOLDER
+app.config['MAX_CONTENT_LENGTH'] = 512 * 1024 * 1024 # 增加上传大小限制到512MB
+
+ALLOWED_EXTENSIONS = {'mp4', 'avi', 'mov'}
+
+# 创建线程池,限制并发处理任务数
+MAX_WORKERS = 2 # 最多同时处理2个视频
+executor = ThreadPoolExecutor(max_workers=MAX_WORKERS)
+
+# 初始化任务状态存储
+tasks = {}
+if os.path.exists(TASK_STATUS_FILE):
+ try:
+ with open(TASK_STATUS_FILE, 'r', encoding='utf-8') as f:
+ tasks = json.load(f)
+ except:
+ logger.error("无法读取任务状态文件,创建新的状态跟踪")
+ tasks = {}
+
+def save_task_status():
+ """保存任务状态到文件"""
+ try:
+ with open(TASK_STATUS_FILE, 'w', encoding='utf-8') as f:
+ json.dump(tasks, f, ensure_ascii=False, indent=2)
+ except Exception as e:
+ logger.error(f"保存任务状态失败: {str(e)}")
+
+def get_free_space_gb(dirname):
+ """获取指定目录所在磁盘的可用空间(GB)"""
+ free_bytes = ctypes.c_ulonglong(0)
+ ctypes.windll.kernel32.GetDiskFreeSpaceExW(
+ ctypes.c_wchar_p(dirname), None, None, ctypes.pointer(free_bytes)
+ )
+ return free_bytes.value / 1024 / 1024 / 1024 # 转换为GB
+
+def allowed_file(filename):
+ return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+def process_video_async(task_id, filepath, filename):
+ """异步处理视频的函数"""
+ try:
+ # 更新任务状态为处理中
+ tasks[task_id]['status'] = 'processing'
+ tasks[task_id]['progress'] = 10
+ save_task_status()
+
+ # 处理视频文件
+ logger.info(f"[任务 {task_id}] 开始处理视频: {filename}")
+
+ # 创建任务专属的输出目录
+ task_output_dir = os.path.join(app.config['OUTPUT_FOLDER'], task_id)
+ os.makedirs(task_output_dir, exist_ok=True)
+
+ # 设置进度回调
+ def progress_callback(progress, message=None):
+ tasks[task_id]['progress'] = progress
+ if message:
+ tasks[task_id]['message'] = message
+ save_task_status()
+
+ # 处理视频
+ start_time = time.time()
+ result = main_process(filepath, output_dir=task_output_dir, progress_callback=progress_callback)
+ end_time = time.time()
+
+ if result:
+ # 处理成功
+ tasks[task_id]['status'] = 'completed'
+ tasks[task_id]['progress'] = 100
+ tasks[task_id]['message'] = '处理完成'
+
+ # 修改报告路径格式,确保前端能正确访问
+ # 检查报告文件是否存在
+ html_path = os.path.join(task_output_dir, "summary.html")
+ if os.path.exists(html_path):
+ # 使用专门的查看报告路由
+ tasks[task_id]['result_path'] = f"/view_report/{task_id}"
+ logger.info(f"[任务 {task_id}] 报告预览路径: {tasks[task_id]['result_path']}")
+ else:
+ # 查找是否有备用报告文件
+ backup_files = [f for f in os.listdir(task_output_dir) if f.endswith('.html')]
+ if backup_files:
+ tasks[task_id]['result_path'] = f"/view_report/{task_id}"
+ logger.info(f"[任务 {task_id}] 使用备用报告预览路径: {tasks[task_id]['result_path']}")
+ else:
+ # 创建一个简单的状态页面
+ simple_html = os.path.join(task_output_dir, "status.html")
+ with open(simple_html, "w", encoding="utf-8") as f:
+ f.write(f"""
+
+
+
+
+ 处理状态
+
+
+
+ 视频处理完成
+
+
视频已成功处理,但未找到标准报告文件。
+
请检查服务器日志以获取详细信息。
+
处理时间: {round(end_time - start_time, 2)}秒
+
+
+
+ """)
+ tasks[task_id]['result_path'] = f"/view_report/{task_id}"
+ logger.info(f"[任务 {task_id}] 创建状态页面: {simple_html}")
+
+ tasks[task_id]['process_time'] = round(end_time - start_time, 2)
+ logger.info(f"[任务 {task_id}] 视频处理成功,耗时: {tasks[task_id]['process_time']}秒")
+ else:
+ # 处理失败
+ tasks[task_id]['status'] = 'failed'
+ tasks[task_id]['message'] = '处理失败,请检查日志'
+ logger.error(f"[任务 {task_id}] 视频处理失败")
+
+ save_task_status()
+
+ except Exception as e:
+ # 处理异常
+ logger.error(f"[任务 {task_id}] 处理过程中发生异常: {str(e)}")
+ try:
+ logger.error(traceback.format_exc())
+ except Exception:
+ logger.error("无法获取详细错误信息,traceback模块不可用")
+ tasks[task_id]['status'] = 'failed'
+ tasks[task_id]['message'] = f'错误: {str(e)}'
+ save_task_status()
+
+@app.route('/')
+def index():
+ return render_template('index.html')
+
+@app.route('/output/')
+def serve_output(filename):
+ """提供输出文件下载"""
+ try:
+ logger.info(f"请求输出文件: {filename}")
+
+ # 检查文件名是否包含任务ID路径
+ if '/' in filename:
+ task_id = filename.split('/')[0]
+ file_path = '/'.join(filename.split('/')[1:])
+ task_output_dir = os.path.join(app.config['OUTPUT_FOLDER'], task_id)
+
+ if not os.path.exists(task_output_dir):
+ logger.error(f"任务输出目录不存在: {task_output_dir}")
+ return f"任务输出目录不存在: {task_id}", 404
+
+ file_full_path = os.path.join(task_output_dir, file_path)
+ if not os.path.exists(file_full_path):
+ logger.error(f"请求的文件不存在: {file_full_path}")
+ return f"请求的文件不存在: {file_path}", 404
+
+ logger.info(f"提供文件: {file_full_path}")
+ return send_from_directory(task_output_dir, file_path)
+ else:
+ # 处理不包含任务ID的直接文件
+ output_dir = app.config['OUTPUT_FOLDER']
+ file_full_path = os.path.join(output_dir, filename)
+
+ if not os.path.exists(file_full_path):
+ logger.error(f"请求的文件不存在: {file_full_path}")
+ return f"请求的文件不存在: {filename}", 404
+
+ logger.info(f"提供文件: {file_full_path}")
+ return send_from_directory(output_dir, filename)
+ except Exception as e:
+ logger.error(f"提供输出文件时出错: {str(e)}")
+ return f"服务器错误: {str(e)}", 500
+
+@app.route('/view_report/')
+def view_report(task_id):
+ """提供报告预览页面"""
+ try:
+ if task_id not in tasks:
+ return "任务不存在", 404
+
+ task = tasks[task_id]
+ if task['status'] != 'completed':
+ return f"任务尚未完成,当前状态: {task['status']}", 400
+
+ task_output_dir = os.path.join(app.config['OUTPUT_FOLDER'], task_id)
+ if not os.path.exists(task_output_dir):
+ return f"任务输出目录不存在: {task_id}", 404
+
+ # 查找HTML报告文件
+ html_files = [f for f in os.listdir(task_output_dir) if f.endswith('.html')]
+ if not html_files:
+ return f"未找到任务 {task_id} 的报告文件", 404
+
+ # 优先使用summary.html
+ if "summary.html" in html_files:
+ report_file = "summary.html"
+ else:
+ report_file = html_files[0]
+
+ logger.info(f"[任务 {task_id}] 提供报告预览: {report_file}")
+ return send_from_directory(task_output_dir, report_file)
+ except Exception as e:
+ logger.error(f"提供报告预览时出错: {str(e)}")
+ return f"服务器错误: {str(e)}", 500
+
+@app.route('/results//')
+def serve_results_file(job_id, filename):
+ """提供任务结果文件"""
+ try:
+ task_output_dir = os.path.join(app.config['OUTPUT_FOLDER'], job_id)
+ if not os.path.exists(task_output_dir):
+ return f"任务目录不存在: {job_id}", 404
+
+ file_path = os.path.join(task_output_dir, filename)
+ if not os.path.exists(file_path):
+ return f"请求的文件不存在: {filename}", 404
+
+ logger.info(f"提供结果文件: {file_path}")
+ return send_from_directory(task_output_dir, filename)
+ except Exception as e:
+ logger.error(f"提供结果文件时出错: {str(e)}")
+ return f"服务器错误: {str(e)}", 500
+
+@app.route('/api/upload', methods=['POST'])
+def upload_file():
+ """处理文件上传"""
+ try:
+ # 检查是否有文件
+ if 'file' not in request.files:
+ logger.error("上传请求中没有文件")
+ return jsonify({'success': False, 'message': '没有找到文件'}), 400
+
+ file = request.files['file']
+
+ # 检查文件名是否为空
+ if file.filename == '':
+ logger.error("上传文件名为空")
+ return jsonify({'success': False, 'message': '空文件名'}), 400
+
+ # 检查文件类型
+ if not allowed_file(file.filename):
+ logger.error(f"不支持的文件类型: {file.filename}")
+ return jsonify({'success': False, 'message': f'不支持的文件类型,请上传 {", ".join(ALLOWED_EXTENSIONS)} 格式的视频'}), 400
+
+ # 检查磁盘空间
+ free_space = get_free_space_gb(UPLOAD_FOLDER)
+ if free_space < 5: # 少于5GB空间警告
+ logger.warning(f"磁盘空间不足: {free_space:.2f}GB")
+ return jsonify({'success': False, 'message': f'磁盘空间不足,仅剩 {free_space:.2f}GB'}), 400
+
+ # 生成安全的文件名
+ filename = secure_filename(file.filename)
+
+ # 生成任务ID
+ task_id = str(uuid.uuid4())
+
+ # 创建上传路径
+ upload_dir = os.path.join(app.config['UPLOAD_FOLDER'], task_id)
+ os.makedirs(upload_dir, exist_ok=True)
+
+ # 保存文件
+ file_path = os.path.join(upload_dir, filename)
+ file.save(file_path)
+ logger.info(f"文件 {filename} 上传成功,路径: {file_path}")
+
+ # 创建任务
+ current_time = time.strftime('%Y-%m-%d %H:%M:%S')
+ tasks[task_id] = {
+ 'id': task_id,
+ 'filename': filename,
+ 'original_filename': file.filename,
+ 'status': 'pending',
+ 'message': '等待处理',
+ 'progress': 0,
+ 'create_time': current_time,
+ 'update_time': current_time,
+ 'file_path': file_path
+ }
+ save_task_status()
+
+ # 异步处理视频
+ executor.submit(process_video_async, task_id, file_path, filename)
+
+ # 返回任务ID
+ return jsonify({
+ 'success': True,
+ 'message': '文件上传成功,开始处理',
+ 'task_id': task_id
+ })
+
+ except Exception as e:
+ logger.error(f"上传处理失败: {str(e)}")
+ try:
+ logger.error(traceback.format_exc())
+ except Exception:
+ pass
+ return jsonify({'success': False, 'message': f'上传失败: {str(e)}'}), 500
+
+@app.route('/api/tasks', methods=['GET'])
+def get_all_tasks():
+ """获取所有任务状态"""
+ return jsonify({'tasks': list(tasks.values())})
+
+@app.route('/api/tasks/', methods=['GET'])
+def get_task_status(task_id):
+ """获取特定任务状态"""
+ if task_id in tasks:
+ return jsonify({'task': tasks[task_id]})
+ else:
+ return jsonify({'success': False, 'message': '任务不存在'}), 404
+
+@app.route('/api/tasks//cancel', methods=['POST'])
+def cancel_task(task_id):
+ """取消任务"""
+ try:
+ if task_id not in tasks:
+ return jsonify({'success': False, 'message': '任务不存在'}), 404
+
+ task = tasks[task_id]
+
+ # 只能取消待处理或处理中的任务
+ if task['status'] not in ['pending', 'processing']:
+ return jsonify({'success': False, 'message': f'无法取消状态为 {task["status"]} 的任务'}), 400
+
+ # 标记任务为已取消
+ task['status'] = 'cancelled'
+ task['message'] = '任务已取消'
+ task['update_time'] = time.strftime('%Y-%m-%d %H:%M:%S')
+ save_task_status()
+
+ # 注意:这里没有实际终止正在运行的处理过程
+ # 如果需要,应该实现一个线程安全的取消机制
+
+ return jsonify({'success': True, 'message': '任务已取消'})
+
+ except Exception as e:
+ logger.error(f"取消任务失败: {str(e)}")
+ return jsonify({'success': False, 'message': f'取消任务失败: {str(e)}'}), 500
+
+@app.route('/api/tasks/', methods=['DELETE'])
+def delete_task(task_id):
+ """删除任务和相关文件"""
+ try:
+ if task_id not in tasks:
+ return jsonify({'success': False, 'message': '任务不存在'}), 404
+
+ # 删除上传文件
+ upload_dir = os.path.join(app.config['UPLOAD_FOLDER'], task_id)
+ if os.path.exists(upload_dir):
+ try:
+ shutil.rmtree(upload_dir)
+ logger.info(f"删除上传文件目录: {upload_dir}")
+ except Exception as e:
+ logger.error(f"删除上传文件目录失败: {str(e)}")
+
+ # 删除输出文件
+ output_dir = os.path.join(app.config['OUTPUT_FOLDER'], task_id)
+ if os.path.exists(output_dir):
+ try:
+ shutil.rmtree(output_dir)
+ logger.info(f"删除输出文件目录: {output_dir}")
+ except Exception as e:
+ logger.error(f"删除输出文件目录失败: {str(e)}")
+
+ # 从任务列表中删除
+ del tasks[task_id]
+ save_task_status()
+
+ return jsonify({'success': True, 'message': '任务已删除'})
+
+ except Exception as e:
+ logger.error(f"删除任务失败: {str(e)}")
+ return jsonify({'success': False, 'message': f'删除任务失败: {str(e)}'}), 500
+
+@app.route('/api/tasks//retry', methods=['POST'])
+def retry_task(task_id):
+ """重试任务"""
+ try:
+ if task_id not in tasks:
+ return jsonify({'success': False, 'message': '任务不存在'}), 404
+
+ task = tasks[task_id]
+
+ # 只能重试失败或取消的任务
+ if task['status'] not in ['failed', 'cancelled']:
+ return jsonify({'success': False, 'message': f'无法重试状态为 {task["status"]} 的任务'}), 400
+
+ # 检查文件是否存在
+ if not os.path.exists(task['file_path']):
+ return jsonify({'success': False, 'message': '原始文件不存在,无法重试'}), 400
+
+ # 更新任务状态
+ task['status'] = 'pending'
+ task['message'] = '等待重新处理'
+ task['progress'] = 0
+ task['update_time'] = time.strftime('%Y-%m-%d %H:%M:%S')
+
+ # 删除旧的输出目录(如果存在)
+ output_dir = os.path.join(app.config['OUTPUT_FOLDER'], task_id)
+ if os.path.exists(output_dir):
+ try:
+ shutil.rmtree(output_dir)
+ logger.info(f"清理旧的输出目录: {output_dir}")
+ except Exception as e:
+ logger.error(f"清理旧的输出目录失败: {str(e)}")
+
+ save_task_status()
+
+ # 重新提交处理任务
+ executor.submit(process_video_async, task_id, task['file_path'], task['filename'])
+
+ return jsonify({'success': True, 'message': '任务已重新提交'})
+
+ except Exception as e:
+ logger.error(f"重试任务失败: {str(e)}")
+ return jsonify({'success': False, 'message': f'重试任务失败: {str(e)}'}), 500
+
+if __name__ == "__main__":
+ try:
+ logger.info("正在启动服务器...")
+ logger.info(f"工作目录: {os.getcwd()}")
+ logger.info(f"静态文件目录: {os.path.abspath('static')}")
+ logger.info(f"模板文件目录: {os.path.abspath('templates')}")
+
+ if not os.path.exists(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'templates')):
+ logger.error("模板目录不存在!")
+
+ if not os.path.exists(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'templates', 'index.html')):
+ logger.error("index.html 不存在!")
+
+ app.run(host='0.0.0.0', port=5001, debug=False)
+ except Exception as e:
+ logger.error(f"服务器启动失败: {str(e)}")
+ logger.error(traceback.format_exc())
\ No newline at end of file