如何在Django导入导出时更新外键关联模型的字段值

我定义了两个Django模型：

class Fund(LabsManagerBudgetMixin, ActiveDateMixin): 
    ref= models.CharField(max_length=30, blank=True, verbose_name=_('Reference'))
    start_date=models.DateField(null=True, blank=True, verbose_name=_('Start Date'))
    end_date=models.DateField(null=True, blank=True, verbose_name=_('End Date'))
    
class Fund_Item(LabsManagerBudgetMixin, CachedModelMixin):
    amount=models.DecimalField(max_digits=12, decimal_places=2, verbose_name=_('Amount'), default=0, null=True)
    type=models.ForeignKey(Cost_Type, on_delete=models.CASCADE, verbose_name=_('Type'))
    fund=models.ForeignKey(Fund, on_delete=models.CASCADE, verbose_name=_('Related Fund'))

其中混入类（mixin）会添加额外字段、方法和管理器。Fund_Item的后台管理继承了ImportExportModelAdmin，其资源类通过before_import_row方法从ref字段识别关联的外键。

通过CSV文件更新Fund_Item对象时，CSV包含关联Fund对象的ref、start_date和end_date字段，希望在导入更新Fund_Item的同时，同步更新关联Fund的日期字段。

查阅Django-Import-Export文档得知，ForeignKeyWidget可用于外键关联数据的导入，但目前仅能更新当前模型的外键，无法更新外键关联模型的属性。不清楚应该重写哪个方法来实现需求，尝试自定义Field的clean和save方法时多次遇到错误：

class FundField(Field):
     def clean(self, data, **kwargs):
        query = Q(amount__gte=-1)

        refI = data.get('Ref', None)
        if refI is not None:
            query = query & Q(ref=refI)
        
        fu = Fund.objects.filter(query).first()
        return fu

class FundDateField(FundField):
    def clean(self, data, **kwargs):        
        fu = super(FundField, self).clean(data, **kwargs)
        return getattr(fu, self.widget.field, None)
    
    def save(self, obj, data, is_m2m=False, **kwargs):
        # self.attribute=self.attribute+"__"+self.widget.field
        super(FundField, self).save(obj, data, is_m2m, **kwargs)

之后根据提示实现了自定义FundDateField来处理字段检查与保存：

class FundDateField(FundField):
    def clean(self, data, **kwargs):   
        fu = super().clean(data, **kwargs)
        field = self.widget.field
        attr =getattr(fu, field, None)
        nAttr=data.get(self.column_name)
        if attr!=nAttr:
            setattr(fu, field, nAttr)
            fu.save()
        print(" - fu : "+str(fu))
        
        return fu

并将其应用到资源类的日期字段中，但仍需确认该实现方向是否正确，或是否有更优的解决方案。

解决方案分析 #

你的实现方向是可行的，但有几个可以优化的点：

1. 避免在clean方法中保存关联对象：clean方法的核心职责是验证和转换数据，而非执行持久化操作。把更新保存Fund的逻辑移到资源类的before_save_instance或after_save_instance方法中，更符合Django-Import-Export的生命周期设计。

2. 优化外键匹配逻辑：FundField中的Q(amount__gte=-1)条件看起来和ref匹配无关，建议移除或明确其用途，避免过滤掉有效数据。

3. 添加日期类型转换：CSV中的日期是字符串格式，直接赋值给DateField会报错，需要先转换为datetime.date对象（根据你的CSV日期格式调整转换规则）。

4. 复用官方组件简化代码：使用官方的ForeignKeyWidget而非自定义FundField，减少自定义代码量，降低维护成本。

优化后的示例代码 #

from import_export import resources, fields
from import_export.widgets import ForeignKeyWidget
from django.db.models import Q
import datetime

class FundForeignKeyWidget(ForeignKeyWidget):
    def get_queryset(self, value, row, *args, **kwargs):
        # 通过ref匹配Fund对象，可按需添加其他过滤条件
        return Fund.objects.filter(Q(ref=row.get('Ref')))

class FundItemResource(resources.ModelResource):
    fund = fields.Field(
        column_name='Ref',
        attribute='fund',
        widget=FundForeignKeyWidget(Fund, 'id')
    )
    
    class Meta:
        model = Fund_Item
        fields = ('id', 'amount', 'type', 'fund',)
        import_id_fields = ('id',)  # 用id作为更新的唯一标识

    def before_save_instance(self, instance, using_transactions, dry_run):
        # 获取CSV中的日期数据
        start_date_str = self.data.get('start_date')
        end_date_str = self.data.get('end_date')
        
        # 更新关联Fund的日期字段
        if instance.fund:
            # 处理start_date
            if start_date_str:
                try:
                    start_date = datetime.datetime.strptime(start_date_str, '%Y-%m-%d').date()
                    if instance.fund.start_date != start_date:
                        instance.fund.start_date = start_date
                        instance.fund.save()
                except ValueError:
                    # 处理日期格式错误，可根据需求记录日志或抛出异常
                    pass
            
            # 处理end_date
            if end_date_str:
                try:
                    end_date = datetime.datetime.strptime(end_date_str, '%Y-%m-%d').date()
                    if instance.fund.end_date != end_date:
                        instance.fund.end_date = end_date
                        instance.fund.save()
                except ValueError:
                    # 处理日期格式错误
                    pass
        return super().before_save_instance(instance, using_transactions, dry_run)

优化点说明 #

• 职责分离：clean阶段只做数据验证和外键匹配，before_save_instance阶段处理关联对象更新，符合单一职责原则。
• 错误容错：添加日期格式转换的异常捕获，避免因单条数据格式错误导致整个导入任务失败。
• 代码精简：复用官方提供的ForeignKeyWidget，减少自定义代码的维护成本。

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