# Copyright 2009-2023 Noviat. # License LGPL-3 or later (http://www.gnu.org/licenses/lgpl). import base64 import logging from copy import deepcopy from sys import exc_info from traceback import format_exception from lxml import etree from odoo import _, fields, models from odoo.exceptions import UserError from odoo.addons.base.models.res_bank import sanitize_account_number _logger = logging.getLogger(__name__) class EbicsFile(models.Model): _name = "ebics.file" _description = "Object to store EBICS Data Files" _order = "date desc" _sql_constraints = [ ( "name_uniq", "unique (name, format_id)", "This File has already been down- or uploaded !", ) ] name = fields.Char(string="Filename") data = fields.Binary(string="File", readonly=True) format_id = fields.Many2one( comodel_name="ebics.file.format", string="EBICS File Formats", readonly=True ) type = fields.Selection(related="format_id.type", readonly=True) date_from = fields.Date( readonly=True, help="'Date From' as entered in the download wizard." ) date_to = fields.Date( readonly=True, help="'Date To' as entered in the download wizard." ) date = fields.Datetime( required=True, readonly=True, help="File Upload/Download date" ) bank_statement_ids = fields.One2many( comodel_name="account.bank.statement", inverse_name="ebics_file_id", string="Generated Bank Statements", readonly=True, ) state = fields.Selection( [("draft", "Draft"), ("done", "Done")], default="draft", required=True, readonly=True, ) user_id = fields.Many2one( comodel_name="res.users", string="User", default=lambda self: self.env.user, readonly=True, ) ebics_userid_id = fields.Many2one( comodel_name="ebics.userid", string="EBICS UserID", ondelete="restrict", readonly=True, ) note = fields.Text(string="Notes") note_process = fields.Text( string="Process Notes", readonly=True, ) company_ids = fields.Many2many( comodel_name="res.company", string="Companies", readonly=True, help="Companies sharing this EBICS file.", ) def unlink(self): ff_methods = self._file_format_methods() for ebics_file in self: if ebics_file.state == "done": raise UserError(_("You can only remove EBICS files in state 'Draft'.")) # execute format specific actions ff = ebics_file.format_id.download_process_method if ff in ff_methods: if ff_methods[ff].get("unlink"): ff_methods[ff]["unlink"]() # remove bank statements ebics_file.bank_statement_ids.unlink() return super().unlink() def set_to_draft(self): return self.write({"state": "draft"}) def set_to_done(self): return self.write({"state": "done"}) def process(self): self.ensure_one() self = self.with_context(allowed_company_ids=self.env.user.company_ids.ids) self.note_process = "" ff_methods = self._file_format_methods() ff = self.format_id.download_process_method if ff in ff_methods: if ff_methods[ff].get("process"): res = ff_methods[ff]["process"]() self.state = "done" return res else: return self._process_undefined_format() def action_open_bank_statements(self): self.ensure_one() action = self.env["ir.actions.act_window"]._for_xml_id( "account.action_bank_statement_tree" ) domain = [("id", "in", self.env.context.get("statement_ids"))] action["domain"] = domain return action def button_close(self): self.ensure_one() return {"type": "ir.actions.act_window_close"} def _file_format_methods(self): """ Extend this dictionary in order to add support for extra file formats. """ res = { "cfonb120": { "process": self._process_cfonb120, "unlink": self._unlink_cfonb120, }, "camt.052": { "process": self._process_camt052, "unlink": self._unlink_camt052, }, "camt.053": { "process": self._process_camt053, "unlink": self._unlink_camt053, }, "camt.054": { "process": self._process_camt054, "unlink": self._unlink_camt054, }, "pain.002": { "process": self._process_pain002, "unlink": self._unlink_pain002, }, } return res def _check_import_module(self, module, raise_if_not_found=True): mod = ( self.env["ir.module.module"] .sudo() .search([("name", "=like", module), ("state", "=", "installed")]) ) if not mod: if raise_if_not_found: raise UserError( _( "The module to process the '%(ebics_format)s' format is not installed " "on your system. " "\nPlease install module '%(module)s'", ebics_format=self.format_id.name, module=module, ) ) return False return True def _lookup_journal(self, res, acc_number, currency_code): currency = self.env["res.currency"].search( [("name", "=ilike", currency_code)], limit=1 ) journal = self.env["account.journal"] if not currency: message = _("Currency %(cc)s not found.", cc=currency_code) res["notifications"].append({"type": "error", "message": message}) return (currency, journal) journals = self.env["account.journal"].search( [ ("type", "=", "bank"), ( "bank_account_id.sanitized_acc_number", "ilike", acc_number, ), ] ) if not journals: message = _( "No financial journal found for Account Number %(nbr)s, " "Currency %(cc)s", nbr=acc_number, cc=currency_code, ) res["notifications"].append({"type": "error", "message": message}) return (currency, journal) for jrnl in journals: journal_currency = jrnl.currency_id or jrnl.company_id.currency_id if journal_currency != currency: continue else: journal = jrnl break if not journal: message = _( "No financial journal found for Account Number %(nbr)s, " "Currency %(cc)s", nbr=acc_number, cc=currency_code, ) res["notifications"].append({"type": "error", "message": message}) return (currency, journal) def _process_download_result(self, res): statement_ids = res["statement_ids"] notifications = res["notifications"] statements = self.env["account.bank.statement"].sudo().browse(statement_ids) st_cnt = len(statement_ids) warning_cnt = error_cnt = 0 if notifications: errors = [] warnings = [] for notif in notifications: if notif["type"] == "error": error_cnt += 1 parts = [notif[k] for k in notif if k in ("message", "details")] errors.append("\n".join(parts)) elif notif["type"] == "warning": warning_cnt += 1 parts = [notif[k] for k in notif if k in ("message", "details")] warnings.append("\n".join(parts)) self.note_process += _("Process file %(fn)s results:", fn=self.name) if error_cnt: self.note_process += "\n\n" + _("Errors") + ":\n" self.note_process += "\n".join(errors) self.note_process += "\n\n" self.note_process += _("Number of errors: %(nr)s", nr=error_cnt) if warning_cnt: self.note_process += "\n\n" + _("Warnings") + ":\n" self.note_process += "\n".join(warnings) self.note_process += "\n\n" self.note_process += _("Number of warnings: %(nr)s", nr=warning_cnt) self.note_process += "\n" if st_cnt: self.note_process += "\n\n" self.note_process += _( "%(st_cnt)s bank statement%(sp)s been imported: ", st_cnt=st_cnt, sp=st_cnt == 1 and _(" has") or _("s have"), ) self.note_process += "\n" for statement in statements: self.note_process += "\n" + _( "Statement %(st)s dated %(date)s (Company: %(cpy)s)", st=statement.name, date=statement.date, cpy=statement.company_id.name, ) if statement_ids: self.sudo().bank_statement_ids = [(4, x) for x in statement_ids] company_ids = self.sudo().bank_statement_ids.mapped("company_id").ids self.company_ids = [(6, 0, company_ids)] ctx = dict(self.env.context, statement_ids=statement_ids) module = __name__.split("addons.")[1].split(".")[0] result_view = self.env.ref("%s.ebics_file_view_form_result" % module) return { "name": _("Import EBICS File"), "res_id": self.id, "view_type": "form", "view_mode": "form", "res_model": self._name, "view_id": result_view.id, "target": "new", "context": ctx, "type": "ir.actions.act_window", } def _process_cfonb120(self): import_module = "account_statement_import_fr_cfonb" self._check_import_module(import_module) res = {"statement_ids": [], "notifications": []} st_datas = self._split_cfonb(res) self._process_bank_statement_oca(res, st_datas) return self._process_download_result(res) def _unlink_cfonb120(self): """ Placeholder for cfonb120 specific actions before removing the EBICS data file and its related bank statements. """ def _split_cfonb(self, res): """ Split CFONB file received via EBICS per statement. Statements without transactions are removed. """ datas = [] file_data = base64.b64decode(self.data) lines = file_data.split(b"\n") st_lines = b"" transactions = False for line in lines: rec_type = line[0:2] currency_code = line[16:19].decode() acc_number = line[21:32].decode() st_lines += line + b"\n" if rec_type == b"04": transactions = True if rec_type == b"07": if transactions: currency, journal = self._lookup_journal( res, acc_number, currency_code ) if currency and journal: datas.append( { "acc_number": acc_number, "journal_id": journal.id, "company_id": journal.company_id.id, "data": base64.b64encode(st_lines), } ) st_lines = b"" transactions = False return datas def _process_camt052(self): import_module = "account_statement_import_camt" self._check_import_module(import_module) return self._process_camt053(self) def _unlink_camt052(self): """ Placeholder for camt052 specific actions before removing the EBICS data file and its related bank statements. """ def _process_camt054(self): import_module = "account_statement_import_camt" self._check_import_module(import_module) return self._process_camt053(self) def _unlink_camt054(self): """ Placeholder for camt054 specific actions before removing the EBICS data file and its related bank statements. """ def _process_camt053(self): """ The Odoo standard statement import is based on manual selection of a financial journal before importing the electronic statement file. An EBICS download may return a single file containing a large number of statements from different companies/journals. Hence we need to split the CAMT file into single statement CAMT files before we can call the logic implemented by the Odoo OE or Community CAMT parsers. """ modules = [ ("oca", "account_statement_import_camt"), ("oe", "account_bank_statement_import_camt"), ] author = False for entry in modules: if self._check_import_module(entry[1], raise_if_not_found=False): author = entry[0] break if not author: raise UserError( _( "The module to process the '%(ebics_format)s' format is " "not installed on your system. " "\nPlease install one of the following modules: \n%(modules)s.", ebics_format=self.format_id.name, modules=", ".join([x[1] for x in modules]), ) ) res = {"statement_ids": [], "notifications": []} st_datas = self._split_camt(res) if author == "oca": self._process_bank_statement_oca(res, st_datas) else: self._process_bank_statement_oe(res, st_datas) return self._process_download_result(res) def _process_bank_statement_oca(self, res, st_datas): for st_data in st_datas: try: with self.env.cr.savepoint(): self._create_bank_statement_oca(res, st_data) except UserError as e: res["notifications"].append( {"type": "error", "message": "".join(e.args)} ) except Exception: tb = "".join(format_exception(*exc_info())) res["notifications"].append({"type": "error", "message": tb}) def _create_bank_statement_oca(self, res, st_data): wiz = ( self.env["account.statement.import"] .with_company(st_data["company_id"]) .with_context(active_model="ebics.file") .create({"statement_filename": self.name}) ) wiz.import_single_file(base64.b64decode(st_data["data"]), res) def _process_bank_statement_oe(self, res, st_datas): """ We execute a cr.commit() after every statement import since we get a 'savepoint does not exist' error when using 'with self.env.cr.savepoint()'. """ for st_data in st_datas: try: self._create_bank_statement_oe(res, st_data) self.env.cr.commit() # pylint: disable=E8102 except UserError as e: msg = "".join(e.args) msg += "\n" msg += _( "Statement for Account Number %(nr)s has not been processed.", nr=st_data["acc_number"], ) res["notifications"].append({"type": "error", "message": msg}) except Exception: tb = "".join(format_exception(*exc_info())) res["notifications"].append({"type": "error", "message": tb}) def _create_bank_statement_oe(self, res, st_data): attachment = ( self.env["ir.attachment"] .with_company(st_data["company_id"]) .create( { "name": self.name, "datas": st_data["data"], "store_fname": self.name, } ) ) journal = ( self.env["account.journal"] .with_company(st_data["company_id"]) .browse(st_data["journal_id"]) ) act = journal._import_bank_statement(attachment) for entry in act["domain"]: if ( isinstance(entry, tuple) and entry[0] == "statement_id" and entry[1] == "in" ): res["statement_ids"].extend(entry[2]) break notifications = act["context"]["notifications"] if notifications: res["notifications"].append(act["context"]["notifications"]) def _unlink_camt053(self): """ Placeholder for camt053 specific actions before removing the EBICS data file and its related bank statements. """ def _split_camt(self, res): """ Split CAMT file received via EBICS per statement. Statements without transactions are removed. """ datas = [] file_data = base64.b64decode(self.data) root = etree.fromstring(file_data, parser=etree.XMLParser(recover=True)) if root is None: message = _("Invalid XML file.") res["notifications"].append({"type": "error", "message": message}) ns = {k or "ns": v for k, v in root.nsmap.items()} stmts = root[0].findall("ns:Stmt", ns) for i, stmt in enumerate(stmts): acc_number = sanitize_account_number( stmt.xpath( "ns:Acct/ns:Id/ns:IBAN/text() | ns:Acct/ns:Id/ns:Othr/ns:Id/text()", namespaces=ns, )[0] ) if not acc_number: message = _("No bank account number found.") res["notifications"].append({"type": "error", "message": message}) continue currency_code = stmt.xpath( "ns:Acct/ns:Ccy/text() | ns:Bal/ns:Amt/@Ccy", namespaces=ns )[0] root_new = deepcopy(root) entries = False for j, el in enumerate(root_new[0].findall("ns:Stmt", ns)): if j != i: el.getparent().remove(el) else: entries = el.findall("ns:Ntry", ns) if not entries: continue else: currency, journal = self._lookup_journal(res, acc_number, currency_code) if not (currency and journal): continue datas.append( { "acc_number": acc_number, "journal_id": journal.id, "company_id": journal.company_id.id, "data": base64.b64encode(etree.tostring(root_new)), } ) return datas def _process_pain002(self): """ Placeholder for processing pain.002 files. TODO: add import logic based upon OCA 'account_payment_return_import' """ def _unlink_pain002(self): """ Placeholder for pain.002 specific actions before removing the EBICS data file. """ raise NotImplementedError def _process_undefined_format(self): raise UserError( _( "The current version of the 'account_ebics' module " "has no support to automatically process EBICS files " "with format %s." ) % self.format_id.name )