Django-如何使用ForeignKey的存在作为条件来设置模型的字段？

如何解决Django-如何使用ForeignKey的存在作为条件来设置模型的字段？

我正在尝试创建一个Django应用，其中有两个主要模型-Team和Person。 一个人可以将应用程序用作用户，也可以属于团队，在这种情况下，用户将成为团队。

class Team(models.Model):
    name = models.CharField(max_length=50)
    team_user = models.ForeignKey(
        User,on_delete = models.CASCADE,null = True,blank = True )

class Person(models.Model):    
    person_user = models.ForeignKey(User,blank = True)
    team = models.ForeignKey(
        Team,blank = True )

我正在尝试开发这样的文件结构：

/root
  /Team_01
    /Person_01 (non-User)
      /files.csv
    /Person_02 (non-User)
      /files.csv
  /Person_03 (User)
    /files.csv

我已经阅读了文档，并试图对this帖子中Evgeni Shudzel的评论进行处理，但是我的问题是我不知道如何为Person模型设置文件路径。如果人员是团队成员，则路径为“ / team_id / person_id”，否则为“ / team_id”部分。我想到的伪代码：

if Person.team exists:
    files = models.FileField(upload_to="TEAM_ID/PERSON_ID/")
else:
    files = models.FileField(upload_to="PERSON_ID/")

如何实现此伪代码？谢谢！

解决方法

您走的路正确。 upload_to可以指向一个函数。

def user_directory_path(instance: Person,filename): 
    # file will be uploaded to MEDIA_ROOT / your path 
    if instance.team:  # 
        return f"team_{instance.team.id}/user_{instance.id}"
    else:
        return f"person_{instance.id}"

# ...
class Person(models.Model): 
    files = models.FileField(upload_to = user_directory_path)