Question

我在模型的FileField中存储了一个csv文件。我想从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()

Answer 1

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

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()

Answer 2

如果您给

DataFrame.to_csv()提供了要保存的文件路径，则返回None。所以new_df是None。

df.to_csv(project.file.name)完成后，您无需执行任何操作，文件已更改。

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

2 个答案: