Update statements balance start and end_real

wizard/account_bank_statement_import.py

@@ -39,28 +39,15 @@ class AccountBankStatementImport(models.TransientModel):
             AccountBankStatementImport,
             self)._complete_stmts_vals(stmts_vals, journal, account_number)
         # Create the bank statements
-        super(AccountBankStatementImport, self)\
+        statement_ids, notifications = super(AccountBankStatementImport, self)\
             ._create_bank_statements(stmts_vals)
+        statements = self.env['account.bank.statement'].browse(statement_ids)
+        # Update Bank Statements balances
+        for statement in statements:
+            total_amt = statement['balance_end'] - statement['balance_start']
+            statement['balance_end_real'] = statement['balance_start']
+            statement['balance_start'] = statement['balance_end_real'] - \
+                total_amt
         # Now that the import worked out, set it as the bank_statements_source
         # of the journal
         journal.bank_statements_source = 'file_import'
-
-    def _parse_file(self, data_file):
-        ofx = super(AccountBankStatementImport, self)._check_ofx(data_file)
-        if not ofx:
-            return super(AccountBankStatementImport, self)._parse_file(data_file)
-
-        currency_code, account_number, stmts_vals = super(
-            AccountBankStatementImport, self)._parse_file(data_file)
-
-        try:
-            available_balance = float(ofx.account.statement.available_balance)
-            stmts_vals[0]['balance_start'] = (float(stmts_vals[0]['balance_start']) -
-                                              float(stmts_vals[0]['balance_end_real'])
-                                              + float(available_balance))
-            stmts_vals[0]['balance_end_real'] = available_balance
-        except NameError:
-            raise UserError(_('Attribute "available_balance" not available \
-                              in OFX file'))
-
-        return currency_code, account_number, stmts_vals