[date2date] date_from choice in the wizard

[burette/remembership.git] / wizard / remembership_invoice.py

diff --git a/wizard/remembership_invoice.py b/wizard/remembership_invoice.py
new file mode 100644 (file)
index 0000000..99fab8b
--- /dev/null
+++ b/wizard/remembership_invoice.py
@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+
+from openerp.osv import osv
+from openerp.osv import fields
+from datetime import datetime
+
+
+class membership_invoice(osv.osv):
+    _inherit = 'membership.invoice'
+
+    _columns = {
+        'date_from': fields.date('This membership start date'),
+        'is_date2date': fields.boolean('This membership product is date to date.', readonly=True),
+    }
+
+    _defaults = {
+        'date_from': datetime.today().strftime('%Y-%m-%d'),
+    }
+
+    def onchange_product(self, cr, uid, ids, product_id=False):
+        res = super(membership_invoice, self).onchange_product(cr, uid, ids, product_id)
+        if product_id:
+            res['value']['is_date2date'] = self.pool.get('product.product').browse(cr, uid, [product_id], context=None)[0].membership_date2date
+        else:
+            res['value']['is_date2date'] = False
+
+        return res
+
+    def membership_invoice(self, cr, uid, ids, context=None):
+        """NOTE:  Copy of orginial function for overwrite it by adding
+        date_from value for date to date membership products. Keep an eye of it
+        for taking advantage of its enhancements (2013-01-11)"""
+        mod_obj = self.pool.get('ir.model.data')
+        partner_obj = self.pool.get('res.partner')
+        datas = {}
+        if context is None:
+            context = {}
+        data = self.browse(cr, uid, ids, context=context)
+        if data:
+            data = data[0]
+            datas = {
+                'membership_product_id': data.product_id.id,
+                'amount': data.member_price,
+                'date_from': data.date_from
+            }
+        invoice_list = partner_obj.create_membership_invoice(cr, uid, context.get('active_ids', []), datas=datas, context=context)
+
+        res = mod_obj.get_object_reference(cr, uid, 'account', 'view_account_invoice_filter')
+
+        return {
+            'domain': [('id', 'in', invoice_list)],
+            'name': 'Membership Invoices',
+            'view_type': 'form',
+            'view_mode': 'tree,form',
+            'res_model': 'account.invoice',
+            'type': 'ir.actions.act_window',
+            'search_view_id': res and res[1] or False
+        }
+
+membership_invoice()