django-import-export修改记

环境

Django==2.0.4
django-import-export==1.0.1

需求:

  1. 导入需要修改主键,由另外一个字段来决定唯一性(替代主键);
  2. 一定要先预览再修改,即:不点【确认导入】就不修改数据库;

解决方案

需求1

使用import_id_fields配置即可,把这个设置为另外一个字段:

class TestModifyProductResource(resources.ModelResource):
    """
    以 test_field 为唯一值的导入。p_id 为数据库主键
    """
    class Meta:
        model = models.TestProduct
        skip_unchanged = True
        import_id_fields = ('test_field',)
        fields = ('p_id', 'field_2', 'field_3', 'field_4', 'test_field')
        use_transactions = False
        instance_loader_class = TestModifyProductInstanceLoader

    def import_row(self, row, *args, **kwargs):
        row['dry_run'] = kwargs.get('dry_run')
        return super().import_row(row, *args, **kwargs)

    def save_instance(self, instance, using_transactions=True, dry_run=False):
        if dry_run:
            # 如果是导入第一阶段,就不存储。
            return
        super().save_instance(instance, using_transactions, dry_run)

问题1:

  • 在mysql中,因为主键无法修改,所以在导入时必须先新增(新数据)后删除(老数据)。解决办法是自定义InstanceLoader,代码如下:
class TestModifyProductInstanceLoader(ModelInstanceLoader):
    """
    导入会出现更新主键的情况,如果已经存在相同的主键就需要先增加再删除。
    """
    def get_instance(self, row):
        # dry_run表示导入阶段,因为不同的阶段行为不一样所以需要把这个信息传递过来。
        dry_run = row.pop('dry_run', True)
        qs = self.get_queryset().filter(test_field=row['test_field'])
        if qs.count() <= 1:
            if dry_run:
                return qs.first()
            instance = super().get_instance(row)
            # 删除p_id和test_field一样的记录,不然后续程序会报错。
            qs.exclude(p_id=row['p_id']).delete()
            return instance
        else:
            # 下面备份p_id是为了提示正确:能够正确的显示更新数量。
            # 如果不需要提示正确,直接对tqs执行delete就行。
            zqs = qs.exclude(p_id=row['p_id'])
            old_p_id = zqs.first().p_id
            zqs.delete()
            instance = super().get_instance(row)
            instance.p_id = old_p_id
            return instance

需求2

按照官方的说法,只需要设置use_transactions = False即可,实际上不行,因官方实现数据缓存使用了事务来实现的。源码如下:

    def import_data(self, dataset, dry_run=False, raise_errors=False,
                    use_transactions=None, collect_failed_rows=False, **kwargs):
        #....
        if use_transactions is None:
            use_transactions = self.get_use_transactions()

        connection = connections[DEFAULT_DB_ALIAS]
        supports_transactions = getattr(connection.features, "supports_transactions", False)

        if use_transactions and not supports_transactions:
            raise ImproperlyConfigured

        using_transactions = (use_transactions or dry_run) and supports_transactions
        #....

我重写save_instance函数,强制不存储即可:

def save_instance(self, instance, using_transactions=True, dry_run=False):
    if dry_run:
        # 如果是导入第一阶段,就不存储。
        return
    super().save_instance(instance, using_transactions, dry_run)

备注: Django是根据主键是否存在来判定是使用insert还是update的,使用sql是可以修改主键的。

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页