addons/account/wizard/account_automatic_reconcile.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 import time
  23
  24 from osv import osv, fields
  25 from tools.translate import _
  26
  27 class account_automatic_reconcile(osv.osv_memory):
  28     _name = 'account.automatic.reconcile'
  29     _description = 'Automatic Reconcile'
  30
  31     _columns = {
  32         'account_ids': fields.many2many('account.account', 'reconcile_account_rel', 'reconcile_id', 'account_id', 'Accounts to Reconcile', domain = [('reconcile','=',1)],),
  33         'writeoff_acc_id': fields.many2one('account.account', 'Account'),
  34         'journal_id': fields.many2one('account.journal', 'Journal'),
  35         'period_id': fields.many2one('account.period', 'Period'),
  36         'max_amount': fields.float('Maximum write-off amount'),
  37         'power': fields.selection([(p, str(p)) for p in range(2, 5)], 'Power', required=True, help='Number of partial amounts that can be combined to find a balance point can be chosen as the power of the automatic reconciliation'),
  38         'reconciled': fields.integer('Reconciled transactions', readonly=True),
  39         'unreconciled': fields.integer('Not reconciled transactions', readonly=True),
  40         'allow_write_off': fields.boolean('Allow write off')
  41     }
  42
  43     def _get_reconciled(self, cr, uid, context=None):
  44         if context is None:
  45             context = {}
  46         return context.get('reconciled', 0)
  47
  48     def _get_unreconciled(self, cr, uid, context=None):
  49         if context is None:
  50             context = {}
  51         return context.get('unreconciled', 0)
  52
  53     _defaults = {
  54         'reconciled': _get_reconciled,
  55         'unreconciled': _get_unreconciled,
  56         'power': 2
  57     }
  58
  59     #TODO: cleanup and comment this code... For now, it is awfulllll
  60     # (way too complex, and really slow)...
  61     def do_reconcile(self, cr, uid, credits, debits, max_amount, power, writeoff_acc_id, period_id, journal_id, context=None):
  62     # for one value of a credit, check all debits, and combination of them
  63     # depending on the power. It starts with a power of one and goes up
  64     # to the max power allowed
  65         move_line_obj = self.pool.get('account.move.line')
  66         if context is None:
  67             context = {}
  68         def check2(value, move_list, power):
  69             def check(value, move_list, power):
  70                 for i in range(len(move_list)):
  71                     move = move_list[i]
  72                     if power == 1:
  73                         if abs(value - move[1]) <= max_amount + 0.00001:
  74                             return [move[0]]
  75                     else:
  76                         del move_list[i]
  77                         res = check(value - move[1], move_list, power-1)
  78                         move_list[i:i] = [move]
  79                         if res:
  80                             res.append(move[0])
  81                             return res
  82                 return False
  83
  84             for p in range(1, power+1):
  85                 res = check(value, move_list, p)
  86                 if res:
  87                     return res
  88             return False
  89
  90         # for a list of credit and debit and a given power, check if there
  91         # are matching tuples of credit and debits, check all debits, and combination of them
  92         # depending on the power. It starts with a power of one and goes up
  93         # to the max power allowed
  94         def check4(list1, list2, power):
  95             def check3(value, list1, list2, list1power, power):
  96                 for i in range(len(list1)):
  97                     move = list1[i]
  98                     if list1power == 1:
  99                         res = check2(value + move[1], list2, power - 1)
 100                         if res:
 101                             return ([move[0]], res)
 102                     else:
 103                         del list1[i]
 104                         res = check3(value + move[1], list1, list2, list1power-1, power-1)
 105                         list1[i:i] = [move]
 106                         if res:
 107                             x, y = res
 108                             x.append(move[0])
 109                             return (x, y)
 110                 return False
 111
 112             for p in range(1, power):
 113                 res = check3(0, list1, list2, p, power)
 114                 if res:
 115                     return res
 116             return False
 117
 118         def check5(list1, list2, max_power):
 119             for p in range(2, max_power+1):
 120                 res = check4(list1, list2, p)
 121                 if res:
 122                     return res
 123
 124         ok = True
 125         reconciled = 0
 126         while credits and debits and ok:
 127             res = check5(credits, debits, power)
 128             if res:
 129                 move_line_obj.reconcile(cr, uid, res[0] + res[1], 'auto', writeoff_acc_id, period_id, journal_id, context)
 130                 reconciled += len(res[0]) + len(res[1])
 131                 credits = [(id, credit) for (id, credit) in credits if id not in res[0]]
 132                 debits = [(id, debit) for (id, debit) in debits if id not in res[1]]
 133             else:
 134                 ok = False
 135         return (reconciled, len(credits)+len(debits))
 136
 137     def reconcile(self, cr, uid, ids, context=None):
 138         move_line_obj = self.pool.get('account.move.line')
 139         obj_model = self.pool.get('ir.model.data')
 140         if context is None:
 141             context = {}
 142         form = self.browse(cr, uid, ids, context=context)[0]
 143         max_amount = form.max_amount or 0.0
 144         power = form.power
 145         allow_write_off = form.allow_write_off
 146         reconciled = unreconciled = 0
 147         if not form.account_ids:
 148             raise osv.except_osv(_('User Error!'), _('You must select accounts to reconcile.'))
 149         for account_id in form.account_ids:
 150             params = (account_id.id,)
 151             if not allow_write_off:
 152                 query = """SELECT partner_id FROM account_move_line WHERE account_id=%s AND reconcile_id IS NULL
 153                 AND state <> 'draft' GROUP BY partner_id
 154                 HAVING ABS(SUM(debit-credit)) = 0.0 AND count(*)>0"""
 155             else:
 156                 query = """SELECT partner_id FROM account_move_line WHERE account_id=%s AND reconcile_id IS NULL
 157                 AND state <> 'draft' GROUP BY partner_id
 158                 HAVING ABS(SUM(debit-credit)) < %s AND count(*)>0"""
 159                 params += (max_amount,)
 160             # reconcile automatically all transactions from partners whose balance is 0
 161             cr.execute(query, params)
 162             partner_ids = [id for (id,) in cr.fetchall()]
 163             for partner_id in partner_ids:
 164                 cr.execute(
 165                     "SELECT id " \
 166                     "FROM account_move_line " \
 167                     "WHERE account_id=%s " \
 168                     "AND partner_id=%s " \
 169                     "AND state <> 'draft' " \
 170                     "AND reconcile_id IS NULL",
 171                     (account_id.id, partner_id))
 172                 line_ids = [id for (id,) in cr.fetchall()]
 173                 if line_ids:
 174                     reconciled += len(line_ids)
 175                     if allow_write_off:
 176                         move_line_obj.reconcile(cr, uid, line_ids, 'auto', form.writeoff_acc_id.id, form.period_id.id, form.journal_id.id, context)
 177                     else:
 178                         move_line_obj.reconcile_partial(cr, uid, line_ids, 'manual', context=context)
 179
 180             # get the list of partners who have more than one unreconciled transaction
 181             cr.execute(
 182                 "SELECT partner_id " \
 183                 "FROM account_move_line " \
 184                 "WHERE account_id=%s " \
 185                 "AND reconcile_id IS NULL " \
 186                 "AND state <> 'draft' " \
 187                 "AND partner_id IS NOT NULL " \
 188                 "GROUP BY partner_id " \
 189                 "HAVING count(*)>1",
 190                 (account_id.id,))
 191             partner_ids = [id for (id,) in cr.fetchall()]
 192             #filter?
 193             for partner_id in partner_ids:
 194                 # get the list of unreconciled 'debit transactions' for this partner
 195                 cr.execute(
 196                     "SELECT id, debit " \
 197                     "FROM account_move_line " \
 198                     "WHERE account_id=%s " \
 199                     "AND partner_id=%s " \
 200                     "AND reconcile_id IS NULL " \
 201                     "AND state <> 'draft' " \
 202                     "AND debit > 0 " \
 203                     "ORDER BY date_maturity",
 204                     (account_id.id, partner_id))
 205                 debits = cr.fetchall()
 206
 207                 # get the list of unreconciled 'credit transactions' for this partner
 208                 cr.execute(
 209                     "SELECT id, credit " \
 210                     "FROM account_move_line " \
 211                     "WHERE account_id=%s " \
 212                     "AND partner_id=%s " \
 213                     "AND reconcile_id IS NULL " \
 214                     "AND state <> 'draft' " \
 215                     "AND credit > 0 " \
 216                     "ORDER BY date_maturity",
 217                     (account_id.id, partner_id))
 218                 credits = cr.fetchall()
 219
 220                 (rec, unrec) = self.do_reconcile(cr, uid, credits, debits, max_amount, power, form.writeoff_acc_id.id, form.period_id.id, form.journal_id.id, context)
 221                 reconciled += rec
 222                 unreconciled += unrec
 223
 224             # add the number of transactions for partners who have only one
 225             # unreconciled transactions to the unreconciled count
 226             partner_filter = partner_ids and 'AND partner_id not in (%s)' % ','.join(map(str, filter(None, partner_ids))) or ''
 227             cr.execute(
 228                 "SELECT count(*) " \
 229                 "FROM account_move_line " \
 230                 "WHERE account_id=%s " \
 231                 "AND reconcile_id IS NULL " \
 232                 "AND state <> 'draft' " + partner_filter,
 233                 (account_id.id,))
 234             additional_unrec = cr.fetchone()[0]
 235             unreconciled = unreconciled + additional_unrec
 236         context.update({'reconciled': reconciled, 'unreconciled': unreconciled})
 237         model_data_ids = obj_model.search(cr,uid,[('model','=','ir.ui.view'),('name','=','account_automatic_reconcile_view1')])
 238         resource_id = obj_model.read(cr, uid, model_data_ids, fields=['res_id'])[0]['res_id']
 239         return {
 240             'view_type': 'form',
 241             'view_mode': 'form',
 242             'res_model': 'account.automatic.reconcile',
 243             'views': [(resource_id,'form')],
 244             'type': 'ir.actions.act_window',
 245             'target': 'new',
 246             'context': context,
 247         }
 248
 249 account_automatic_reconcile()
 250
 251 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: