原来有的导入功能相信很多小伙伴对其功能不是很满意，不过没关系，我们可以二开啊，把它的功能改造成你想要的样子，接下来让我们看看怎么办吧
例如我想把员工导入功能中添加上用户同步注册功能
首先，我要找到原模块中导入时调用的模型——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