odoo/addons/account/account_bank_statement.py

# -*- encoding: utf-8 -*-
##############################################################################
#
# Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# $Id: account.py 1005 2005-07-25 08:41:42Z nicoe $
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################
import time
import netsvc
from osv import fields, osv

from tools.misc import currency

import mx.DateTime
from mx.DateTime import RelativeDateTime, now, DateTime, localtime

#
# use a sequence for names ?
# 
class account_bank_statement(osv.osv):
	def _default_journal_id(self, cr, uid, context={}):
		if context.get('journal_id', False):
			return context['journal_id']
		return False

	def _default_balance_start(self, cr, uid, context={}):
		cr.execute('select id from account_bank_statement where journal_id=%d order by date desc limit 1', (1,))
		res = cr.fetchone()
		if res:
			return self.browse(cr, uid, [res[0]], context)[0].balance_end
		return 0.0

	def _end_balance(self, cr, uid, ids, prop, unknow_none, unknow_dict):
		res = {}
		statements = self.browse(cr, uid, ids)
		for statement in statements:
			res[statement.id] = statement.balance_start
			for line in statement.move_line_ids:
				if line.debit>0:
					if line.account_id.id == (statement.journal_id.default_debit_account_id.id):
						res[statement.id]+=line.debit
				else:
					if line.account_id.id == (statement.journal_id.default_credit_account_id.id):
						res[statement.id]+=line.credit
			if statement.state=='draft':
				for line in statement.line_ids:
					res[statement.id] += line.amount
		for r in res:
			res[r] = round(res[r], 2)
		return res

	def _get_period(self, cr, uid, context={}):
		periods = self.pool.get('account.period').find(cr, uid)
		if periods:
			return periods[0]
		else:
			return False

	_order = "date desc"
	_name = "account.bank.statement"
	_description = "Bank Statement"
	_columns = {
		'name': fields.char('Name', size=64, required=True),
		'date': fields.date('Date', required=True, states={'confirm':[('readonly',True)]}),
		'journal_id': fields.many2one('account.journal', 'Journal', required=True, states={'confirm':[('readonly',True)]}, domain=[('type','=','cash')]),
		'period_id': fields.many2one('account.period', 'Period', required=True, states={'confirm':[('readonly',True)]}),
		'balance_start': fields.float('Starting Balance', digits=(16,2), states={'confirm':[('readonly',True)]}),
		'balance_end_real': fields.float('Ending Balance', digits=(16,2), states={'confirm':[('readonly',True)]}),
		'balance_end': fields.function(_end_balance, method=True, string='Balance'),
		'line_ids': fields.one2many('account.bank.statement.line', 'statement_id', 'Statement lines', states={'confirm':[('readonly',True)]}),
		'move_line_ids': fields.one2many('account.move.line', 'statement_id', 'Entry lines', states={'confirm':[('readonly',True)]}),
		'state': fields.selection([('draft','Draft'),('confirm','Confirm')], 'State', required=True, states={'confirm':[('readonly',True)]}, readonly="1"),
	}
	
	_defaults = {
		'name': lambda self,cr,uid,context={}: self.pool.get('ir.sequence').get(cr, uid, 'account.bank.statement'),
		'date': lambda *a: time.strftime('%Y-%m-%d'),
		'state': lambda *a: 'draft',
		'balance_start': _default_balance_start,
		'journal_id': _default_journal_id,
		'period_id': _get_period,
	}
	def button_confirm(self, cr, uid, ids, context={}):
		done = []
		for st in self.browse(cr, uid, ids, context):
			if not st.state=='draft':
				continue
			if not (abs(st.balance_end - st.balance_end_real) < 0.0001):
				raise osv.except_osv('Error !', 'The statement balance is incorrect !\nCheck that the ending balance equals the computed one.')
			if (not st.journal_id.default_credit_account_id) or (not st.journal_id.default_debit_account_id):
				raise osv.except_osv('Configration Error !', 'Please verify that an account is defined in the journal.')
			for line in st.move_line_ids:
				if line.state <> 'valid':
					raise osv.except_osv('Error !', 'The account entries lines are not valid.')
			for move in st.line_ids:
				move_id = self.pool.get('account.move').create(cr, uid, {
					'journal_id': st.journal_id.id,
					'period_id': st.period_id.id,
				}, context=context)
				self.pool.get('account.bank.statement.line').write(cr, uid, [move.id], {
					'move_ids': [(4,move_id, False)]
				})
				if not move.amount:
					continue
				torec = []
				amount = move.amount
				if move.reconcile_id and move.reconcile_id.line_new_ids:
					for newline in move.reconcile_id.line_new_ids:
						amount += newline.amount
				torec.append(self.pool.get('account.move.line').create(cr, uid, {
					'name': move.name,
					'date': move.date,
					'ref': move.ref,
					'move_id': move_id,
					'partner_id': ((move.partner_id) and move.partner_id.id) or False,
					'account_id': (move.account_id) and move.account_id.id,
					'credit': ((amount>0) and amount) or 0.0,
					'debit': ((amount<0) and -amount) or 0.0,
					'statement_id': st.id,
					'journal_id': st.journal_id.id,
					'period_id': st.period_id.id,
				}, context=context))
				if move.reconcile_id and move.reconcile_id.line_new_ids:
					for newline in move.reconcile_id.line_new_ids:
						self.pool.get('account.move.line').create(cr, uid, {
							'name': newline.name or move.name,
							'date': move.date,
							'ref': move.ref,
							'move_id': move_id,
							'partner_id': ((move.partner_id) and move.partner_id.id) or False,
							'account_id': (newline.account_id) and newline.account_id.id,
							'debit': newline.amount>0 and newline.amount or 0.0,
							'credit': newline.amount<0 and -newline.amount or 0.0,
							'statement_id': st.id,
							'journal_id': st.journal_id.id,
							'period_id': st.period_id.id,
						}, context=context)

				c = context.copy()
				c['journal_id'] = st.journal_id.id
				c['period_id'] = st.period_id.id
				fields = ['move_id','name','date','partner_id','account_id','credit','debit']
				default = self.pool.get('account.move.line').default_get(cr, uid, fields, context=c)
				default.update({
					'statement_id': st.id,
					'journal_id': st.journal_id.id,
					'period_id': st.period_id.id,
					'move_id': move_id,
					'ref': move.ref,
				})
				self.pool.get('account.move.line').create(cr, uid, default, context=context)
				if move.reconcile_id and move.reconcile_id.line_ids:
					torec += map(lambda x: x.id, move.reconcile_id.line_ids)
					try:
						self.pool.get('account.move.line').reconcile(cr, uid, torec, 'statement', context)
					except:
						raise osv.except_osv('Error !', 'Unable to reconcile entry "%s": %.2f'%(move.name, move.amount))

			done.append(st.id)
		self.write(cr, uid, done, {'state':'confirm'}, context=context)
		return True
	def button_cancel(self, cr, uid, ids, context={}):
		done = []
		for st in self.browse(cr, uid, ids, context):
			if st.state=='draft':
				continue
			ids = []
			for line in st.line_ids:
				ids += [x.id for x in line.move_ids]
			self.pool.get('account.move').unlink(cr, uid, ids, context)
			done.append(st.id)
		self.write(cr, uid, done, {'state':'draft'}, context=context)
		return True
	def onchange_journal_id(self, cr, uid, id, journal_id, context={}):
		if not journal_id:
			return {}
		cr.execute('select balance_end_real from account_bank_statement where journal_id=%d order by date desc limit 1', (journal_id,))
		res = cr.fetchone()
		if res:
			return {'value': {'balance_start': res[0] or 0.0}}
		return {}
account_bank_statement()

class account_bank_statement_reconcile(osv.osv):
	_name = "account.bank.statement.reconcile"
	_description = "Statement reconcile"
	def _total_entry(self, cr, uid, ids, prop, unknow_none, context={}):
		result = {}
		for o in self.browse(cr, uid, ids, context):
			result[o.id] = 0.0
			for line in o.line_ids:
				result[o.id] += line.debit - line.credit
		return result

	def _total_new(self, cr, uid, ids, prop, unknow_none, context={}):
		result = {}
		for o in self.browse(cr, uid, ids, context):
			result[o.id] = 0.0
			for line in o.line_new_ids:
				result[o.id] += line.amount
		return result

	def _total_balance(self, cr, uid, ids, prop, unknow_none, context={}):
		result = {}
		for o in self.browse(cr, uid, ids, context):
			result[o.id] = o.total_new-o.total_entry+o.total_amount
		return result

	def _total_amount(self, cr, uid, ids, prop, unknow_none, context={}):
		return dict(map(lambda x: (x,context.get('amount', 0.0)), ids))

	def name_get(self, cr, uid, ids, context):
		res= []
		for o in self.browse(cr, uid, ids, context):
			res.append((o.id, '[%.2f/%.2f]' % (o.total_entry, o.total_new)))
		return res

	_columns = {
		'name': fields.char('Date', size=64, required=True),
		'partner_id': fields.many2one('res.partner', 'Partner', readonly=True),
		'line_new_ids': fields.one2many('account.bank.statement.reconcile.line', 'line_id', 'Write-Off'),
		'total_entry': fields.function(_total_entry, method=True, string='Total entries'),
		'total_new': fields.function(_total_new, method=True, string='Total write-off'),
		'total_amount': fields.function(_total_amount, method=True, string='Payment amount'),
		'total_balance': fields.function(_total_balance, method=True, string='Balance'),
		#line_ids define in account.py
	}
	_defaults = {
		'name': lambda *a:time.strftime('%Y-%m-%d'),
		'partner_id': lambda s,cr,u,c={}: c.get('partner', False),
		'total_amount': lambda s,cr,u,c={}: c.get('amount', 0.0)
	}
account_bank_statement_reconcile()

class account_bank_statement_reconcile_line(osv.osv):
	_name = "account.bank.statement.reconcile.line"
	_description = "Statement reconcile line"
	_columns = {
		'name': fields.char('Description', size=64),
		'account_id': fields.many2one('account.account', 'Account', required=True),
		'line_id': fields.many2one('account.bank.statement.reconcile', 'Reconcile'),
		'amount': fields.float('Amount', required=True),
	}
account_bank_statement_reconcile_line()


class account_bank_statement_line(osv.osv):
	def onchange_partner_id(self, cr, uid, id, partner_id, type, context={}):
		if not partner_id:
			return {}
		part = self.pool.get('res.partner').browse(cr, uid, partner_id, context)
		if type=='supplier':
			account_id = part.property_account_payable.id
		else:
			account_id =  part.property_account_receivable.id
		cr.execute('select sum(debit-credit) from account_move_line where (reconcile_id is null) and partner_id=%d and account_id=%d', (partner_id, account_id))
		balance = cr.fetchone()[0] or 0.0
		val = {'amount': balance, 'account_id':account_id}
		return {'value':val}
	_order = "date,name desc"
	_name = "account.bank.statement.line"
	_description = "Bank Statement Line"
	_columns = {
		'name': fields.char('Name', size=64, required=True),
		'date': fields.date('Date', required=True),
		'amount': fields.float('Amount'),
		'type': fields.selection([('supplier','Supplier'),('customer','Customer'),('general','General')], 'Type', required=True),
		'partner_id': fields.many2one('res.partner', 'Partner'),
		'account_id': fields.many2one('account.account','Account', required=True),
		'statement_id': fields.many2one('account.bank.statement', 'Statement', select=True, required=True),
		'reconcile_id': fields.many2one('account.bank.statement.reconcile', 'Reconcile', states={'confirm':[('readonly',True)]}),
		'move_ids': fields.many2many('account.move', 'account_bank_statement_line_move_rel', 'move_id','statement_id', 'Moves'),
		'ref': fields.char('Ref.', size=32),
		'note': fields.text('Notes')
	}
	_defaults = {
		'name': lambda self,cr,uid,context={}: self.pool.get('ir.sequence').get(cr, uid, 'account.bank.statement.line'),
		'date': lambda *a: time.strftime('%Y-%m-%d'),
		'type': lambda *a: 'general',
	}
account_bank_statement_line()