bank-statement-import-api/account_statement_import_api_bridge/models/account_journal.py at b253c558a23599349c35d142ac17fe5edc76424b · akretion/bank-statement-import-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Copyright 2025 Akretion France (https://www.akretion.com/)
# @author: Alexis de Lattre <alexis.delattre@akretion.com>
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from datetime import timedelta

import pytz

from odoo import fields, models


class AccountJournal(models.Model):
    _inherit = "account.journal"

    bridge_preferred_date = fields.Selection(
        [
            ("booking_date", "Booking Date"),
            ("value_date", "Value Date"),
            ("transaction_date", "Transaction Date"),
            ("date", "date"),
        ],
        default="booking_date",
    )

    def _api_import_bridge(self, result, speedy):
        self.ensure_one()
        if not self.statement_import_api_account_identifier:
            self._api_import_error_log(result, "Bridge Account Identifier is not set")
            return
        speedy["bridge_preferred_date"] = self.bridge_preferred_date
        import_api = self.statement_import_api_id
        headers = import_api._bridge_get_headers(self.company_id, result, speedy)
        if not headers:
            return  # The error is already in the logs
        # I would like to filter-out future lines, but it not possible in params
        params = {
            "account_id": self.statement_import_api_account_identifier,
        }
        if self.statement_import_api_last_success:
            # rewind 1h, just in case
            since_dt = self.statement_import_api_last_success - timedelta(hours=1)
            since_dt_aware = pytz.utc.localize(since_dt)
            since_iso = since_dt_aware.isoformat(timespec="milliseconds")
            if since_iso.endswith("+00:00"):
                since_iso = f"{since_iso[:-6]}Z"
            params["since"] = since_iso
        else:
            params["min_date"] = self.statement_import_api_start_date

        transactions = import_api._bridge_get_all_pages(
            "aggregation/transactions", headers, result, speedy, params
        )
        if transactions:
            for trans in transactions:
                pivot = self._api_import_bridge_prepare_pivot_line(
                    trans, result, speedy
                )
                if pivot:
                    result["lines"].append(pivot)

    def _api_import_bridge_prepare_pivot_line(self, trans, result, speedy):
        assert str(trans["account_id"]) == self.statement_import_api_account_identifier
        if speedy["bridge_preferred_date"]:
            date = trans.get(speedy["bridge_preferred_date"])
        else:
            date = trans.get("booking_date")
        # 'date' is the only field that is always set
        if not date:
            date = trans["date"]
        last_update_dt = self._api_import_timestamp_iso8601_to_datetime(
            trans["updated_at"], speedy
        )
        pivot = {
            "currency_code": trans["currency_code"],
            "date": date,
            "payment_ref": trans["provider_description"],
            "amount": trans["amount"],
            "unique_import_id": str(trans["id"]),
            "transaction_type": trans.get("operation_type"),
            "last_update_dt": last_update_dt,
            "to_delete": trans["deleted"],
        }
        if trans["future"]:
            self._api_import_info_log(
                result,
                f"Skipped transaction dated {pivot['date']} "
                f"amount {pivot['amount']} label '{pivot['payment_ref']}' "
                f"which has future flag",
            )
            return False
        return pivot