[ADD] account_bank_statement_import_split: split statements during import

account_bank_statement_import_split/models/__init__.py

@@ -0,0 +1,3 @@
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from . import account_bank_statement_import

account_bank_statement_import_split/models/account_bank_statement_import.py

@@ -0,0 +1,153 @@
+# Copyright 2019 Brainbean Apps (https://brainbeanapps.com)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+
+from odoo import api, fields, models
+
+from dateutil.relativedelta import relativedelta, MO
+from decimal import Decimal
+
+
+class AccountBankStatementImport(models.TransientModel):
+    _inherit = 'account.bank.statement.import'
+
+    import_mode = fields.Selection(
+        selection=[
+            ('single', 'Single statement'),
+            ('daily', 'Daily statements'),
+            ('weekly', 'Weekly statements'),
+            ('monthly', 'Monthly statements'),
+        ],
+        default='single',
+    )
+
+    def _complete_stmts_vals(self, stmts_vals, journal, account_number):
+        stmts_vals = super()._complete_stmts_vals(
+            stmts_vals,
+            journal,
+            account_number
+        )
+        if not self.import_mode or self.import_mode == 'single':
+            return stmts_vals
+        statements = []
+        for st_vals in stmts_vals:
+            transactions = list(sorted(
+                map(
+                    lambda transaction: self._prepare_transaction(
+                        transaction
+                    ),
+                    st_vals['transactions']
+                ),
+                key=lambda transaction: transaction['date']
+            ))
+            if not transactions:
+                continue
+            del st_vals['transactions']
+
+            balance_start = Decimal(st_vals['balance_start']) \
+                if 'balance_start' in st_vals else None
+            balance_end = Decimal(st_vals['balance_end_real']) \
+                if 'balance_end_real' in st_vals else None
+            statement_date_since = self._get_statement_date_since(
+                transactions[0]['date']
+            )
+            while transactions:
+                statement_date_until = (
+                    statement_date_since + self._get_statement_date_step()
+                )
+
+                last_transaction_index = None
+                for index, transaction in enumerate(transactions):
+                    if transaction['date'] >= statement_date_until:
+                        break
+                    last_transaction_index = index
+                if last_transaction_index is None:
+                    # NOTE: No transactions for current period
+                    statement_date_since = statement_date_until
+                    continue
+
+                statement_transactions = \
+                    transactions[0:last_transaction_index + 1]
+                transactions = transactions[last_transaction_index + 1:]
+
+                statement_values = dict(st_vals)
+                statement_values.update({
+                    'name': self._get_statement_name(
+                        journal,
+                        statement_date_since,
+                        statement_date_until,
+                    ),
+                    'date': self._get_statement_date(
+                        statement_date_since,
+                        statement_date_until,
+                    ),
+                    'transactions': statement_transactions,
+                })
+                if balance_start is not None:
+                    statement_values.update({
+                        'balance_start': float(balance_start),
+                    })
+                    for transaction in statement_transactions:
+                        balance_start += Decimal(transaction['amount'])
+                if balance_end is not None:
+                    statement_balance_end = balance_end
+                    for transaction in transactions:
+                        statement_balance_end -= Decimal(transaction['amount'])
+                    statement_values.update({
+                        'balance_end_real': float(statement_balance_end),
+                    })
+
+                statements.append(statement_values)
+                statement_date_since = statement_date_until
+        return statements
+
+    @api.multi
+    def _prepare_transaction(self, transaction):
+        transaction.update({
+            'date': fields.Date.from_string(transaction['date']),
+        })
+        return transaction
+
+    @api.multi
+    def _get_statement_date_since(self, date):
+        self.ensure_one()
+        if self.import_mode == 'daily':
+            return date
+        elif self.import_mode == 'weekly':
+            return date + relativedelta(weekday=MO(-1))
+        elif self.import_mode == 'monthly':
+            return date.replace(
+                day=1,
+            )
+
+    @api.multi
+    def _get_statement_date_step(self):
+        self.ensure_one()
+        if self.import_mode == 'daily':
+            return relativedelta(
+                days=1,
+            )
+        elif self.import_mode == 'weekly':
+            return relativedelta(
+                weeks=1,
+                weekday=MO,
+            )
+        elif self.import_mode == 'monthly':
+            return relativedelta(
+                months=1,
+                day=1,
+            )
+
+    @api.multi
+    def _get_statement_date(self, date_since, date_until):
+        self.ensure_one()
+        # NOTE: Statement date is treated by Odoo as start of period. Details
+        #  - addons/account/models/account_journal_dashboard.py
+        #  - def get_line_graph_datas()
+        return date_since
+
+    @api.multi
+    def _get_statement_name(self, journal, date_since, date_until):
+        self.ensure_one()
+        return journal.sequence_id.with_context(
+            ir_sequence_date=self._get_statement_date(date_since, date_until)
+        ).next_by_id()