django_celery_results记录泄露凭证：如何从任务结果中排除敏感参数？

以下是几个无需加密的更优方案：

1. 避免直接传递敏感参数，改用临时缓存中转 #

调用Celery任务时，不直接把用户名、密码作为参数传入，而是先将敏感凭证存入安全的临时存储（比如Django缓存，推荐用Redis等内存缓存），生成一个唯一的随机key，再把这个key作为参数传给任务：

# 调用任务前的代码
import uuid
from django.core.cache import cache

# 生成唯一key
cred_key = f"email_cred_{uuid.uuid4().hex}"
# 存入缓存，设置短过期时间（比如5分钟，足够任务执行完成）
cache.set(cred_key, {"username": "xxx", "password": "xxx"}, timeout=300)

# 调用Celery任务，只传key
send_mail_task.delay(cred_key, subject, message, recipient_list)

然后在Celery任务内部，通过key取出凭证，使用后立即删除缓存：

# Celery任务代码
from celery import shared_task
from django.core.cache import cache
from django.core.mail import send_mail

@shared_task
def send_mail_task(cred_key, subject, message, recipient_list):
    # 取出凭证
    creds = cache.get(cred_key)
    if not creds:
        raise ValueError("凭证已过期或不存在")
    # 使用凭证发送邮件
    send_mail(
        subject,
        message,
        creds["username"],
        recipient_list,
        auth_user=creds["username"],
        auth_password=creds["password"],
        fail_silently=False,
    )
    # 立即删除缓存，避免泄露
    cache.delete(cred_key)

这种方式下，django_celery_results只会记录cred_key这个非敏感参数，不会存储实际的凭证信息。

2. 自定义任务类，过滤敏感参数 #

通过自定义Celery任务类，在任务执行后修改要存储的元数据，移除敏感的kwargs字段：

from celery import Task

class SafeTask(Task):
    def after_return(self, status, retval, task_id, args, kwargs, einfo):
        # 过滤掉kwargs中的敏感字段
        if "username" in kwargs:
            kwargs["username"] = "[REDACTED]"
        if "password" in kwargs:
            kwargs["password"] = "[REDACTED]"
        # 调用父类方法，确保正常存储结果
        super().after_return(status, retval, task_id, args, kwargs, einfo)

然后在定义任务时使用这个自定义类：

@shared_task(base=SafeTask)
def send_mail_task(username, password, subject, message, recipient_list):
    send_mail(
        subject,
        message,
        username,
        recipient_list,
        auth_user=username,
        auth_password=password,
        fail_silently=False,
    )

这样存储到django_celery_results的task_kwargs里，敏感字段会被替换为占位符，实际凭证不会被记录。

3. 配置django_celery_results不存储任务参数 #

如果不需要记录任何任务参数，可以修改Celery配置，禁用任务参数的存储：

# settings.py中的Celery配置
CELERY_RESULT_EXTENDED = False

注意：这个配置会完全关闭task_args和task_kwargs的存储，如果你还需要记录其他非敏感参数，这个方案可能不适用。

内容的提问来源于stack exchange，提问作者nigel239