如何在django里上传csv文件并进行入库处理的方法

下面是在Django中上传CSV文件并进行入库处理的详细攻略：

1. 添加模型

首先，我们需要在Django中创建一个模型来存储CSV文件中的数据。我们可以使用Django自带的模型（例如在models.py文件中添加一个名为CSVData的模型）：

from django.db import models

class CSVData(models.Model):
    column1 = models.CharField(max_length=255)
    column2 = models.CharField(max_length=255)
    column3 = models.CharField(max_length=255)
    # 其他需要存储的列

在这里，我们使用了CharField来存储每一列的值，但是根据实际情况需要使用不同的字段类型。

2. 创建表单

接下来，我们需要为上传CSV文件创建一个表单。在forms.py文件中添加以下表单：

from django import forms

class CSVUploadForm(forms.Form):
    file = forms.FileField()

在这里，我们使用了Django自带的文件上传表单字段FileField，用于从用户的计算机中选取一个CSV文件。

3. 创建视图

在这一步中，我们将创建负责上传CSV文件并将数据保存到数据库中的视图。在views.py文件中添加以下代码：

import csv
from django.shortcuts import render
from .forms import CSVUploadForm
from .models import CSVData

def upload_csv(request):
    if request.method == 'POST':
        form = CSVUploadForm(request.POST, request.FILES)
        if form.is_valid():
            # 读取CSV文件并将数据存入数据库
            file = request.FILES.get('file')
            if file:
                rows = csv.reader(file.read().decode().splitlines())
                for row in rows:
                    _, created = CSVData.objects.get_or_create(
                        column1=row[0], column2=row[1], column3=row[2]
                    )
    else:
        form = CSVUploadForm()
    return render(request, 'csv_upload.html', {'form': form})

在这个视图中，我们使用了csv模块来读取CSV文件的每一行，并将其存储到数据库中。这里使用了get_or_create方法来避免在插入重复的数据时出现错误。

4. 创建模板

我们需要为该视图创建一个模板，以便用户能够上传CSV文件。在templates目录下创建一个名为csv_upload.html的模板，并添加以下代码：

{% extends 'base.html' %}

{% block content %}
<h2>Upload CSV File</h2>
<form method="post" enctype="multipart/form-data">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">Upload</button>
</form>
{% endblock %}

在这个模板中，我们使用了Django自带的方式渲染表单，并在提交表单时执行视图中的代码。

5. 配置路由

最后，我们需要将该视图与URL绑定起来。在urls.py文件中添加以下代码：

from django.urls import path
from .views import upload_csv

urlpatterns = [
    path('upload-csv/', upload_csv, name='upload_csv'),
]

在这里，我们将upload_csv视图绑定到了/upload-csv/这个URL上（你可以根据实际情况更改）。

示例1：解析CSV文件中的多行数据

为了演示如何解析CSV文件中的多行数据，我们假设每一行数据的第一列都是唯一的标识符，且我们需要将其存储到一个字典中。下面是具体的代码示例：

def upload_csv(request):
    if request.method == 'POST':
        form = CSVUploadForm(request.POST, request.FILES)
        if form.is_valid():
            # 读取CSV文件并将数据存入数据库
            file = request.FILES.get('file')
            if file:
                reader = csv.DictReader(file.read().decode().splitlines())
                for row in reader:
                    data = {
                        "column2": row["column2"],
                        "column3": row["column3"],
                        # 其他需要存储的列
                    }
                    # 使用第一列作为唯一标识符存储数据到字典中
                    unique_id = row["column1"]
                    my_dict[unique_id] = data
    else:
        form = CSVUploadForm()
    return render(request, 'csv_upload.html', {'form': form})

在这个例子中，我们使用了csv.DictReader来读取CSV文件中的每一行数据并将其存储到字典中。

示例2：忽略CSV文件中的第一行

CSV文件的第一行通常是列名，我们可以使用next()方法跳过它。下面是具体的代码示例：

def upload_csv(request):
    if request.method == 'POST':
        form = CSVUploadForm(request.POST, request.FILES)
        if form.is_valid():
            # 读取CSV文件并将数据存入数据库
            file = request.FILES.get('file')
            if file:
                rows = csv.reader(file.read().decode().splitlines())
                # 跳过CSV文件的第一行
                next(rows)
                for row in rows:
                    _, created = CSVData.objects.get_or_create(
                        column1=row[0], column2=row[1], column3=row[2]
                    )
    else:
        form = CSVUploadForm()
    return render(request, 'csv_upload.html', {'form': form})

在这个例子中，我们使用了next()方法来跳过CSV文件的第一行。这对于数据集较大的CSV文件尤其有用，因为我们需要避免在读取大量数据时耗费不必要的时间和资源。

希望这个攻略能够对你有所帮助！