繁体   English   中英

如何在Django模型中获取当前用户?

[英]How to get current user in django model?

我正在尝试创建一个动态文件夹来为每个用户存储一些csv文件。 我有DemoUser和文档。 这是一对一的关系。 我想将用户的所有文档存储在用户ID命名的文件夹中。

但是我不知道该怎么做。 我找不到所有者的名字。

这是我的代码:

models.py

def get_upload_path(instance, filename):
    now = timezone.now().strftime("%Y_%m_%d")
    if(instance.owner.id is not None):
        return 'graphs/documents/{0}/email/' + now +'_{1}'.format(instance.owner.id, filename)

class Document(models.Model):
    docfile = models.FileField(upload_to=get_upload_path, validators=[import_document_validator])
    owner = models.ForeignKey(DemoUser)
    date_published = models.DateTimeField(_('date published'), default=timezone.now)

views.py:

if request.method == 'POST':
    form = DocumentForm(request.POST, request.FILES)
    if form.is_valid():
        newdoc = Document(owner=DemoUser.objects.get(id=request.user.id))
        newdoc.save()
        newdoc.docfile = request.FILES['docfile']
        newdoc.save()

该文件已上传,但地址为:

图形/文档/ {0} /email/2015_10_07_MOCK_DATA_2_qyt9xGe.csv

它没有获得instance.owner.id值。 我尝试在添加docfile之前保存文档,以尝试为该对象设置所有者并获取owner.id值,但没有成功。 你能帮我么?

非常感谢!

它应该是:

return 'graphs/documents/{0}/email/{1}_{2}'.format(instance.owner.id, now, filename)

我发现为每个用户创建用于存储文件的动态目录的简单方法是为模型创建一个新方法,该方法将一个字符串分配给代表用户名的实例。 这仅适用于像这样的简单情况,在这种情况下,您需要当前用户作为字符串来帮助格式化文件路径。

#models.py
def get_upload_path(instance, filename):
    now = timezone.now().strftime("%Y_%m_%d")
    return 'graphs/documents/{0}/email/{1}_{2}'.format(instance.owner, now, filename)

class Document(models.Model):
    docfile = models.FileField(upload_to=get_upload_path)
    date_published = models.DateTimeField(_('date published'), default=timezone.now)

    def keep_owner(self,owner):
        self.owner = owner

#views.py
if request.method == 'POST':
    form = DocumentForm(request.POST, request.FILES)
    if form.is_valid():
        newdoc = form.save(commit=False)
        newdoc.keep_owner(request.user)
        newdoc.save()

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM