如何在Django的choices中自动读取CSV文件，无需逐项调用？

问题概述 #

需要实现函数为Django Model生成choices选项，当前需手动读取CSV内容逐个添加选项，希望通过循环自动读取CSV完成生成，尝试for循环未成功。现有代码如下：

def Equipe():
    with open("controle_pav/static/texto/equipes.csv", 'r') as arquivo:
        equipes = arquivo.read()
            EQUIPE = (                
                ('equipes', equipes.split()[0]),
                ('equipes', equipes.split()[1]),
                ('equipes', equipes.split()[2]),
                        )
        return EQUIPE

可行方案 #

原代码问题在于直接按空格分割整个文件内容，且手动指定索引，无法适配CSV的行结构。推荐使用Python内置csv模块逐行读取，自动生成符合Django Choices格式的元组。

优化代码 #

import csv

def get_equipe_choices():
    choices = []
    # 打开CSV文件，指定编码避免乱码
    with open("controle_pav/static/texto/equipes.csv", 'r', encoding='utf-8') as arquivo:
        csv_reader = csv.reader(arquivo)
        # 遍历每行数据生成选项
        for row in csv_reader:
            # 跳过空行
            if not row:
                continue
            # 按CSV结构调整索引，格式为(实际存储值, 页面显示文本)
            choices.append((row[0].strip(), row[1].strip()))
    # 转换为元组返回（Django接受列表或元组格式的choices）
    return tuple(choices)

核心要点 #

• 用csv.reader处理CSV文件，能正确解析包含特殊字符的字段，比手动分割更稳定。
• 循环遍历自动生成所有选项，无需手动指定条目数量，适配任意行数的CSV。
• 增加strip()清理字段前后空白，避免无效空格影响选项显示。
• 跳过空行，防止生成无效选项。

Django Model中调用示例 #

from django.db import models
from .utils import get_equipe_choices  # 假设函数存放在utils.py中

class ExampleModel(models.Model):
    equipe = models.CharField(max_length=100, choices=get_equipe_choices())

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