CSV导入到具有外键字段的空Django数据库

from django.db import models

class A(models.Model):
    name = models.CharField(max_length=50)

class B(models.Model):
    name = models.CharField(max_length=50)
    a = models.ForeignKey(A, on_delete=models.CASCADE)

python manage.py makemigrations
python manage.py migrate

import csv
from django.core.exceptions import ObjectDoesNotExist
from myapp.models import A, B

def import_data_from_csv(csv_file):
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        for row in reader:
            try:
                a = A.objects.get(id=row['a_id'])  # 根据外键字段的值获取关联模型对象
                b = B(name=row['name'], a=a)
                b.save()
            except ObjectDoesNotExist:
                print(f"Object with id {row['a_id']} does not exist.")

# 调用导入函数
import_data_from_csv('data.csv')