Django 表单的数据导入和导出

2023-04-11 00:00:00 导入表单导出

Django 表单的数据导入和导出是指将表单中的数据从一个文件（如 CSV、Excel）导入到数据库中，或从数据库中导出到一个文件中。这对于数据的备份、恢复和迁移都非常有用。

首先，我们需要安装一个 Python 库来处理文件格式，比如 csv 和 openpyxl。可以通过 pip 安装：

pip install csv
pip install openpyxl

接下来，在 Django 中，我们可以创建一个表单，定义想要导入或导出的字段。举个例子：

from django import forms

class MyForm(forms.Form):
    name = forms.CharField(max_length=50)
    email = forms.EmailField()
    age = forms.IntegerField()

这个表单有三个字段：name、email、age。接下来，我们需要在视图中实现导入和导出功能。首先是导入：

import csv
from django.shortcuts import render
from django.http import HttpResponse
from .forms import MyForm

def import_csv(request):
    if request.method == 'POST':
        form = MyForm(request.POST, request.FILES)
        if form.is_valid():
            file = request.FILES['file']
            reader = csv.reader(file)
            for row in reader:
                name, email, age = row
                # 处理数据
            return HttpResponse('Imported successfully')
    else:
        form = MyForm()
    return render(request, 'import.html', {'form': form})

这个导入视图接收一个 POST 请求，表单包含一个名为 file 的文件字段。视图先检查表单是否有效，如果有效则读取上传的文件并迭代读取每一行，将数据存入数据库中。最后返回一个成功的 HTTP 响应。

接下来是导出视图：

import csv
from django.http import HttpResponse
from .models import Person

def export_csv(request):
    response = HttpResponse(content_type='text/csv')
    response['Content-Disposition'] = 'attachment; filename="mydata.csv"'

    writer = csv.writer(response)
    writer.writerow(['Name', 'Email', 'Age'])

    persons = Person.objects.all()
    for person in persons:
        writer.writerow([person.name, person.email, person.age])

    return response

这个导出视图创建一个 CSV 文件，向其中写入列标题和数据库中的数据。最后，返回一个 HTTP 响应，将 CSV 文件作为附件提供给客户端下载。

以上是 Django 表单数据导入和导出的基本方法，可以根据实际需要进行调整和扩展。

相关文章