wizard/remembership_invoice.py

   1 # -*- coding: utf-8 -*-
   2
   3 from openerp.osv import osv
   4 from openerp.osv import fields
   5 from datetime import datetime
   6
   7
   8 class membership_invoice(osv.osv):
   9     _inherit = 'membership.invoice'
  10
  11     _columns = {
  12         'date_from': fields.date('This membership start date'),
  13         'is_date2date': fields.boolean('This membership product is date to date.', readonly=True),
  14     }
  15
  16     _defaults = {
  17         'date_from': datetime.today().strftime('%Y-%m-%d'),
  18     }
  19
  20     def onchange_product(self, cr, uid, ids, product_id=False):
  21         res = super(membership_invoice, self).onchange_product(cr, uid, ids, product_id)
  22         if product_id:
  23             res['value']['is_date2date'] = self.pool.get('product.product').browse(cr, uid, [product_id], context=None)[0].membership_date2date
  24         else:
  25             res['value']['is_date2date'] = False
  26
  27         return res
  28
  29     def membership_invoice(self, cr, uid, ids, context=None):
  30         """NOTE:  Copy of orginial function for overwrite it by adding
  31         date_from value for date to date membership products. Keep an eye of it
  32         for taking advantage of its enhancements (2013-01-11)"""
  33         mod_obj = self.pool.get('ir.model.data')
  34         partner_obj = self.pool.get('res.partner')
  35         datas = {}
  36         if context is None:
  37             context = {}
  38         data = self.browse(cr, uid, ids, context=context)
  39         if data:
  40             data = data[0]
  41             datas = {
  42                 'membership_product_id': data.product_id.id,
  43                 'amount': data.member_price,
  44                 'date_from': data.date_from
  45             }
  46         invoice_list = partner_obj.create_membership_invoice(cr, uid, context.get('active_ids', []), datas=datas, context=context)
  47
  48         res = mod_obj.get_object_reference(cr, uid, 'account', 'view_account_invoice_filter')
  49
  50         return {
  51             'domain': [('id', 'in', invoice_list)],
  52             'name': 'Membership Invoices',
  53             'view_type': 'form',
  54             'view_mode': 'tree,form',
  55             'res_model': 'account.invoice',
  56             'type': 'ir.actions.act_window',
  57             'search_view_id': res and res[1] or False
  58         }
  59
  60 membership_invoice()