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

Question

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

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, on_delete = models.CASCADE, null = True, blank = True)
team = models.ForeignKey(
Team, 
on_delete = models.CASCADE,
null = True,
blank = True )

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

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

我已经阅读了文档，我试图对Evgeni Shudzel在这篇文章中的评论做些什么，但我的问题是，我不知道如何为Person模型设置文件的路径，这样，如果Person是团队的成员，则路径为"/team_ id/perse_；如果不是/team_ id"；部分被排除在外。我脑海中的伪代码：

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

如何实现此伪代码？非常感谢。

Answer 1

您的路径正确。CCD_ 1可以指向一个函数。

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)