bug fix on membership

bzr revid: mra@tinyerp.com-20081217124800-6nrloa8zxdo8s0zo
This commit is contained in:
Mustufa Rangwala 2008-12-17 18:18:00 +05:30
parent 269e7a6ec1
commit b035f71692
2 changed files with 58 additions and 40 deletions

View File

@ -1,7 +1,7 @@
# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
# $Id$
#
@ -225,11 +225,11 @@ membership_line()
class Partner(osv.osv):
'''Partner'''
def _membership_state_search_inv(self, cr, uid, ids, context=None):
data_inv = self.pool.get('account.invoice').browse(cr, uid, ids, context)
def _get_partner_id(self, cr, uid, ids, context=None):
data_inv = self.pool.get('membership.membership_line').browse(cr, uid, ids, context)
list_partner = []
for data in data_inv:
list_partner.append(data.partner_id.id)
list_partner.append(data.partner.id)
return list_partner
def _membership_state(self, cr, uid, ids, name, args, context=None):
@ -237,7 +237,6 @@ class Partner(osv.osv):
for id in ids:
res[id] = 'none'
today = time.strftime('%Y-%m-%d')
for id in ids:
partner_data = self.browse(cr,uid,id)
if partner_data.membership_cancel and today > partner_data.membership_cancel:
@ -437,7 +436,8 @@ class Partner(osv.osv):
# 'membership_state': fields.function(_membership_state, method=True, string='Current membership state',
# type='selection', selection=STATE, fnct_search=_membership_state_search),
'membership_state': fields.function(_membership_state, method=True, string='Current membership state',
type='selection',selection=STATE,store={'account.invoice':(_membership_state_search_inv,['state'], 10)}),
type='selection',selection=STATE,store={'membership.membership_line':(_get_partner_id,['state'], 10),
'res.partner':(lambda self,cr,uid,ids,c={}:ids, ['free_member'], 10)}),
# 'associate_member': fields.many2one('res.partner', 'Associate member'),
'free_member': fields.boolean('Free member'),
'membership_start': fields.function(_membership_start, method=True,
@ -501,36 +501,53 @@ class Invoice(osv.osv):
_inherit = 'account.invoice'
def action_move_create(self, cr, uid, ids, context=None):
'''Create membership.membership_line if the product is for membership'''
if context is None:
context = {}
def create(self, cr, uid, vals, context={}):
result = super(Invoice, self).create(cr, uid, vals, context)
a = self.browse(cr, uid, result)
member_line_obj = self.pool.get('membership.membership_line')
partner_obj = self.pool.get('res.partner')
for invoice in self.browse(cr, uid, ids):
# fetch already existing member lines
former_mlines = member_line_obj.search(cr,uid,
[('account_invoice_line','in',
[ l.id for l in invoice.invoice_line])], context)
# empty them :
if former_mlines:
member_line_obj.write(cr,uid,former_mlines, {'account_invoice_line':False}, context)
for line in invoice.invoice_line:
if line.product_id and line.product_id.membership:
date_from = line.product_id.membership_date_from
date_to = line.product_id.membership_date_to
if invoice.date_invoice > date_from and invoice.date_invoice < date_to:
date_from = invoice.date_invoice
member_line_obj.create(cr, uid, {
'partner': invoice.partner_id.id,
for i in a.invoice_line:
if i.product_id and i.product_id.membership:
date_from = i.product_id.membership_date_from
date_to = i.product_id.membership_date_to
if a.date_invoice > date_from and a.date_invoice < date_to:
date_from = a.date_invoice
line_id = member_line_obj.create(cr, uid, {
'partner': a.partner_id.id,
'date_from': date_from,
'date_to': date_to,
'account_invoice_line': line.id,
'account_invoice_line': i.id,
})
return super(Invoice, self).action_move_create(cr, uid, ids, context)
return result
# def action_move_create(self, cr, uid, ids, context=None):
# '''Create membership.membership_line if the product is for membership'''
# if context is None:
# context = {}
# member_line_obj = self.pool.get('membership.membership_line')
# partner_obj = self.pool.get('res.partner')
# for invoice in self.browse(cr, uid, ids):
#
# # fetch already existing member lines
# former_mlines = member_line_obj.search(cr,uid,
# [('account_invoice_line','in',
# [ l.id for l in invoice.invoice_line])], context)
# # empty them :
# if former_mlines:
# member_line_obj.write(cr,uid,former_mlines, {'account_invoice_line':False}, context)
#
# for line in invoice.invoice_line:
# if line.product_id and line.product_id.membership:
# date_from = line.product_id.membership_date_from
# date_to = line.product_id.membership_date_to
# if invoice.date_invoice > date_from and invoice.date_invoice < date_to:
# date_from = invoice.date_invoice
# line_id = member_line_obj.create(cr, uid, {
# 'partner': invoice.partner_id.id,
# 'date_from': date_from,
# 'date_to': date_to,
# 'account_invoice_line': line.id,
# })
# return super(Invoice, self).action_move_create(cr, uid, ids, context)
def action_cancel(self, cr, uid, ids, context=None):
'''Create a 'date_cancel' on the membership_line object'''

View File

@ -1,7 +1,7 @@
# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
# $Id$
#
@ -60,15 +60,8 @@ def _invoice_membership(self, cr, uid, data, context):
for partner_id in partner_ids:
account_id = partner_obj.read(cr, uid, partner_id, ['property_account_receivable'])['property_account_receivable'][0]
invoice_id = invoice_obj.create(cr, uid, {
'partner_id' : partner_id,
'address_invoice_id': partner_address_ids[partner_id]['id'],
'account_id': account_id,
}
)
line_value = {
'invoice_id' : invoice_id,
'product_id' : product_id,
}
@ -81,12 +74,20 @@ def _invoice_membership(self, cr, uid, data, context):
tax_tab = [(6, 0, line_value['invoice_line_tax_id'])]
line_value['invoice_line_tax_id'] = tax_tab
invoice_line_id = invoice_line_obj.create(cr, uid, line_value)
invoice_id = invoice_obj.create(cr, uid, {
'partner_id' : partner_id,
'address_invoice_id': partner_address_ids[partner_id]['id'],
'account_id': account_id,
'invoice_line':[(6,0,[invoice_line_id])]
}
)
invoice_list.append(invoice_id)
if line_value['invoice_line_tax_id']:
invoice_obj.write(cr, uid, [invoice_id], {'tax_line':tax_tab})
tax_value = invoice_tax_obj.compute(cr, uid, invoice_id).values()[0]
invoice_tax_obj.create(cr, uid, tax_value)
value = {
'domain': [
('id', 'in', invoice_list),