account-reconcile/account_move_transactionid_.../parser/transactionid_file_parser.py

90 lines
3.1 KiB
Python

# © 2011-2019 Camptocamp SA
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
import datetime
from odoo.tools import ustr
from odoo.addons.account_move_base_import.parser.file_parser import (
FileParser,
float_or_zero,
)
class TransactionIDFileParser(FileParser):
"""TransactionID parser that use a define format in csv or xls to import
bank statement.
"""
def __init__(
self,
profile,
ftype="csv",
extra_fields=None,
header=None,
**kwargs):
"""Add transaction_id in header keys
:param char: profile: Reference to the profile
:param char: ftype: extension of the file (could be csv or xls)
:param dict: extra_fields: extra fields to add to the conversion
dict. In the format {fieldname: fieldtype}
:param list: header : specify header fields if the csv file has no
header
"""
conversion_dict = {
"transaction_id": ustr,
"label": ustr,
"date": datetime.datetime,
"amount": float_or_zero,
"commission_amount": float_or_zero,
}
super().__init__(
profile,
extra_fields=conversion_dict,
ftype=ftype,
header=header,
**kwargs
)
self.support_multi_moves = True
@classmethod
def parser_for(cls, parser_name):
"""Used by the new_bank_statement_parser class factory. Return true if
the providen name is generic_csvxls_transaction
"""
return parser_name == "generic_csvxls_transaction"
def get_move_line_vals(self, line, *args, **kwargs):
"""This method must return a dict of vals that can be passed to create
method of statement line in order to record it. It is the
responsibility of every parser to give this dict of vals, so each one
can implement his own way of recording the lines.
:param: line: a dict of vals that represent a line of
result_row_list
:return: dict of values to give to the create method of statement
line, it MUST contain at least:
{
'name':value,
'date':value,
'amount':value,
'ref':value,
'label':value,
'commission_amount':value,
}
In this generic parser, the commission is given for every line, so we
store it for each one.
"""
amount = line.get("amount", 0.0)
return {
"name": line.get("label", "/"),
"date_maturity": line.get("date", datetime.datetime.now().date()),
"credit": amount > 0.0 and amount or 0.0,
"debit": amount < 0.0 and -amount or 0.0,
"ref": line.get("transaction_id", "/"),
}
def get_move_vals(self):
res = super().get_move_vals()
if "ref" in res:
res.pop("ref")
if res.get("name") == "/":
res["name"] = self.move_ref
return res