odoo导入功能二开

Posted 2021-02-13 chenshuquan

原来有的导入功能相信很多小伙伴对其功能不是很满意，不过没关系，我们可以二开啊，把它的功能改造成你想要的样子，接下来让我们看看怎么办吧

• 例如我想把员工导入功能中添加上用户同步注册功能

• 首先，我要找到原模块中导入时调用的模型——importUser.py文件

• 找到这个文件之后打开 HtkjtImport 类，该类继承import基础模块

    class HtkjtImport(models.TransientModel):
        _inherit = "base_import.import"

• 在该类下面找到 do 方法，将其重写

• 当然咯，重写该方法或者其他方法都可以，不一定就选择这个

• 代码风格规范一点的也可以自己新建一个方法，然后调用就可以了

  @api.multi
  def do(self, fields, columns, options, dryrun=False):

    self.ensure_one()
    self._cr.execute('SAVEPOINT import')
  
    try:
        data, import_fields = self._convert_import_data(fields, options)
        # Parse date and float field
        data = self._parse_import_data(data, import_fields, options)
    except ValueError as error:
        return {
            'messages': [{
                'type': 'error',
                'message': pycompat.text_type(error),
                'record': False,
            }]
        }
  
    _logger.info('importing %d rows...', len(data))
  
    name_create_enabled_fields = options.pop('name_create_enabled_fields', {})
    model = self.env[self.res_model].with_context(import_file=True,
                                                  name_create_enabled_fields=name_create_enabled_fields)
    import_result = model.load(import_fields, data)
    _logger.info('done')
  
    # If transaction aborted, RELEASE SAVEPOINT is going to raise
    # an InternalError (ROLLBACK should work, maybe). Ignore that.
    try:
        if dryrun:
            self._cr.execute('ROLLBACK TO SAVEPOINT import')
            # cancel all changes done to the registry/ormcache
            self.pool.reset_changes()
        else:
            self._cr.execute('RELEASE SAVEPOINT import')
    except psycopg2.InternalError:
        pass
  
    # 导入成功后更新插入映射列
    if import_result['ids'] and options.get('headers'):
        BaseImportMapping = self.env['base_import.mapping']
        for index, column_name in enumerate(columns):
            if column_name:
                # Update to latest selected field
                exist_records = BaseImportMapping.search(
                    [('res_model', '=', self.res_model), ('column_name', '=', column_name)])
                if exist_records:
                    exist_records.write({'field_name': fields[index]})
                else:
                    BaseImportMapping.create({
                        'res_model': self.res_model,
                        'column_name': column_name,
                        'field_name': fields[index]
                    })
  
    for da in data:
        print(da[1])
        print(da[3])
  
        data = {
            'name': da[1],
            'display_name': da[1],
            'email': da[3],
        }
        partner = self.env['res.partner'].create(data)
        print(partner.id)
  
        data = {
            'login': da[3],
            'password': 123456,
            'partner_id': partner.id,
        }
        user = self.env['res.users'].create(data)
  
    return import_result