如何将仅含部分字段的CSV文件导入Django的Album模型？

下面提供两种原生实现方式，按需选择：

方法一：Django Shell手动导入 #

步骤： #

1. 确认CSV文件列名与模型字段匹配：确保CSV包含title、release_date、artist、spotify_url这四列（字段名大小写和拼写要和模型完全一致）
2. 启动Django Shell：

python manage.py shell

3. 导入依赖模块与模型：

import csv
from your_app_name.models import Album  # 替换成你的实际app名称

4. 读取CSV并批量创建Album对象：

# 替换为你的CSV文件绝对路径
with open('/full/path/to/your/albums.csv', 'r', encoding='utf-8') as csv_file:
    reader = csv.DictReader(csv_file)
    for row in reader:
        # 仅传入CSV中存在的字段，其他字段使用模型默认值或留空
        album = Album(
            title=row['title'],
            release_date=row['release_date'],
            artist=row['artist'],
            spotify_url=row['spotify_url']
            # 说明：
            # - description 因模型设置blank=True，默认是空字符串
            # - image 会自动使用模型定义的default值 'default2.jpg'
            # - genre 需特别处理：模型未设置blank=True，若CSV无此列，要么给默认选项，要么修改模型
        )
        album.save()

关于genre字段的处理：

如果CSV里没有genre数据，你有两个选择：

• 修改模型允许为空：
  在Album模型中给genre字段添加blank=True：

  genre = models.CharField(choices=GENRE_CHOICES, max_length=20, blank=True)
  

  然后执行迁移：

  python manage.py makemigrations && python manage.py migrate
  

• 指定默认选项：
  假设你的GENRE_CHOICES里有('OTHER', '其他')这类兜底选项，创建对象时可以写：

  genre=row.get('genre', 'OTHER')
  

方法二：自定义管理命令（适合重复导入） #

如果需要多次执行导入操作，写一个自定义manage命令更方便：

1. 在你的app目录下创建management/commands文件夹（需包含__init__.py空文件），新建import_albums.py文件：

import csv
from django.core.management.base import BaseCommand
from your_app_name.models import Album  # 替换为实际app名称

class Command(BaseCommand):
    help = '批量从CSV导入专辑数据'

    def add_arguments(self, parser):
        parser.add_argument('csv_path', type=str, help='CSV文件的路径')

    def handle(self, *args, **options):
        csv_path = options['csv_path']
        with open(csv_path, 'r', encoding='utf-8') as f:
            reader = csv.DictReader(f)
            count = 0
            for row in reader:
                Album.objects.create(
                    title=row['title'],
                    release_date=row['release_date'],
                    artist=row['artist'],
                    spotify_url=row['spotify_url']
                    # 同样处理genre字段，参考方法一的说明
                )
                count += 1
        self.stdout.write(self.style.SUCCESS(f'成功导入{count}条专辑数据'))

2. 执行导入命令：

python manage.py import_albums /full/path/to/your/albums.csv

关键注意点 #

• CSV列名必须和模型字段名完全匹配，否则会出现KeyError
• 务必处理genre字段的非空约束，否则导入时会抛出数据库完整性错误
• 确保CSV文件路径正确，读取时指定正确的编码（如utf-8）避免乱码

内容的提问来源于stack exchange，提问作者NART