Skip to content

Commit

Permalink
Merge commit 'refs/pull/455/head' of https://github.com/oca/bank-stat…
Browse files Browse the repository at this point in the history
…ement-import into 13.0-1313
  • Loading branch information
docker-odoo committed Feb 1, 2023
2 parents 5aa465b + 6905967 commit a9b6544
Show file tree
Hide file tree
Showing 10 changed files with 290 additions and 29 deletions.
3 changes: 3 additions & 0 deletions account_bank_statement_import_txt_xlsx/data/map_data.xml
Expand Up @@ -10,12 +10,15 @@
model="account.bank.statement.import.sheet.mapping"
>
<field name="name">Sample Statement</field>
<field name="footer_lines_count">0</field>
<field name="column_labels_row">1</field>
<field name="float_thousands_sep">comma</field>
<field name="float_decimal_sep">dot</field>
<field name="delimiter">comma</field>
<field name="quotechar">"</field>
<field name="timestamp_format">%m/%d/%Y</field>
<field name="timestamp_column">Date</field>
<field name="amount_type">simple_value</field>
<field name="amount_column">Amount</field>
<field name="original_currency_column">Currency</field>
<field name="original_amount_column">Amount Currency</field>
Expand Down
@@ -0,0 +1,15 @@
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).

from openupgradelib import openupgrade


@openupgrade.migrate()
def migrate(env, version):
openupgrade.logged_query(
env.cr,
"""
UPDATE account_bank_statement_import_sheet_mapping
SET amount_type = 'absolute_value'
WHERE debit_credit_column IS NOT NULL;
""",
)
Expand Up @@ -86,10 +86,43 @@ class AccountBankStatementImportSheetMapping(models.Model):
"transaction amount in original transaction currency from"
),
)
amount_type = fields.Selection(
selection=[
("simple_value", "Simple value"),
("absolute_value", "Absolute value"),
("distinct_credit_debit", "Distinct Credit/debit Column"),
],
string="Amount type",
required=True,
default="simple_value",
help=(
"Simple value: use igned amount in ammount comlumn\n"
"Absolute Value: use a same comlumn for debit and credit\n"
"(absolute value + indicate sign)\n"
"Distinct Credit/debit Column: use a distinct comlumn for debit and credit"
),
)
amount_column = fields.Char(
string="Amount column",
help=(
'Used if amount type is "Simple value" or "Absolute value"\n'
"Amount of transaction in journal's currency\n"
"Some statement formats use credit/debit columns"
),
)
debit_column = fields.Char(
string="Debit column",
help='Used if amount type is "Distinct Credit/debit Column"',
)
credit_column = fields.Char(
string="Credit column",
help='Used if amount type is "Distinct Credit/debit Column"\n',
)
debit_credit_column = fields.Char(
string="Debit/credit column",
help=(
"Some statement formats use absolute amount value and indicate sign"
'Used if amount type is "Absolute value"\n'
"Some statement formats use absolute amount value and indicate sign\n"
"of the transaction by specifying if it was a debit or a credit one"
),
)
Expand All @@ -112,6 +145,18 @@ class AccountBankStatementImportSheetMapping(models.Model):
bank_account_column = fields.Char(
string="Bank Account column", help="Partner's bank account",
)
footer_lines_count = fields.Integer(
string="Footer lines number",
help="Set the Footer lines number."
"Used in some csv file that integrate meta data in"
"last lines.",
default="0",
)
column_labels_row = fields.Integer(
string="Row number for column labels",
help="The number of line that contain column names.",
default="1",
)

@api.onchange("float_thousands_sep")
def onchange_thousands_separator(self):
Expand Down
Expand Up @@ -25,21 +25,23 @@ class AccountBankStatementImportSheetParser(models.TransientModel):
_description = "Account Bank Statement Import Sheet Parser"

@api.model
def parse_header(self, data_file, encoding, csv_options):
def parse_header(self, data_file, encoding, csv_options, column_labels_row=1):
try:
workbook = xlrd.open_workbook(
file_contents=data_file,
encoding_override=encoding if encoding else None,
)
sheet = workbook.sheet_by_index(0)
values = sheet.row_values(0)
values = sheet.row_values(column_labels_row - 1)
return [str(value) for value in values]
except xlrd.XLRDError:
pass

data = StringIO(data_file.decode(encoding or "utf-8"))
csv_data = reader(data, **csv_options)
return list(next(csv_data))
csv_data_lst = list(csv_data)
header = [value.strip() for value in csv_data_lst[column_labels_row - 1]]
return header

@api.model
def parse(self, data_file, mapping):
Expand All @@ -51,9 +53,12 @@ def parse(self, data_file, mapping):
if not lines:
return currency_code, account_number, [{"transactions": []}]

lines = list(sorted(lines, key=lambda line: line["timestamp"]))
first_line = lines[0]
last_line = lines[-1]
if lines[0]["timestamp"] > lines[-1]["timestamp"]:
first_line = lines[-1]
last_line = lines[0]
else:
first_line = lines[0]
last_line = lines[-1]
data = {
"date": first_line["timestamp"].date(),
}
Expand Down Expand Up @@ -102,16 +107,32 @@ def _parse_lines(self, mapping, data_file, currency_code):
StringIO(data_file.decode(mapping.file_encoding or "utf-8")),
**csv_options
)

if isinstance(csv_or_xlsx, tuple):
header = [str(value) for value in csv_or_xlsx[1].row_values(0)]
header = [
str(value).strip()
for value in csv_or_xlsx[1].row_values(mapping.column_labels_row - 1)
]
else:
header = [value.strip() for value in next(csv_or_xlsx)]
columns["timestamp_column"] = header.index(mapping.timestamp_column)
columns["currency_column"] = (
header.index(mapping.currency_column) if mapping.currency_column else None
)
columns["amount_column"] = header.index(mapping.amount_column)
columns["amount_column"] = (
header.index(mapping.amount_column)
if mapping.amount_type != "distinct_credit_debit"
else None
)
columns["debit_column"] = (
header.index(mapping.debit_column)
if mapping.amount_type == "distinct_credit_debit"
else None
)
columns["credit_column"] = (
header.index(mapping.credit_column)
if mapping.amount_type == "distinct_credit_debit"
else None
)
columns["balance_column"] = (
header.index(mapping.balance_column) if mapping.balance_column else None
)
Expand Down Expand Up @@ -163,17 +184,25 @@ def _parse_lines(self, mapping, data_file, currency_code):

def _parse_rows(self, mapping, currency_code, csv_or_xlsx, columns): # noqa: C901
if isinstance(csv_or_xlsx, tuple):
rows = range(1, csv_or_xlsx[1].nrows)
rows = range(
mapping.column_labels_row,
csv_or_xlsx[1].nrows - mapping.footer_lines_count,
)
else:
rows = csv_or_xlsx
stat_first_index = mapping.column_labels_row
stat_last_index = -mapping.footer_lines_count
if stat_last_index:
rows = csv_or_xlsx[stat_first_index:stat_last_index]
else:
rows = csv_or_xlsx[stat_first_index:]

lines = []
for row in rows:
if isinstance(csv_or_xlsx, tuple):
book = csv_or_xlsx[0]
sheet = csv_or_xlsx[1]
values = []
for col_index in range(sheet.row_len(row)):
for col_index in range(0, sheet.row_len(row)):
cell_type = sheet.cell_type(row, col_index)
cell_value = sheet.cell_value(row, col_index)
if cell_type == xlrd.XL_CELL_DATE:
Expand All @@ -188,7 +217,6 @@ def _parse_rows(self, mapping, currency_code, csv_or_xlsx, columns): # noqa: C9
if columns["currency_column"] is not None
else currency_code
)
amount = values[columns["amount_column"]]
balance = (
values[columns["balance_column"]]
if columns["balance_column"] is not None
Expand Down Expand Up @@ -251,17 +279,25 @@ def _parse_rows(self, mapping, currency_code, csv_or_xlsx, columns): # noqa: C9
if isinstance(timestamp, str):
timestamp = datetime.strptime(timestamp, mapping.timestamp_format)

amount = self._parse_decimal(amount, mapping)
if balance:
balance = self._parse_decimal(balance, mapping)
else:
balance = None

if debit_credit:
amount_column = columns["amount_column"]
if amount_column and values[columns["amount_column"]]:
amount = self._parse_decimal(values[columns["amount_column"]], mapping)
if debit_credit is not None:
amount = amount.copy_abs()
if debit_credit == mapping.debit_value:
amount = -amount

debit_column = columns["debit_column"]
credit_column = columns["credit_column"]
if debit_column and credit_column:
debit_amount = self._parse_decimal(values[debit_column], mapping)
debit_amount = debit_amount.copy_abs()
credit_amount = self._parse_decimal(values[credit_column], mapping)
amount = credit_amount - debit_amount

if balance is not None:
balance = self._parse_decimal(balance, mapping)

if not original_currency:
original_currency = currency
original_amount = amount
Expand Down Expand Up @@ -371,6 +407,7 @@ def _parse_decimal(self, value, mapping):
return value
elif isinstance(value, float):
return Decimal(value)
value = value or "0"
thousands, decimal = mapping._get_float_separators()
value = value.replace(thousands, "")
value = value.replace(decimal, ".")
Expand Down
@@ -1,5 +1,6 @@
* Alexis de Lattre <alexis.delattre@akretion.com>
* Sebastien BEAU <sebastien.beau@akretion.com>
* Mourad EL HADJ MIMOUNE <mourad.elhadj.mimoune@akretion.com>
* Tecnativa (https://www.tecnativa.com)

* Vicent Cubells <vicent.cubells@tecnativa.com>
Expand Down
@@ -0,0 +1,10 @@
Bank code : 1001010101,Agency Code : 10000,Download start date : 01/04/2020,Download end date : 02/04/2020,,
Account Number : 08088804068,Account Name : Account Owner,: EUR,,,
,,,,,
Balance at end of period,,,,"+31070,11",
Date,Operation Number,Label,Debit,Credit,Detail
01/04/20,UNIQUE OP 1,LABEL 1,"-50,00",,DETAILS 1
01/04/20,UNIQUE OP 2,LABEL 2,"-100,00",,CLIENTS X
02/04/20,UNIQUE OP 3,LABEL 3,"-80,68",,DETAILS 2
02/04/20,UNIQUE OP 4,LABEL 4,,"1300,00",DETAILS 3
Balance at start of period,,,,"+30000,77",
Expand Up @@ -372,3 +372,54 @@ def test_debit_credit(self):
self.assertEqual(statement.balance_start, 10.0)
self.assertEqual(statement.balance_end_real, 1510.0)
self.assertEqual(statement.balance_end, 1510.0)

def test_metadata_separated_debit_credit(self):
journal = self.AccountJournal.create(
{
"name": "Bank",
"type": "bank",
"code": "BANK",
"currency_id": self.currency_usd.id,
}
)
statement_map = self.sample_statement_map.copy(
{
"footer_lines_count": 1,
"column_labels_row": 5,
"amount_column": None,
"partner_name_column": None,
"bank_account_column": None,
"float_thousands_sep": "none",
"float_decimal_sep": "comma",
"timestamp_format": "%m/%d/%y",
"original_currency_column": None,
"original_amount_column": None,
"amount_type": "distinct_credit_debit",
"debit_column": "Debit",
"credit_column": "Credit",
}
)
wizard = self.AccountBankStatementImport.with_context(
{"journal_id": journal.id}
).create(
{
"filename": "fixtures/meta_data_separated_credit_debit.csv",
"data_file": self._data_file(
"fixtures/meta_data_separated_credit_debit.csv", "utf-8"
),
"sheet_mapping_id": statement_map.id,
}
)
wizard.with_context(
{
"journal_id": journal.id,
"account_bank_statement_import_txt_xlsx_test": True,
}
).import_file()
statement = self.AccountBankStatement.search([("journal_id", "=", journal.id)])
self.assertEqual(len(statement), 1)
self.assertEqual(len(statement.line_ids), 4)
line1 = statement.line_ids.filtered(lambda x: x.name == "LABEL 1")
line4 = statement.line_ids.filtered(lambda x: x.name == "LABEL 4")
self.assertEqual(line1.amount, -50)
self.assertEqual(line4.amount, 1300)
Expand Up @@ -51,15 +51,46 @@
attrs="{'required': [('debit_credit_column', '!=', False)]}"
/>
</group>
<group>
<field name="column_labels_row" />
<field name="footer_lines_count" />
</group>
</group>
<group string="Columns">
<field name="timestamp_column" />
<field name="currency_column" />
<field name="amount_column" />
<field name="amount_type" />
<field
name="amount_column"
attrs="{
'required': [('amount_type', '!=', 'distinct_credit_debit'),],
'invisible': [('amount_type', '=', 'distinct_credit_debit')],
}"
/>
<field
name="debit_column"
attrs="{
'required': [('amount_type', '=', 'distinct_credit_debit')],
'invisible': [('amount_type', '!=', 'distinct_credit_debit')],
}"
/>
<field
name="credit_column"
attrs="{
'required': [('amount_type', '=', 'distinct_credit_debit')],
'invisible': [('amount_type', '!=', 'distinct_credit_debit')],
}"
/>
<field
name="debit_credit_column"
attrs="{
'required': [('amount_type', '=', 'absolute_value')],
'invisible': [('amount_type', '!=', 'absolute_value')],
}"
/>
<field name="balance_column" />
<field name="original_currency_column" />
<field name="original_amount_column" />
<field name="debit_credit_column" />
<field name="transaction_id_column" />
<field name="description_column" />
<field name="notes_column" />
Expand Down

0 comments on commit a9b6544

Please sign in to comment.