bank-statement-import/account_bank_statement_import_ofx/account_bank_statement_import_ofx.py

# -*- coding: utf-8 -*-
# noqa: This is a backport from Odoo. OCA has no control over style here.
# flake8: noqa

import logging
import StringIO

from openerp.osv import osv
from openerp.tools.translate import _
from openerp.exceptions import Warning

_logger = logging.getLogger(__name__)

try:
    from ofxparse import OfxParser as ofxparser
except ImportError:
    _logger.error("OFX parser unavailable because the `ofxparse` Python library cannot be found."
                    "It can be downloaded and installed from `https://pypi.python.org/pypi/ofxparse`.")
    ofxparser = None

class account_bank_statement_import(osv.TransientModel):
    _inherit = 'account.bank.statement.import'

    def _check_ofx(self, cr, uid, file, context=None):
        if ofxparser is None:
            return False
        try:
            ofx = ofxparser.parse(file)
        except:
            return False
        return ofx

    def _parse_file(self, cr, uid, data_file, context=None):
        ofx = self._check_ofx(cr, uid, StringIO.StringIO(data_file), context=context)
        if not ofx:
            return super(account_bank_statement_import, self)._parse_file(cr, uid, data_file, context=context)

        transactions = []
        total_amt = 0.00
        try:
            for transaction in ofx.account.statement.transactions:
                # Since ofxparse doesn't provide account numbers, we'll have to find res.partner and res.partner.bank here
                # (normal behavious is to provide 'account_number', which the generic module uses to find partner/bank)
                bank_account_id = partner_id = False
                ids = self.pool.get('res.partner.bank').search(cr, uid, [('owner_name', '=', transaction.payee)], context=context)
                if ids:
                    bank_account_id = bank_account_id = ids[0]
                    partner_id = self.pool.get('res.partner.bank').browse(cr, uid, bank_account_id, context=context).partner_id.id
                vals_line = {
                    'date': transaction.date,
                    'name': transaction.payee + (transaction.memo and ': ' + transaction.memo or ''),
                    'ref': transaction.id,
                    'amount': transaction.amount,
                    'unique_import_id': transaction.id,
                    'bank_account_id': bank_account_id,
                    'partner_id': partner_id,
                }
                total_amt += float(transaction.amount)
                transactions.append(vals_line)
        except Exception, e:
            raise Warning(_("The following problem occurred during import. The file might not be valid.\n\n %s" % e.message))

        vals_bank_statement = {
            'name': ofx.account.routing_number,
            'transactions': transactions,
            'balance_start': float(ofx.account.statement.balance) - total_amt,
            'balance_end_real': float(ofx.account.statement.balance),
        }
        return ofx.account.statement.currency, ofx.account.number, [vals_bank_statement]