From 421ec99363edaace7f9a1e5611df7a3323271f36 Mon Sep 17 00:00:00 2001
From: "watcha.h" <watcha.h@almacom.co.th>
Date: Tue, 8 Mar 2016 18:07:50 +0700
Subject: [PATCH] add action to create invoice payment

---
 .../actions/create_invoice_payment.xml        |   6 +
 .../layouts/clinic_hdcase_payment_form.xml    |  16 +++
 .../layouts/clinic_import_payment_form.xml    |   5 -
 .../layouts/clinic_payment_form.xml           |  21 +--
 .../layouts/create_invoice_payment_form.xml   |  18 +++
 netforce_clinic/layouts/payment_list.xml      |  14 ++
 netforce_clinic/models/__init__.py            |   1 +
 netforce_clinic/models/account_invoice.py     |   1 +
 netforce_clinic/models/account_payment.py     |   2 +-
 .../models/create_invoice_payment.py          | 131 ++++++++++++++++++
 netforce_clinic/models/partner.py             |   2 +-
 11 files changed, 194 insertions(+), 23 deletions(-)
 create mode 100644 netforce_clinic/actions/create_invoice_payment.xml
 create mode 100644 netforce_clinic/layouts/clinic_hdcase_payment_form.xml
 delete mode 100644 netforce_clinic/layouts/clinic_import_payment_form.xml
 create mode 100644 netforce_clinic/layouts/create_invoice_payment_form.xml
 create mode 100644 netforce_clinic/layouts/payment_list.xml
 create mode 100644 netforce_clinic/models/create_invoice_payment.py

diff --git a/netforce_clinic/actions/create_invoice_payment.xml b/netforce_clinic/actions/create_invoice_payment.xml
new file mode 100644
index 0000000..fdc68e6
--- /dev/null
+++ b/netforce_clinic/actions/create_invoice_payment.xml
@@ -0,0 +1,6 @@
+<action>                                                                                                         
+    <field name="view_cls">form_view</field>                                                                        
+    <field name="model">create.invoice.payment</field>
+    <field name="view_xml">create_invoice_payment_form</field>                                                      
+    <field name="menu">account_menu</field>                                                                  
+</action>  
diff --git a/netforce_clinic/layouts/clinic_hdcase_payment_form.xml b/netforce_clinic/layouts/clinic_hdcase_payment_form.xml
new file mode 100644
index 0000000..398842b
--- /dev/null
+++ b/netforce_clinic/layouts/clinic_hdcase_payment_form.xml
@@ -0,0 +1,16 @@
+<form model="clinic.hd.case.payment">
+    <field name="hd_case_id" invisible="1"/>
+    <field name="bill_no"/>
+    <field name="pay_amount" onchange="onchange_amount"/>
+    <newline/>
+    <field name="to_pay" readonly="1"/>
+    <newline/>
+    <group attrs='{"invisible":[["pay_amount","=",0]]}'>
+        <field name="complete" help="Mark as complete"/>
+    </group>
+    <foot>
+        <button attrs='{"invisible":[["pay_amount","=",0]]}' string="Cash" type="success" icon="ok" method="cash" confirm="Are you sure?"/>
+        <button attrs='{"invisible":[["pay_amount","=",0]]}' string="Credit" type="default" icon="check" method="credit"/>
+        <button attrs='{"invisible":[["pay_amount","!=",0]]}' string="OK" type="success" icon="ok" method="credit"/>
+    </foot>
+</form>
diff --git a/netforce_clinic/layouts/clinic_import_payment_form.xml b/netforce_clinic/layouts/clinic_import_payment_form.xml
deleted file mode 100644
index 3b0f39d..0000000
--- a/netforce_clinic/layouts/clinic_import_payment_form.xml
+++ /dev/null
@@ -1,5 +0,0 @@
-<inherit model="account.payment" inherit="payment_form">
-    <item string="Copy" position="after">
-        <item string="Import Payment (UC)" method="import_payment"/>
-    </item>
-</inherit>
diff --git a/netforce_clinic/layouts/clinic_payment_form.xml b/netforce_clinic/layouts/clinic_payment_form.xml
index 398842b..5ccf627 100644
--- a/netforce_clinic/layouts/clinic_payment_form.xml
+++ b/netforce_clinic/layouts/clinic_payment_form.xml
@@ -1,16 +1,5 @@
-<form model="clinic.hd.case.payment">
-    <field name="hd_case_id" invisible="1"/>
-    <field name="bill_no"/>
-    <field name="pay_amount" onchange="onchange_amount"/>
-    <newline/>
-    <field name="to_pay" readonly="1"/>
-    <newline/>
-    <group attrs='{"invisible":[["pay_amount","=",0]]}'>
-        <field name="complete" help="Mark as complete"/>
-    </group>
-    <foot>
-        <button attrs='{"invisible":[["pay_amount","=",0]]}' string="Cash" type="success" icon="ok" method="cash" confirm="Are you sure?"/>
-        <button attrs='{"invisible":[["pay_amount","=",0]]}' string="Credit" type="default" icon="check" method="credit"/>
-        <button attrs='{"invisible":[["pay_amount","!=",0]]}' string="OK" type="success" icon="ok" method="credit"/>
-    </foot>
-</form>
+<inherit inherit="payment_form">
+    <item string="Copy" position="after">
+        <item string="Import Payment (UC)" method="import_payment"/>
+    </item>
+</inherit>
diff --git a/netforce_clinic/layouts/create_invoice_payment_form.xml b/netforce_clinic/layouts/create_invoice_payment_form.xml
new file mode 100644
index 0000000..fea299a
--- /dev/null
+++ b/netforce_clinic/layouts/create_invoice_payment_form.xml
@@ -0,0 +1,18 @@
+<form model="create.invoice.payment">
+    <head>
+        <field name="state"/>
+    </head>
+    <field name="date" mode="month" onchange="onchange_date" span="3"/>
+    <field name="date_from" span="3"/>
+    <field name="date_to" span="3"/>
+    <newline/>
+    <field name="partner_id" span="3"/>
+    <field name="account_id" domain='[["type","!=","view"]]' states="ready_payment" span="3"/>
+    <related>
+        <field name="invoices"/>
+    </related>
+    <foot replace="1">
+        <button string="Back" states="ready_payment"  method="back_step" type="default" icon="arrow-left"/>
+        <button string="Next" method="do_next" type="primary" icon="arrow-right"/>
+    </foot>
+</form>
diff --git a/netforce_clinic/layouts/payment_list.xml b/netforce_clinic/layouts/payment_list.xml
new file mode 100644
index 0000000..740a2e0
--- /dev/null
+++ b/netforce_clinic/layouts/payment_list.xml
@@ -0,0 +1,14 @@
+<list model="account.payment">
+    <top>
+        <button string="Create Invoice Payment" action="create_invoice_payment"/>
+    </top>
+    <field name="number"/>
+    <field name="date"/>
+    <field name="partner_id"/>
+    <field name="type"/>
+    <field name="pay_type"/>
+    <field name="account_id"/>
+    <field name="ref"/>
+    <field name="amount_total"/>
+    <field name="state"/>
+</list>
diff --git a/netforce_clinic/models/__init__.py b/netforce_clinic/models/__init__.py
index a12468c..749868c 100644
--- a/netforce_clinic/models/__init__.py
+++ b/netforce_clinic/models/__init__.py
@@ -144,3 +144,4 @@ from . import account_move_line
 from . import invoice_payment
 from . import document
 from . import payment_matching
+from . import create_invoice_payment
diff --git a/netforce_clinic/models/account_invoice.py b/netforce_clinic/models/account_invoice.py
index 37613aa..fccc873 100644
--- a/netforce_clinic/models/account_invoice.py
+++ b/netforce_clinic/models/account_invoice.py
@@ -30,6 +30,7 @@ class AccountInvoice(Model):
         'patient_partner_id': fields.Many2One("partner","Partner Patient",search=True),
         'patient_id': fields.Many2One("clinic.patient","Patient",function="_get_patient", function_multi=True,store=True,search=True),
         'patient_type_id': fields.Many2One("clinic.patient.type","Patient Type",function="_get_patient", function_multi=True,store=True,search=True),
+        'create_invoice_id': fields.Many2One("create.invoice.payment","Create Invoice Payment"),
     }
 
     def _get_number(self,context={}):
diff --git a/netforce_clinic/models/account_payment.py b/netforce_clinic/models/account_payment.py
index 18e13af..329f0a2 100644
--- a/netforce_clinic/models/account_payment.py
+++ b/netforce_clinic/models/account_payment.py
@@ -31,7 +31,7 @@ class AccountPayment(Model):
         'number': _get_number,
     }
 
-    
+
     def run_report(self,ids,context={}):
         obj=self.browse(ids)[0]
         hd_case_id=obj.related_id.id
diff --git a/netforce_clinic/models/create_invoice_payment.py b/netforce_clinic/models/create_invoice_payment.py
new file mode 100644
index 0000000..1dd284c
--- /dev/null
+++ b/netforce_clinic/models/create_invoice_payment.py
@@ -0,0 +1,131 @@
+import time
+from calendar import monthrange
+
+from netforce.model import Model, fields , get_model
+from netforce.database import get_connection
+
+class CreateInvoicePayment(Model):
+    _name="create.invoice.payment"
+    _transient=True
+
+    _fields={
+        "date": fields.Date("Month", required=True),
+        'date_from': fields.Date("From",required=True),
+        'date_to': fields.Date("To",required=True),
+        'partner_id': fields.Many2One("partner","Contact",required=True),
+        'invoices': fields.One2Many("account.invoice","create_invoice_id","Invoices"),
+        'state': fields.Selection([["find_invoice","Find Invoice"],["ready_payment","Ready Payment"]],'State'),
+        'account_id': fields.Many2One("account.account","Account"),
+    }
+
+    def _get_date_from(self,context={}):
+        year,month=time.strftime("%Y-%m").split("-")
+        return '%s-%s-01'%(year,month)
+
+    def _get_date_to(self,context={}):
+        year,month,day=time.strftime("%Y-%m-%d").split("-")
+        weekday, total_day=monthrange(int(year), int(month))
+        return "%s-%s-%s"%(year,month,total_day)
+
+    _defaults={
+        'date': lambda *a: time.strftime("%Y-%m-%d"),        
+        'date_from': _get_date_from,
+        'date_to': _get_date_to,
+        'state': 'find_invoice',
+    }
+
+    def onchange_date(self,context={}):
+        data=context['data']
+        date=data['date']
+        year,month,day=date.split("-")
+        weekday, total_day=monthrange(int(year), int(month))
+        data['date_from']="%s-%s-01"%(year,month)
+        data['date_to']="%s-%s-%s"%(year,month,total_day)
+        return data
+
+    def back_step(self,ids,context={}):
+        obj=self.browse(ids)[0]
+        state='find_invoice'
+        if obj.state=='ready_payment':
+            db=get_connection()
+            res=db.query("""
+                select id from account_invoice where create_invoice_id=%s
+            """,obj.id)
+            if res:
+                inv_ids=[r['id'] for r in res]
+                db.execute("update account_invoice set create_invoice_id=null where id in %s",tuple(inv_ids))
+
+        obj.write({
+            'state': state,
+        })
+
+    def find_invoice(self,ids,context={}):
+        obj=self.browse(ids)[0]
+        obj.write({
+            'state': 'ready_payment',
+        })
+        db=get_connection()
+        res=db.query("""
+            select id from account_invoice where create_invoice_id=%s
+        """,obj.id)
+        if res:
+            inv_ids=[r['id'] for r in res]
+            db.execute("update account_invoice set create_invoice_id=null where id in %s",tuple(inv_ids))
+
+        res=db.query("""
+            select id from account_invoice where date>=%s and date <=%s and state='waiting_payment'
+            and partner_id=%s
+        """,obj.date_from, obj.date_to,obj.partner_id.id)
+        if res:
+            inv_ids=[r['id'] for r in res]
+            db.execute("update account_invoice set create_invoice_id=%s where id in %s",obj.id,tuple(inv_ids))
+        
+
+    def create_payment(self,ids,context={}):
+        obj=self.browse(ids)[0]
+        if not obj.account_id:
+            raise Exception("Missing account")
+        if not obj.invoices:
+            raise Exception("Nothing to pay!")
+        partner=obj.partner_id
+        type="in"
+        vals={
+            'type': type, # receivable
+            'account_id': obj.account_id.id,
+            'partner_id': partner.id,
+            'pay_type': 'invoice',
+            'invoice_lines': [],
+        }
+        datenow=time.strftime("%Y-%m-%d")
+        rate_type="sell" # receivable
+        settings=get_model("settings").browse(1)
+        currency=settings.currency_id
+        if not currency:
+            raise Exception("Missing setting.currency")
+        for inv in obj.invoices:
+            vals['invoice_lines'].append(('create',{
+                'invoice_id': inv.id,
+                "amount": get_model("currency").convert(inv.amount_due,inv.currency_id.id,currency.id,date=datenow,rate_type=rate_type),
+            }))
+        ctx={
+            'type': type,
+        }
+        payment_id=get_model("account.payment").create(vals,context=ctx)
+        return {
+            'next': {
+                'name': 'payment',
+                'mode': 'form',
+                'active_id': payment_id,
+            },
+            'flash': 'Create Payment Successfull',
+        }
+
+    def do_next(self,ids,context={}):
+        obj=self.browse(ids)[0]
+        if obj.state=='find_invoice':
+            obj.find_invoice()
+        elif obj.state=='ready_payment':
+            res=obj.create_payment()
+            return res
+
+CreateInvoicePayment.register()
diff --git a/netforce_clinic/models/partner.py b/netforce_clinic/models/partner.py
index 4d8cf32..844da35 100644
--- a/netforce_clinic/models/partner.py
+++ b/netforce_clinic/models/partner.py
@@ -24,7 +24,7 @@ class Partner(Model):
             vals.append((obj.id,name))
         return vals
 
-    def name_search(self,name,domain=None,condition=[],context={},**kw):
+    def _name_search(self,name,domain=None,condition=[],context={},**kw):
         dom=[["name","ilike","%"+name+"%"]]
         if domain:
              dom=[dom,domain]