Merge lp:~openbig/bigconsulting/changes_account_invoice_discount into lp:bigconsulting

[gpa(OpenERP)]

Changes on the account_invoice_cash_discount

[lp:~openbig/bigconsulting/changes_account_invoice_discount]

7. By gpa(OpenERP) on 2010-05-27

changes on the account_invoice_cash_discount

[Husen Daudi (husendaudi)]

In __terp__.py
Website should be : http://www.openbig.org
there must be short description.

In account_invoice_cash_discount.py useless import statements are there.
Clean up code
import mx.DateTime
from mx.DateTime import RelativeDateTime

Wizard_pay_invoice.py
useless import : import netsvc

clean up code and resubmit merge.

[lp:~openbig/bigconsulting/changes_account_invoice_discount]

8. By gpa(OpenERP) on 2010-05-27

Remove useless import statements and added Description

9. By gpa(OpenERP) on 2010-05-27

change in get_payment function

=== added directory 'account_invoice_cash_discount'
=== added file 'account_invoice_cash_discount/__init__.py'
--- account_invoice_cash_discount/__init__.py	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/__init__.py	2010-05-27 07:44:26 +0000
@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution	
+#    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    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 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+import account_invoice_cash_discount
+import wizard
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
+

=== added file 'account_invoice_cash_discount/__terp__.py'
--- account_invoice_cash_discount/__terp__.py	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/__terp__.py	2010-05-27 07:44:26 +0000
@@ -0,0 +1,45 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution	
+#    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    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 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+{
+    "name" : "Payement Term with Cash Discount",
+    "version" : "1.0",
+    "depends" : ["account",],
+    "author" : "Big Consulting",
+    "website" : "",
+    "category" : "",
+    "description": """
+    The Module add automatically account move lines in case of payment with cash discount conditions in reconcilation form.
+    """,
+    "init_xml" : [
+    ],
+    "demo_xml" : [
+    ],
+    "update_xml" : [
+        "account_invoice_cash_discount_view.xml",
+        "account_wizard.xml"
+    ],
+    "active": False,
+    "installable": True,
+
+}
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
+

=== added file 'account_invoice_cash_discount/account_invoice_cash_discount.py'
--- account_invoice_cash_discount/account_invoice_cash_discount.py	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/account_invoice_cash_discount.py	2010-05-27 07:44:26 +0000
@@ -0,0 +1,195 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution	
+#    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    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 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+from osv import fields, osv
+from tools import config
+import time
+from mx import DateTime
+
+class account_payment_term(osv.osv):
+    _name = "account.payment.term"
+    _inherit = "account.payment.term"
+    _columns = {
+        'cash_discount_ids': fields.one2many('account.cash.discount', 'payment_id', 'Cash Discounts'),
+    }
+account_payment_term()
+
+class account_cash_discount(osv.osv):
+    _name = "account.cash.discount"
+    _description = "Cash Discount" #A reduction in the price  if payment is made within a stipulated period.
+    _columns = {
+        'name': fields.char('Name', size=32),
+        'delay': fields.integer('Number of Days', required=True),
+        'discount': fields.float('Discount (%)', digits=(16,6),required=True),
+        'payment_id': fields.many2one('account.payment.term','Associated Payment Term'),
+        'credit_account_id': fields.many2one('account.account', 'Credit Account'),
+        'debit_account_id': fields.many2one('account.account', 'Debit Account'),
+    }
+account_cash_discount()
+
+class account_invoice(osv.osv):
+    _inherit = "account.invoice"
+    
+    def _get_payment(self, cr, uid, ids, resudial_amonut, payment_term, context=None):
+
+        #This function used to count the discount payment term
+        if context is None:
+           context = {}
+           
+        tax_obj = self.pool.get('account.tax')
+        invoice = self.browse(cr, uid, ids[0], context=context)
+        
+        if invoice.date_due:
+            date1 = invoice.date_due
+        else:
+            date1 =  time.strftime('%Y-%m-%d')
+                
+        if 'date_p' in context and context['date_p']:    
+            date2 = context['date_p']
+        else:
+            date2 =  time.strftime('%Y-%m-%d')   
+
+        from_dt = time.mktime(time.strptime(date1,'%Y-%m-%d'))
+        to_dt = time.mktime(time.strptime(date2,'%Y-%m-%d'))
+        diff_day = (from_dt-to_dt)/(3600*24)
+        if payment_term:
+            pt = self.pool.get('account.payment.term').browse(cr, uid, payment_term, context=context)
+            if pt.cash_discount_ids:
+                    res = 0.0
+                    for d in pt.cash_discount_ids:
+                        if diff_day == d.delay or diff_day < d.delay:
+                            if resudial_amonut:
+                                res = resudial_amonut * d.discount
+                            ted = 0.0
+                            for line in invoice.invoice_line:
+                                if line.invoice_line_tax_id:
+                                    tt = tax_obj.compute_inv(cr, uid, line.invoice_line_tax_id, res, line.quantity)
+                                    for tex in tt:
+                                        rr = 0.0 
+                                        rr += tex['amount']
+                                    ted +=rr
+                            discunt = res + ted
+                            return discunt
+            
+    def pay_and_reconcile(self, cr, uid, ids, pay_amount, pay_account_id, period_id, pay_journal_id, writeoff_acc_id, writeoff_period_id, writeoff_journal_id, context=None, name=''):
+        
+        if context is None:
+            context = {}
+        #TODO check if we can use different period for payment and the writeoff line
+        assert len(ids)==1, "Can only pay one invoice at a time"
+        invoice = self.browse(cr, uid, ids[0])
+        
+        if 'amount_currency' in context and context['amount_currency']:
+            amount_currency = context['amount_currency']
+        else:
+            amount_currency = invoice.residual
+
+        if invoice.payment_term:
+            # Return the discount on for the payment term
+            discunt1 = self._get_payment(cr, uid, ids, amount_currency, invoice.payment_term.id, context=context)
+
+        src_account_id = invoice.account_id.id
+        # Take the seq as name for move
+        types = {'out_invoice': -1, 'in_invoice': 1, 'out_refund': 1, 'in_refund': -1}
+        direction = types[invoice.type]
+        #take the choosen date
+        if 'date_p' in context and context['date_p']:
+            date=context['date_p']
+        else:
+            date=time.strftime('%Y-%m-%d')
+            
+        # Take the amount in currency and the currency of the payment
+        if 'amount_currency' in context and context['amount_currency'] and 'currency_id' in context and context['currency_id']:
+            amount_currency = context['amount_currency']
+            currency_id = context['currency_id']
+        else:
+            amount_currency = False
+            currency_id = False
+        
+        if invoice.type in ('in_invoice', 'in_refund'):
+            ref = invoice.reference
+        else:
+            ref = self._convert_ref(cr, uid, invoice.number)        
+        # Pay attention to the sign for both debit/credit AND amount_currency
+        l1 = {
+            'debit': direction * (pay_amount)>0 and direction * (pay_amount),
+            'credit': direction * (pay_amount)<0 and - direction * (pay_amount),
+            'account_id': src_account_id,
+            'partner_id': invoice.partner_id.id,
+            'ref':ref,
+            'date': date,
+            'currency_id':currency_id,
+            'amount_currency':amount_currency and direction * amount_currency or 0.0,
+        }
+        l2 = {
+            'debit': direction * (pay_amount-discunt1)<0 and - direction * (pay_amount-discunt1),
+            'credit': direction * (pay_amount-discunt1)>0 and direction * (pay_amount-discunt1),
+            'account_id': pay_account_id,
+            'partner_id': invoice.partner_id.id,
+            'ref':ref,
+            'date': date,
+            'currency_id':currency_id,
+            'amount_currency':amount_currency and - direction * amount_currency or 0.0,
+        }
+        l3 = {
+            'debit': direction * discunt1<0 and - direction * discunt1,
+            'credit': direction * discunt1>0 and  direction * discunt1,
+            'account_id': context['account_id'],
+            'partner_id': invoice.partner_id.id,
+            'ref':ref,
+            'date': date,
+            'currency_id':currency_id,
+            'amount_currency':amount_currency and direction * amount_currency or 0.0,
+        }
+
+        if not name:
+            name = invoice.invoice_line and invoice.invoice_line[0].name or invoice.number
+        l1['name'] = name
+        l2['name'] = name
+        l3['name'] = name
+        lines = [(0, 0, l1), (0, 0, l2), (0, 0, l3)]
+
+        move = {'ref': ref, 'line_id': lines, 'journal_id': pay_journal_id, 'period_id': period_id, 'date': date}
+        move_id = self.pool.get('account.move').create(cr, uid, move, context=context)
+
+        line_ids = []
+        total = 0.0
+        line = self.pool.get('account.move.line')
+        cr.execute('select id from account_move_line where move_id in ('+str(move_id)+','+str(invoice.move_id.id)+')')
+        lines = line.browse(cr, uid, map(lambda x: x[0], cr.fetchall()) )
+        for l in lines+invoice.payment_ids:
+            if l.account_id.id==src_account_id:
+                line_ids.append(l.id)
+                total += (l.debit or 0.0) - (l.credit or 0.0)
+        if (not round(total,int(config['price_accuracy']))) or writeoff_acc_id:
+            self.pool.get('account.move.line').reconcile(cr, uid, line_ids, 'manual', writeoff_acc_id, writeoff_period_id, writeoff_journal_id, context)
+        else:
+            self.pool.get('account.move.line').reconcile_partial(cr, uid, line_ids, 'manual', context)
+
+        # Update the stored value (fields.function), so we write to trigger recompute
+        self.pool.get('account.invoice').write(cr, uid, ids, {}, context=context)
+        return True
+    
+account_invoice()
+
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
+

=== added file 'account_invoice_cash_discount/account_invoice_cash_discount_view.xml'
--- account_invoice_cash_discount/account_invoice_cash_discount_view.xml	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/account_invoice_cash_discount_view.xml	2010-05-27 07:44:26 +0000
@@ -0,0 +1,47 @@
+<?xml version="1.0"?>
+<openerp>
+    <data>
+
+        <!-- cash discount -->
+
+
+        <record model="ir.ui.view" id="view_cash_discount_form">
+            <field name="name">account.cash.discount.form</field>
+            <field name="model">account.cash.discount</field>
+            <field name="type">form</field>
+            <field name="arch" type="xml">
+                <form string="Cash Discount">
+                    <field name="name" select="1"/>
+                    <field name="delay" select="1"/>
+                    <field name="discount" select="1"/>
+                </form>
+            </field>
+        </record>
+
+        <record model="ir.ui.view" id="view_cash_discount_tree">
+            <field name="name">account.cash.discount.tree</field>
+            <field name="model">account.cash.discount</field>
+            <field name="type">tree</field>
+            <field name="arch" type="xml">
+                <tree string="Cash Discount" editable="bottom">
+                    <field name="name" select="1"/>
+                    <field name="delay" select="1"/>
+                    <field name="discount" select="1"/>
+                </tree>
+            </field>
+        </record>
+
+        <record model="ir.ui.view" id="view_payment_term_form">
+            <field name="name">account.payment.term.form</field>
+            <field name="model">account.payment.term</field>
+            <field name="inherit_id" ref="account.view_payment_term_form"/>
+            <field name="arch" type="xml">
+                <field name="line_ids" position="after">
+                    <separator string="Cash Discount" colspan="4"/>
+                    <field name="cash_discount_ids" colspan="4"/>
+                </field>
+            </field>
+        </record>
+
+    </data>
+</openerp>

=== added file 'account_invoice_cash_discount/account_wizard.xml'
--- account_invoice_cash_discount/account_wizard.xml	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/account_wizard.xml	2010-05-27 07:44:26 +0000
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="utf-8"?>
+<openerp>
+    <data>
+        <wizard id="wizard_invoice_pay" model="account.invoice" name="account.invoice.pay1" string="Pay invoice" groups="base.group_user"/>
+    </data>
+</openerp>

=== added directory 'account_invoice_cash_discount/wizard'
=== added file 'account_invoice_cash_discount/wizard/__init__.py'
--- account_invoice_cash_discount/wizard/__init__.py	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/wizard/__init__.py	2010-05-27 07:44:26 +0000
@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution
+#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    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 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+import wizard_pay_invoice
+
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
+

=== added file 'account_invoice_cash_discount/wizard/wizard_pay_invoice.py'
--- account_invoice_cash_discount/wizard/wizard_pay_invoice.py	1970-01-01 00:00:00 +0000
+++ account_invoice_cash_discount/wizard/wizard_pay_invoice.py	2010-05-27 07:44:26 +0000
@@ -0,0 +1,203 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution
+#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    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 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+import wizard
+import pooler
+import time
+from tools.translate import _
+import tools
+
+pay_form = '''<?xml version="1.0"?>
+<form string="Pay invoice">
+    <field name="amount"/>
+    <newline/>
+    <field name="name"/>
+    <field name="date"/>
+    <field name="journal_id"/>
+    <field name="period_id"/>
+    <field name="cash_amount"/>
+    <field name="account_id"/>
+</form>'''
+
+pay_fields = {
+    'amount': {'string': 'Amount paid', 'type':'float', 'required':True, 'digits': (16,int(tools.config['price_accuracy']))},
+    'name': {'string': 'Entry Name', 'type':'char', 'size': 64, 'required':True},
+    'cash_amount': {'string': 'Cash Discount Amount', 'type':'float', 'required':True, 'digits': (16,int(tools.config['price_accuracy']))},
+    'account_id': {'string': 'Account For Discount', 'type': 'many2one', 'relation':'account.account', 'required':True},
+    'date': {'string': 'Payment date', 'type':'date', 'required':True, 'default':lambda *args: time.strftime('%Y-%m-%d')},
+    'journal_id': {'string': 'Journal/Payment Mode', 'type': 'many2one', 'relation':'account.journal', 'required':True, 'domain':[('type','=','cash')]},
+    'period_id': {'string': 'Period', 'type': 'many2one', 'relation':'account.period', 'required':True},
+}
+
+def _pay_and_reconcile(self, cr, uid, data, context):
+    form = data['form']
+    period_id = form.get('period_id', False)
+    journal_id = form.get('journal_id', False)
+    writeoff_account_id = form.get('writeoff_acc_id', False)
+    writeoff_journal_id = form.get('writeoff_journal_id', False)
+    pool = pooler.get_pool(cr.dbname)
+    cur_obj = pool.get('res.currency')
+    amount = form['amount']
+    context['analytic_id'] = form.get('analytic_id', False)
+
+    invoice = pool.get('account.invoice').browse(cr, uid, data['id'], context)
+    journal = pool.get('account.journal').browse(cr, uid, data['form']['journal_id'], context)
+    # Compute the amount in company's currency, with the journal currency (which is equal to payment currency) 
+    # when it is needed :  If payment currency (according to selected journal.currency) is <> from company currency
+    if journal.currency and invoice.company_id.currency_id.id<>journal.currency.id:
+        ctx = {'date':data['form']['date']}
+        amount = cur_obj.compute(cr, uid, journal.currency.id, invoice.company_id.currency_id.id, amount, context=ctx)
+        currency_id = journal.currency.id
+        # Put the paid amount in currency, and the currency, in the context if currency is different from company's currency
+        context.update({'amount_currency':form['amount'],'currency_id':currency_id})
+    
+    if invoice.company_id.currency_id.id<>invoice.currency_id.id:
+        ctx = {'date':data['form']['date']}
+        amount = cur_obj.compute(cr, uid, invoice.currency_id.id, invoice.company_id.currency_id.id, amount, context=ctx)
+        currency_id = invoice.currency_id.id
+        # Put the paid amount in currency, and the currency, in the context if currency is different from company's currency
+        context.update({'amount_currency':form['amount'],'currency_id':currency_id})
+        
+    # Take the choosen date
+    if form.has_key('comment'):
+        context.update({'date_p':form['date'],'comment':form['comment']})
+    else:
+        context.update({'date_p':form['date'],'comment':False})      
+    
+    context.update({'account_id':form['account_id'],'cash_amount':form['cash_amount'],'amount_currency':form['amount']})    
+    
+    acc_id = journal.default_credit_account_id and journal.default_credit_account_id.id
+    if not acc_id:
+        raise wizard.except_wizard(_('Error !'), _('Your journal must have a default credit and debit account.'))
+    pool.get('account.invoice').pay_and_reconcile(cr, uid, [data['id']],
+            amount, acc_id, period_id, journal_id, writeoff_account_id,
+            period_id, writeoff_journal_id, context, data['form']['name'])
+    return {}
+
+def _wo_check(self, cr, uid, data, context):
+    pool = pooler.get_pool(cr.dbname)
+    invoice = pool.get('account.invoice').browse(cr, uid, data['id'], context)
+    journal = pool.get('account.journal').browse(cr, uid, data['form']['journal_id'], context)
+    cur_obj = pool.get('res.currency')
+    # Here we need that:
+    #    The invoice total amount in company's currency <> paid amount in company currency
+    #    (according to the correct day rate, invoicing rate and payment rate are may be different)
+    #    => Ask to a write-off of the difference. This could happen even if both amount are equal,
+    #    because if the currency rate
+    # Get the amount in company currency for the invoice (according to move lines)
+    inv_amount_company_currency = 0
+    for aml in invoice.move_id.line_id:
+        if aml.account_id.id == invoice.account_id.id or aml.account_id.type in ('receivable', 'payable'):
+            inv_amount_company_currency += aml.debit
+            inv_amount_company_currency -= aml.credit
+    inv_amount_company_currency = abs(inv_amount_company_currency)
+
+    # Get the current amount paid in company currency
+    if journal.currency and invoice.company_id.currency_id.id<>journal.currency.id:
+        ctx = {'date':data['form']['date']}
+        amount_paid = cur_obj.compute(cr, uid, journal.currency.id, invoice.company_id.currency_id.id, data['form']['amount'], round=True, context=ctx)
+    else:
+        amount_paid = data['form']['amount']
+    # Get the old payment if there are some
+    if invoice.payment_ids:
+        debit=credit=0.0
+        for payment in invoice.payment_ids:
+            debit+=payment.debit
+            credit+=payment.credit
+        amount_paid+=abs(debit-credit)
+        
+    # Test if there is a difference according to currency rouding setting
+    if pool.get('res.currency').is_zero(cr, uid, invoice.company_id.currency_id,
+            (amount_paid - inv_amount_company_currency)):
+        return 'reconcile'
+    return 'addendum'
+
+_transaction_add_form = '''<?xml version="1.0"?>
+<form string="Information addendum">
+    <separator string="Write-Off Move" colspan="4"/>
+    <field name="writeoff_journal_id"/>
+    <field name="writeoff_acc_id" domain="[('type','&lt;&gt;','view'),('type','&lt;&gt;','consolidation')]"/>
+    <field name="comment"/>
+    <separator string="Analytic" colspan="4"/>
+    <field name="analytic_id"/>
+</form>'''
+
+_transaction_add_fields = {
+    'writeoff_acc_id': {'string':'Write-Off account', 'type':'many2one', 'relation':'account.account', 'required':True},
+    'writeoff_journal_id': {'string': 'Write-Off journal', 'type': 'many2one', 'relation':'account.journal', 'required':True},
+    'comment': {'string': 'Comment', 'type':'char', 'size': 64 , 'required':True},
+    'analytic_id': {'string':'Analytic Account', 'type': 'many2one', 'relation':'account.analytic.account'},
+}
+
+def _get_value_addendum(self, cr, uid, data, context={}):
+    return {'comment': _('Write-Off')}
+
+def _compute(self, cr, uid, data, context={}):
+    pool = pooler.get_pool(cr.dbname)
+    invoice = pool.get('account.invoice').browse(cr, uid, data['id'], context)
+    
+
+def _get_period(self, cr, uid, data, context={}):
+    pool = pooler.get_pool(cr.dbname)
+    ids = pool.get('account.period').find(cr, uid, context=context)
+    period_id = False
+    if len(ids):
+        period_id = ids[0]
+    invoice = pool.get('account.invoice').browse(cr, uid, data['id'], context)
+    cash_amount = pool.get('account.invoice')._get_payment(cr, uid, [data['id']], invoice.residual, invoice.payment_term.id, context=context)
+    if invoice.state in ['draft', 'proforma2', 'cancel']:
+        raise wizard.except_wizard(_('Error !'), _('Can not pay draft/proforma/cancel invoice.'))
+    return {
+        'period_id': period_id,
+        'amount': invoice.residual,
+        'date': time.strftime('%Y-%m-%d'),
+        'cash_amount': cash_amount,
+    }
+
+class wizard_pay_invoice(wizard.interface):
+    states = {
+        'init': {
+            'actions': [_get_period],
+            'result': {'type':'form', 'arch':pay_form, 'fields':pay_fields, 'state':[('end','Cancel'),('reconcile','Partial Payment'),('writeoff_check','Full Payment')]}
+        },
+        'writeoff_check': {
+            'actions': [],
+            'result' : {'type': 'choice', 'next_state': _wo_check }
+        },
+        'addendum': {
+            'actions': [_get_value_addendum],
+            'result': {'type': 'form', 'arch':_transaction_add_form, 'fields':_transaction_add_fields, 'state':[('end','Cancel'),('reconcile','Pay and reconcile')]}
+        },
+        'reconcile': {
+            'actions': [_pay_and_reconcile],
+            'result': {'type':'state', 'state':'end'}
+        },
+        'compute_dis': {
+            'actions': [_compute],
+            'result': {'type':'state', 'state':'end'}
+        }
+    }
+wizard_pay_invoice('account.invoice.pay1')
+
+
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
+