Django多级推荐系统报错：无法将User实例赋值给Referral.parent

核心原因：Referral.parent是关联自身的TreeForeignKey，必须传入Referral实例，你现在直接传了User实例，类型不匹配导致报错。之前用ForeignKey关联User正常是因为字段类型本身接受User实例，现在字段要求变了，逻辑也要对应调整。

具体修正步骤 #

1. 获取推荐人对应的Referral实例
   假设你的Referral模型中有一个关联User的字段（比如user = models.OneToOneField(User, on_delete=models.CASCADE)），在注册逻辑里，先通过推荐人的User对象找到其对应的Referral实例：

   # views.py 中注册逻辑的修改部分
   try:
       # 先拿到推荐人的User实例
       referrer_user = User.objects.get(username=request.POST.get('referrer_username'))
       # 通过User关联的Referral字段获取对应的Referral实例
       referrer_referral = Referral.objects.get(user=referrer_user)
   except (User.DoesNotExist, Referral.DoesNotExist):
       # 处理推荐人不存在或无对应Referral的情况，比如设为None或抛出提示
       referrer_referral = None
   
   # 创建新用户的Referral实例时，传入正确的Referral实例作为parent
   new_referral = Referral.objects.create(
       user=new_user,
       parent=referrer_referral,
       # 其他字段...
   )
   

2. 确认信号逻辑正常
   检查signals.py中User创建时生成Referral的逻辑，确保每个新创建的User都能自动生成对应的Referral实例，避免出现用户存在但无Referral的情况：

   # signals.py 示例（确保逻辑正确）
   from django.db.models.signals import post_save
   from django.dispatch import receiver
   from django.contrib.auth.models import User
   from .models import Referral
   from .utils import generate_referral_code
   
   @receiver(post_save, sender=User)
   def create_user_referral(sender, instance, created, **kwargs):
       if created:
           Referral.objects.create(
               user=instance,
               referral_code=generate_referral_code(),
               parent=None  # 新用户默认无上级
           )
   

3. 简化关联获取（可选）
   如果Referral和User是OneToOne关联，可以在模型中设置related_name，直接通过User实例获取Referral：

   # models.py 中修改关联字段
   class Referral(models.Model):
       user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='referral')
       parent = TreeForeignKey('self', on_delete=models.SET_NULL, null=True, blank=True, related_name='children')
       referral_code = models.CharField(max_length=20, unique=True)
       # 其他字段...
   

   之后在views里可以直接写：

   referrer_referral = referrer_user.referral  # 无需再用Referral.objects.get
   

关键注意事项 #

• 永远不要把非Referral类型的对象赋值给parent字段，严格匹配字段要求的实例类型。
• 处理推荐人不存在的异常场景，避免程序崩溃，同时给用户合理的提示（比如“推荐码无效”）。

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