Files
account-reconcile/account_statement_one_move/statement.py

241 lines
11 KiB
Python

# -*- coding: utf-8 -*-
###############################################################################
#
# account_statement_one_move for OpenERP
# Copyright (C) 2013-TODAY Akretion <http://www.akretion.com>.
# @author Sébastien BEAU <sebastien.beau@akretion.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
###############################################################################
from openerp.osv import fields, orm, osv
from openerp.tools.translate import _
class AccountStatementProfile(orm.Model):
_inherit = "account.statement.profile"
_columns = {
'one_move': fields.boolean(
'Group Journal Items',
help="Only one Journal Entry will be generated on the "
"validation of the bank statement."),
'split_transfer_line': fields.boolean(
'Split Transfer Line',
help="Two transfer lines will be automatically generated : one "
"for the refunds and one for the payments.")
}
class account_bank_statement(orm.Model):
_inherit = "account.bank.statement"
def _prepare_move_line_vals(self, cr, uid, st_line, *args, **kwargs):
res = super(account_bank_statement, self)._prepare_move_line_vals(cr, uid, st_line,
*args, **kwargs)
period_id = self._get_period(cr, uid, st_line.statement_id.date,
context=kwargs.get('context'))
if st_line.statement_id.profile_id.one_move:
res.update({
'period_id': period_id,
'date': st_line.statement_id.date,
'name': st_line.ref,
})
return res
return res
def _prepare_move(self, cr, uid, st_line, st_line_number, context=None):
res = super(account_bank_statement, self).\
_prepare_move(cr, uid, st_line, st_line_number, context=context)
res.update({
'ref': st_line.statement_id.name,
'name': st_line.statement_id.name,
'date': st_line.statement_id.date,
})
return res
def create_move_from_st_line(self, cr, uid, st_line_id, company_currency_id,
st_line_number, context=None):
if context is None:
context = {}
account_move_obj = self.pool.get('account.move')
account_bank_statement_line_obj = self.pool.get('account.bank.statement.line')
st_line = account_bank_statement_line_obj.browse(cr, uid, st_line_id,
context=context)
st = st_line.statement_id
if st.profile_id.one_move:
if not context.get('move_id'):
move_vals = self._prepare_move(cr, uid, st_line, st_line_number, context=context)
context['move_id'] = account_move_obj.create(cr, uid, move_vals, context=context)
self.create_move_line_from_st_line(cr, uid, context['move_id'],
st_line_id, company_currency_id,
context=context)
return context['move_id']
else:
return super(account_bank_statement, self).create_move_from_st_line(cr, uid, st_line_id,
company_currency_id,
st_line_number,
context=context)
def create_move_line_from_st_line(self, cr, uid, move_id, st_line_id,
company_currency_id, context=None):
"""Create the account move line from the statement line.
:param int/long move_id: ID of the account.move
:param int/long st_line_id: ID of the account.bank.statement.line to create the move line from.
:param int/long company_currency_id: ID of the res.currency of the company
:return: ID of the account.move created
"""
if context is None:
context = {}
res_currency_obj = self.pool.get('res.currency')
account_move_line_obj = self.pool.get('account.move.line')
account_bank_statement_line_obj = self.pool.get('account.bank.statement.line')
st_line = account_bank_statement_line_obj.browse(cr, uid, st_line_id, context=context)
st = st_line.statement_id
context.update({'date': st_line.date})
acc_cur = ((st_line.amount<=0) and st.journal_id.default_debit_account_id) or st_line.account_id
context.update({
'res.currency.compute.account': acc_cur,
})
amount = res_currency_obj.compute(cr, uid, st.currency.id,
company_currency_id,
st_line.amount,
context=context)
bank_move_vals = self._prepare_bank_move_line(cr, uid, st_line, move_id, amount,
company_currency_id, context=context)
return account_move_line_obj.create(cr, uid, bank_move_vals, context=context)
def _valid_move(self, cr, uid, move_id, context=None):
move_obj = self.pool.get('account.move')
move = move_obj.browse(cr, uid, move_id, context=context)
move_obj.post(cr, uid, [move_id], context=context)
return True
def create_statement_payment_line(self, cr, uid, st, payment_amount, context=None):
st_line_obj = self.pool.get('account.bank.statement.line')
payment_account_id = st.profile_id.journal_id.default_credit_account_id.id
partner_id = st.profile_id.partner_id and profile.partner_id.id or False
vals = {
'name': _('Payment Transfer'),
'date': st.date,
'amount': -payment_amount,
'partner_id': partner_id,
'type': 'general',
'statement_id': st.id,
'account_id': payment_account_id,
'ref': _('Transfer'),
'already_completed': True
}
payment_line_id = st_line_obj.create(cr, uid, vals, context=context)
return payment_line_id
def create_statement_refund_line(self, cr, uid, st, refund_amount, context=None):
st_line_obj = self.pool.get('account.bank.statement.line')
refund_account_id = st.profile_id.journal_id.default_credit_account_id.id
partner_id = st.profile_id.partner_id and profile.partner_id.id or False
vals = {
'name': _('Refund Transfer'),
'date': st.date,
'amount': -refund_amount,
'partner_id': partner_id,
'type': 'general',
'statement_id': st.id,
'account_id': refund_account_id,
'ref': _('Transfer'),
'already_completed': True
}
refund_line_id = st_line_obj.create(cr, uid, vals, context=context)
return refund_line_id
def prepare_statement_transfer_lines(self, cr, uid, st, context=None):
refund_amount = 0.0
payment_amount = 0.0
transfer_line_ids = []
#Calculate the part of the refund amount and the payment amount
for st_line in st.line_ids:
if st_line.amount < 0.0:
refund_amount += st_line.amount
else:
payment_amount += st_line.amount
#Create 2 Transfer lines or One global tranfer line
if st.profile_id.split_transfer_line:
transfer_line_ids.append(self.create_statement_refund_line(cr, uid, st,
refund_amount,
context=context))
transfer_line_ids.append(self.create_statement_payment_line(cr, uid, st,
payment_amount,
context=context))
else:
global_amount = refund_amount + payment_amount
#The global transfer line can be a refund or a payment transfer
if global_amount < 0.00:
transfer_line_ids.append(self.create_statement_refund_line(cr, uid, st,
global_amount,
context=context))
else:
transfer_line_ids.append(self.create_statement_payment_line(cr, uid, st,
global_amount,
context=context))
return transfer_line_ids
def button_confirm_bank(self, cr, uid, ids, context=None):
st_line_obj = self.pool.get('account.bank.statement.line')
if context is None:
context = {}
for st in self.browse(cr, uid, ids, context=context):
if st.profile_id.one_move:
refund_line_ids = self.prepare_statement_transfer_lines(cr, uid, st,
context=context)
super(account_bank_statement, self).button_confirm_bank(cr, uid, ids,
context=context)
if st.profile_id.one_move:
move_id = context['move_id']
self._valid_move(cr, uid, move_id, context=context)
lines_ids = [x.id for x in st.line_ids]
st_line_obj.write(cr, uid, lines_ids,
{'move_ids': [(4, move_id, False)]},
context=context)
return True
def button_cancel(self, cr, uid, ids, context=None):
done = []
for st in self.browse(cr, uid, ids, context=context):
if st.profile_id.one_move:
assert st.line_ids, "This statement does not contain any lines"
for move in st.line_ids[0].move_ids:
if move.state != 'draft':
move.button_cancel(context=context)
move.unlink(context=context)
st.write({'state':'draft'}, context=context)
else:
super(account_bank_statement, self).button_cancel(cr, uid, ids,
context=context)
return True