bank-statement-import/account_bank_statement_import_txt_xlsx/models/account_bank_statement_import.py

# Copyright 2020 CorporateHub (https://corporatehub.eu)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import api, fields, models

import logging
_logger = logging.getLogger(__name__)


class AccountBankStatementImport(models.TransientModel):
    _inherit = 'account.bank.statement.import'

    def _get_default_mapping_id(self):
        return self.env["account.journal"].browse(
            self.env.context.get('journal_id')).default_sheet_mapping_id

    sheet_mapping_id = fields.Many2one(
        string='Sheet mapping',
        comodel_name='account.bank.statement.import.sheet.mapping',
        default=_get_default_mapping_id,
    )

    @api.multi
    def _parse_file(self, data_file):
        self.ensure_one()
        try:
            Parser = self.env['account.bank.statement.import.sheet.parser']
            return Parser.parse(
                self.sheet_mapping_id,
                data_file,
                self.filename
            )
        except:
            if self.env.context.get(
                    'account_bank_statement_import_txt_xlsx_test'):
                raise
            _logger.warning('Sheet parser error', exc_info=True)
        return super()._parse_file(data_file)