原来有的导入功能相信很多小伙伴对其功能不是很满意,不过没关系,我们可以二开啊,把它的功能改造成你想要的样子,接下来让我们看看怎么办吧
- 例如我想把员工导入功能中添加上用户同步注册功能
- 首先,我要找到原模块中导入时调用的模型——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
原文地址:https://www.cnblogs.com/chenshuquan/p/10531862.html
时间: 2024-10-30 07:45:36