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

我已经阅读了文档，我正在尝试对 this post 中 Evgeni Shudzel 的评论做一些事情，但我的问题是我不知道如何设置文件的Person 模型的路径，因此如果 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/")

如何实现这个伪代码？谢谢！

Answer 1

你走的路是对的。 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)

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

Django - How can I set a Model's field using the existance of a ForeignKey as a condition?

filefield