Django：将CharField改为选择字段后自动更新历史数据

需求说明 #

原本用CharField收集用户爱好，现在将该字段改为带HOBBY_CHOICES选项的选择字段，需要自动处理历史数据：

• 原数据为"skiing"、"football"的，更新为"sport"
• 其余所有值统一更新为"other"

解决方案 #

1. 处理历史数据：自定义Django迁移脚本 #

历史数据的批量更新适合通过Django迁移脚本完成，这是一次性的操作，不会影响后续业务逻辑。

步骤：

• 先确保模型已修改为带选项的CharField（即你提供的abstract_models.py中的代码）
• 生成空迁移文件：

  python manage.py makemigrations --empty your_app_name
  

• 打开生成的迁移文件，添加数据更新逻辑：

from django.db import migrations

def update_hobby_data(apps, schema_editor):
    ReferencesProfileBase = apps.get_model('your_app_name', 'ReferencesProfileBase')
    SPORT_VALUES = ["skiing", "football"]
    
    # 批量更新属于sport类的旧数据
    ReferencesProfileBase.objects.filter(hobby__in=SPORT_VALUES).update(hobby="sport")
    # 批量更新剩余的旧数据为other（排除已更新的和空值）
    ReferencesProfileBase.objects.exclude(hobby__in=["sport", None, ""]).update(hobby="other")

class Migration(migrations.Migration):

    dependencies = [
        ('your_app_name', '之前的迁移文件名'),  # 替换为实际的依赖迁移文件名
    ]

    operations = [
        migrations.RunPython(update_hobby_data),
    ]

• 执行迁移：

  python manage.py migrate
  

2. 处理新提交数据：表单/模型层自动转换 #

如果后续用户提交的表单仍可能传入旧的爱好值（比如前端未同步更新），可以在以下位置处理转换逻辑：

方式一：表单的clean方法

在处理提交的表单中，添加字段清洗逻辑：

from django import forms
from .models import ReferencesProfileBase

class ReferencesProfileForm(forms.ModelForm):
    class Meta:
        model = ReferencesProfileBase
        fields = ['hobby']
    
    def clean_hobby(self):
        hobby = self.cleaned_data.get('hobby')
        SPORT_VALUES = ["skiing", "football"]
        if hobby in SPORT_VALUES:
            return "sport"
        elif hobby and hobby not in dict(ReferencesProfileBase.HOBBY_CHOICES).keys():
            return "other"
        return hobby

方式二：模型的save方法

在模型中重写save方法，确保无论从哪里修改数据，都会自动转换：

class ReferencesProfileBase(models.Model, UsernameNaturalKeyMixin):
    HOBBY_CHOICES = (
        ('sport', _('sport')),
        ('music', _('music')),
        ('travel', _('travel')),
        ('other', _('other')),
    )
    # 其他字段定义...
    hobby = models.CharField(
        _('Hobby'),
        max_length=250,
        choices=HOBBY_CHOICES,
        blank=True,
        null=True,
    )

    def save(self, *args, **kwargs):
        SPORT_VALUES = ["skiing", "football"]
        if self.hobby in SPORT_VALUES:
            self.hobby = "sport"
        elif self.hobby and self.hobby not in dict(self.HOBBY_CHOICES).keys():
            self.hobby = "other"
        super().save(*args, **kwargs)

注意事项 #

• 迁移脚本执行前建议先备份数据库，避免数据意外丢失
• 模型中HOBBY_CHOICES的写法建议修正为('sport', _('sport'))，确保显示文本支持翻译
• 若字段允许空值，注意在逻辑中排除None和空字符串，避免误转换

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