113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
from io import BytesIO
|
|
|
|
import ofxparse
|
|
|
|
from trytond.pool import PoolMeta, Pool
|
|
|
|
|
|
class StatementImportStart:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'account.statement.import.start'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(StatementImportStart, cls).__setup__()
|
|
cls.file_format.selection.append(('ofx', "OFX"))
|
|
|
|
|
|
class StatementImport:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'account.statement.import'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(StatementImport, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'ofx_no_journal': (
|
|
'No journal found for the bank account "%(account)s".'),
|
|
'ofx_wrong_currency': (
|
|
'The journal "%(journal)s" must have '
|
|
'the currency "%(ofx_currency)s" instead of '
|
|
'"%(currency)s".'),
|
|
'ofx_not_statement': (
|
|
"The OFX file is not a statement."),
|
|
})
|
|
|
|
def parse_ofx(self):
|
|
file_ = BytesIO(self.start.file_)
|
|
ofx = ofxparse.OfxParser.parse(file_)
|
|
|
|
for account in ofx.accounts:
|
|
statement = self.ofx_statement(ofx, account)
|
|
origins = []
|
|
for transaction in account.statement.transactions:
|
|
origins.extend(self.ofx_origin(account, transaction))
|
|
statement.origins = origins
|
|
yield statement
|
|
|
|
def ofx_statement(self, ofx, ofx_account):
|
|
pool = Pool()
|
|
Statement = pool.get('account.statement')
|
|
Journal = pool.get('account.statement.journal')
|
|
|
|
statement = Statement()
|
|
statement.name = ofx.trnuid
|
|
statement.company = self.start.company
|
|
statement.journal = Journal.get_by_bank_account(
|
|
statement.company, ofx_account.number)
|
|
if not statement.journal:
|
|
self.raise_user_error('ofx_no_journal', {
|
|
'account': ofx_account.number,
|
|
})
|
|
if statement.journal.currency.code != ofx_account.curdef:
|
|
self.raise_user_error('ofx_wrong_currency', {
|
|
'journal': statement.journal.rec_name,
|
|
'currency': statement.journal.currency.rec_name,
|
|
'ofx_currency': ofx_account.curdef,
|
|
})
|
|
if not isinstance(ofx_account.statement, ofxparse.Statement):
|
|
self.raise_user_error('ofx_not_statement')
|
|
statement.date = ofx_account.statement.balance_date.date()
|
|
statement.total_amount = ofx_account.statement.balance
|
|
statement.number_of_lines = len(ofx_account.statement.transactions)
|
|
return statement
|
|
|
|
def ofx_origin(self, ofx_account, transaction):
|
|
pool = Pool()
|
|
Origin = pool.get('account.statement.origin')
|
|
|
|
origin = Origin()
|
|
origin.number = transaction.id
|
|
origin.date = transaction.date.date()
|
|
origin.amount = transaction.amount
|
|
origin.party = self.ofx_party(ofx_account, transaction)
|
|
origin.description = transaction.memo
|
|
origin.informations = self.ofx_informations(ofx_account, transaction)
|
|
return [origin]
|
|
|
|
def ofx_party(self, ofx_account, transaction):
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
|
|
if not transaction.payee:
|
|
return
|
|
parties = Party.search([('rec_name', 'ilike', transaction.payee)])
|
|
if len(parties) == 1:
|
|
party, = parties
|
|
return party
|
|
|
|
def ofx_informations(self, ofx_account, transaction):
|
|
informations = {}
|
|
for name in [
|
|
'checknum',
|
|
'mcc',
|
|
'sic',
|
|
'type',
|
|
]:
|
|
value = getattr(transaction, name)
|
|
if value:
|
|
informations['ofx_' + name] = value
|
|
return informations
|