将带有外键的CSV文件上传到Django

# models.py
from django.db import models

class Category(models.Model):
    name = models.CharField(max_length=100)

class Product(models.Model):
    name = models.CharField(max_length=100)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

# views.py
import csv
from django.http import HttpResponse
from django.shortcuts import render
from .models import Category, Product

def upload_csv(request):
    if request.method == 'POST' and request.FILES['csv_file']:
        csv_file = request.FILES['csv_file']
        if not csv_file.name.endswith('.csv'):
            return HttpResponse('Invalid file format')

        # Save CSV file to temporary directory
        with open('temp.csv', 'wb+') as destination:
            for chunk in csv_file.chunks():
                destination.write(chunk)

        # Parse CSV file and save data to database
        with open('temp.csv', 'r') as file:
            reader = csv.reader(file)
            for row in reader:
                category_name = row[0]
                product_name = row[1]

                # Find or create category
                category, _ = Category.objects.get_or_create(name=category_name)

                # Create product with foreign key to category
                Product.objects.create(name=product_name, category=category)

        return HttpResponse('CSV file uploaded successfully')

    return render(request, 'upload_csv.html')