时间戳

首页 > 解决方案 > 在 django 视图中手动更新模型的 FileField

django - 在 django 视图中手动更新模型的 FileField

问题描述

我有一个 csv 文件存储在模型的 FileField 中。我想从 csv 文件中删除选定的列并更新存储在项目的 FileField 中的文件。

我已成功从模型中检索文件并将其存储到 Pandas DataFrame 中。选定的列已成功从数据框中删除，但未能更新存储在项目的 FileField 中的文件。

这是我的模型：

class Project(models.Model):
    name = models.CharField(max_length=100)
    description = models.TextField()
    base_file = models.FileField(upload_to='project-files', default=True)
    date_created = models.DateTimeField(default=timezone.now)
    author = models.ForeignKey(User, on_delete=models.CASCADE)

这是假设通过 pandas 数据框操作文件的视图：

def PreprocessView(request, **kwargs):
    proj_pk = kwargs.get('pk')

    project = Project.objects.get(id=proj_pk)  
    df      = pd.read_csv(project.base_file)
    n_cols  = df.keys

    context ={}
    context['df']     = df
    context['n_cols'] = n_cols

    if request.method == 'POST':       
        checked_values = request.POST.getlist(u'predictors')
        str_check_values = ', '.join(checked_values)
        df.drop(columns=checked_values, inplace=True)
        new_df = df.to_csv(project.base_file.name)
        project.base_file = new_df
        project.save()

        messages.success(request, f'{str_check_values} has been successfuly removed!')

    return render(request, 'projects/preprocess/preprocess.html', context)

以下行返回错误并且无法更新数据库中的文件。

        new_df = df.to_csv(project.base_file.name)
        project.base_file = new_df
        project.save()

标签： django

解决方案

您必须创建一个文件对象，然后将其附加到模型。像这样

from django.core.files.base import ContentFile


if request.method == 'POST':       
    checked_values = request.POST.getlist(u'predictors')
    str_check_values = ', '.join(checked_values)
    df.drop(columns=checked_values, inplace=True)
    new_df = df.to_csv()

    updated_file = ContentFile(new_df)
    updated_file.name = "filename.csv"

    project.base_file = updated_file
    project.save()

推荐阅读