Flask 后台线程中的请求上下文问题分析与解决方案

2025-05-15 23:20:08,759 - app - ERROR - 处理出错: 保存操作日志失败
Traceback (most recent call last):
  File "/doudian-phone-tool/services/order_service.py", line 129, in save_operation_log
    auth_token, user_id = PassportService.current_user_id()
                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/doudian-phone-tool/libs/passport.py", line 33, in current_user_id
    auth_header = request.headers.get("Authorization", "")
                  ^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/site-packages/werkzeug/local.py", line 311, in __get__
    obj = instance._get_current_object()
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/site-packages/werkzeug/local.py", line 508, in _get_current_object
    raise RuntimeError(unbound_message) from None
RuntimeError: Working outside of request context.

def process_file_background(user_id):  # 接收 user_id 参数
    """后台线程处理文件"""
    from app import app
    with app.app_context():
        try:
            output_file = process_and_export_results(
                raw_results=raw_results,
                filepath=filepath,
                original_filename=original_filename,
                cookie=cookie,
                nationwide=nationwide,
                userId=user_id,  # 直接使用传入的 user_id
                receiver_email=receiver_email
            )
            logging.info(f"文件处理完成: {output_file}")
            if os.path.exists(filepath):
                os.remove(filepath)
        except Exception as e:
            logging.error(f"后台文件处理失败: {str(e)}", exc_info=True)

# 在主线程中获取 user_id，并传递给后台线程
auth_token, user_id = PassportService.current_user_id()
thread = threading.Thread(target=process_file_background, args=(user_id,))  # 传入 user_id
thread.start()

@staticmethod
def save_operation_log(
        business_type: str = '上传',
        operation_type: str = '开始上传',
        operation_content: str = None,
        operation_params: str = None,
        user_id: int = None,  # 新增可选参数
):
    """保存操作日志"""
    try:
        from app import app
        with app.app_context():
            if user_id is None:  # 如果没有传入 user_id，尝试获取
                try:
                    auth_token, user_id = PassportService.current_user_id()
                except RuntimeError:  # 如果不在请求上下文，记录匿名日志
                    user_id = 0  # 或用 None，取决于业务需求
            memberOperationLog = MemberOperationLog(
                user_id=user_id,
                business_type=business_type,
                operation_type=operation_type,
                operation_content=operation_content,
                operation_params=operation_params,
                operation_time=datetime.now(),
                create_time=datetime.now(),
                update_time=datetime.now()
            )
            db.session.add(memberOperationLog)
            db.session.commit()
    except Exception as e:
        raise MemberOperationLogError("保存操作日志失败")

from flask import copy_current_request_context

def process_file_background():
    """后台线程处理文件（携带请求上下文）"""
    @copy_current_request_context  # 复制请求上下文
    def run_in_context():
        from app import app
        with app.app_context():
            try:
                output_file = process_and_export_results(
                    raw_results=raw_results,
                    filepath=filepath,
                    original_filename=original_filename,
                    cookie=cookie,
                    nationwide=nationwide,
                    userId=user_id,
                    receiver_email=receiver_email
                )
                logging.info(f"文件处理完成: {output_file}")
                if os.path.exists(filepath):
                    os.remove(filepath)
            except Exception as e:
                logging.error(f"后台文件处理失败: {str(e)}", exc_info=True)
    run_in_context()  # 执行带上下文的函数

# 启动线程
thread = threading.Thread(target=process_file_background)
thread.start()