Goffin Simon
0fd773a486
Unify and refactor exception handling in framework and addons. The generic `except_osv` is now deprecated, and replaced by more specialized exception subtypes: - `UserError` (renamed from Warning, as it conflicts with the built-in `Warning`) raised when a non-technical error occurs during a business operation. It could be a missing information in the data provided by the user, or a misconfiguration. - `AccessError`: raised when any operation is denied because the user conducting it does not have the required access rights. - `AccessDenied`: raised when an operation that requires authenticated access is attempted via an unauthenticated request. - `MissingError`: raised when an operation is attempted on a record that does not exist. - `ValidationError`: raised when an operation violates a SQL or Python constraint. - All other exceptions are internal errors due to a system problem or bug, and raised untouched to the client-side, which should display a traceback. All exceptions take a single message argument. The `test_exceptions` module has been updated to showcase both new and old (deprecated) exceptions. A great many old `except_osv` had a useless title with "Error!" or "Warning", those have been removed, as this is handled by the client-side widget that displays the messages. This commit introduces a more consistent policy for logging errors and warnings: - All messages that do not require administrator attention should be logged at INFO level or lower. This includes all errors that are notified to the user in a friendly manner, even for access right problems or validation errors during business operations. - All messages that indicate a likely misconfiguration or malicious use by the users should be logged at WARNING level, as they typically require administrator attention. - All other unhandled internal errors cannot typically be handled by the user and should be logged at ERROR or higher level, as they require immediate administrator attention.
551 lines
27 KiB
Python
551 lines
27 KiB
Python
# -*- coding: utf-8 -*-
|
||
##############################################################################
|
||
#
|
||
# OpenERP, Open Source Management Solution
|
||
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU Affero 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 Affero General Public License for more details.
|
||
#
|
||
# You should have received a copy of the GNU Affero General Public License
|
||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
#
|
||
##############################################################################
|
||
|
||
import time
|
||
|
||
from openerp.osv import fields, osv
|
||
import openerp.addons.decimal_precision as dp
|
||
from openerp.tools.translate import _
|
||
from openerp.exceptions import UserError
|
||
|
||
STATE = [
|
||
('none', 'Non Member'),
|
||
('canceled', 'Cancelled Member'),
|
||
('old', 'Old Member'),
|
||
('waiting', 'Waiting Member'),
|
||
('invoiced', 'Invoiced Member'),
|
||
('free', 'Free Member'),
|
||
('paid', 'Paid Member'),
|
||
]
|
||
|
||
STATE_PRIOR = {
|
||
'none': 0,
|
||
'canceled': 1,
|
||
'old': 2,
|
||
'waiting': 3,
|
||
'invoiced': 4,
|
||
'free': 6,
|
||
'paid': 7
|
||
}
|
||
|
||
class membership_line(osv.osv):
|
||
'''Member line'''
|
||
|
||
def _get_partners(self, cr, uid, ids, context=None):
|
||
list_membership_line = []
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
for partner in self.pool.get('res.partner').browse(cr, uid, ids, context=context):
|
||
if partner.member_lines:
|
||
list_membership_line += member_line_obj.search(cr, uid, [('id', 'in', [ l.id for l in partner.member_lines])], context=context)
|
||
return list_membership_line
|
||
|
||
def _get_membership_lines(self, cr, uid, ids, context=None):
|
||
list_membership_line = []
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
for invoice in self.pool.get('account.invoice').browse(cr, uid, ids, context=context):
|
||
if invoice.invoice_line:
|
||
list_membership_line += member_line_obj.search(cr, uid, [('account_invoice_line', 'in', [ l.id for l in invoice.invoice_line])], context=context)
|
||
return list_membership_line
|
||
|
||
def _check_membership_date(self, cr, uid, ids, context=None):
|
||
"""Check if membership product is not in the past
|
||
@param self: The object pointer
|
||
@param cr: the current row, from the database cursor,
|
||
@param uid: the current user’s ID for security checks,
|
||
@param ids: List of Membership Line IDs
|
||
@param context: A standard dictionary for contextual values
|
||
"""
|
||
|
||
cr.execute('''
|
||
SELECT MIN(ml.date_to - ai.date_invoice)
|
||
FROM membership_membership_line ml
|
||
JOIN account_invoice_line ail ON (
|
||
ml.account_invoice_line = ail.id
|
||
)
|
||
JOIN account_invoice ai ON (
|
||
ai.id = ail.invoice_id)
|
||
WHERE ml.id IN %s''', (tuple(ids),))
|
||
res = cr.fetchall()
|
||
for r in res:
|
||
if r[0] and r[0] < 0:
|
||
return False
|
||
return True
|
||
|
||
def _state(self, cr, uid, ids, name, args, context=None):
|
||
"""Compute the state lines
|
||
@param self: The object pointer
|
||
@param cr: the current row, from the database cursor,
|
||
@param uid: the current user’s ID for security checks,
|
||
@param ids: List of Membership Line IDs
|
||
@param name: Field Name
|
||
@param context: A standard dictionary for contextual values
|
||
@param return: Dictionary of state Value
|
||
"""
|
||
res = {}
|
||
inv_obj = self.pool.get('account.invoice')
|
||
for line in self.browse(cr, uid, ids, context=context):
|
||
cr.execute('''
|
||
SELECT i.state, i.id FROM
|
||
account_invoice i
|
||
WHERE
|
||
i.id = (
|
||
SELECT l.invoice_id FROM
|
||
account_invoice_line l WHERE
|
||
l.id = (
|
||
SELECT ml.account_invoice_line FROM
|
||
membership_membership_line ml WHERE
|
||
ml.id = %s
|
||
)
|
||
)
|
||
''', (line.id,))
|
||
fetched = cr.fetchone()
|
||
if not fetched:
|
||
res[line.id] = 'canceled'
|
||
continue
|
||
istate = fetched[0]
|
||
state = 'none'
|
||
if (istate == 'draft') | (istate == 'proforma'):
|
||
state = 'waiting'
|
||
elif istate == 'open':
|
||
state = 'invoiced'
|
||
elif istate == 'paid':
|
||
state = 'paid'
|
||
inv = inv_obj.browse(cr, uid, fetched[1], context=context)
|
||
for payment in inv.payment_ids:
|
||
if payment.invoice and payment.invoice.type == 'out_refund':
|
||
state = 'canceled'
|
||
elif istate == 'cancel':
|
||
state = 'canceled'
|
||
res[line.id] = state
|
||
return res
|
||
|
||
|
||
_description = __doc__
|
||
_name = 'membership.membership_line'
|
||
_columns = {
|
||
'partner': fields.many2one('res.partner', 'Partner', ondelete='cascade', select=1),
|
||
'membership_id': fields.many2one('product.product', string="Membership", required=True),
|
||
'date_from': fields.date('From', readonly=True),
|
||
'date_to': fields.date('To', readonly=True),
|
||
'date_cancel': fields.date('Cancel date'),
|
||
'date': fields.date('Join Date', help="Date on which member has joined the membership"),
|
||
'member_price': fields.float('Membership Fee', digits_compute= dp.get_precision('Product Price'), required=True, help='Amount for the membership'),
|
||
'account_invoice_line': fields.many2one('account.invoice.line', 'Account Invoice line', readonly=True),
|
||
'account_invoice_id': fields.related('account_invoice_line', 'invoice_id', type='many2one', relation='account.invoice', string='Invoice', readonly=True),
|
||
'state': fields.function(_state,
|
||
string='Membership Status', type='selection',
|
||
selection=STATE, store = {
|
||
'account.invoice': (_get_membership_lines, ['state'], 10),
|
||
'res.partner': (_get_partners, ['membership_state'], 12),
|
||
}, help="""It indicates the membership status.
|
||
-Non Member: A member who has not applied for any membership.
|
||
-Cancelled Member: A member who has cancelled his membership.
|
||
-Old Member: A member whose membership date has expired.
|
||
-Waiting Member: A member who has applied for the membership and whose invoice is going to be created.
|
||
-Invoiced Member: A member whose invoice has been created.
|
||
-Paid Member: A member who has paid the membership amount."""),
|
||
'company_id': fields.related('account_invoice_line', 'invoice_id', 'company_id', type="many2one", relation="res.company", string="Company", readonly=True, store=True)
|
||
}
|
||
_rec_name = 'partner'
|
||
_order = 'id desc'
|
||
_constraints = [
|
||
(_check_membership_date, 'Error, this membership product is out of date', [])
|
||
]
|
||
|
||
|
||
|
||
class Partner(osv.osv):
|
||
'''Partner'''
|
||
_inherit = 'res.partner'
|
||
|
||
def _get_partner_id(self, cr, uid, ids, context=None):
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
res_obj = self.pool.get('res.partner')
|
||
data_inv = member_line_obj.browse(cr, uid, ids, context=context)
|
||
list_partner = []
|
||
for data in data_inv:
|
||
list_partner.append(data.partner.id)
|
||
ids2 = list_partner
|
||
while ids2:
|
||
ids2 = res_obj.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
|
||
list_partner += ids2
|
||
return list_partner
|
||
|
||
def _get_invoice_partner(self, cr, uid, ids, context=None):
|
||
inv_obj = self.pool.get('account.invoice')
|
||
res_obj = self.pool.get('res.partner')
|
||
data_inv = inv_obj.browse(cr, uid, ids, context=context)
|
||
list_partner = []
|
||
for data in data_inv:
|
||
list_partner.append(data.partner_id.id)
|
||
ids2 = list_partner
|
||
while ids2:
|
||
ids2 = res_obj.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
|
||
list_partner += ids2
|
||
return list_partner
|
||
|
||
def _membership_state(self, cr, uid, ids, name, args, context=None):
|
||
"""This Function return Membership State For Given Partner.
|
||
@param self: The object pointer
|
||
@param cr: the current row, from the database cursor,
|
||
@param uid: the current user’s ID for security checks,
|
||
@param ids: List of Partner IDs
|
||
@param name: Field Name
|
||
@param context: A standard dictionary for contextual values
|
||
@param return: Dictionary of Membership state Value
|
||
"""
|
||
res = {}
|
||
for id in ids:
|
||
res[id] = 'none'
|
||
today = time.strftime('%Y-%m-%d')
|
||
for id in ids:
|
||
partner_data = self.browse(cr, uid, id, context=context)
|
||
if partner_data.membership_cancel and today > partner_data.membership_cancel:
|
||
res[id] = 'canceled'
|
||
continue
|
||
if partner_data.membership_stop and today > partner_data.membership_stop:
|
||
res[id] = 'old'
|
||
continue
|
||
s = 4
|
||
if partner_data.member_lines:
|
||
for mline in partner_data.member_lines:
|
||
if mline.date_to >= today:
|
||
if mline.account_invoice_line and mline.account_invoice_line.invoice_id:
|
||
mstate = mline.account_invoice_line.invoice_id.state
|
||
if mstate == 'paid':
|
||
s = 0
|
||
inv = mline.account_invoice_line.invoice_id
|
||
for payment in inv.payment_ids:
|
||
if payment.invoice.type == 'out_refund':
|
||
s = 2
|
||
break
|
||
elif mstate == 'open' and s!=0:
|
||
s = 1
|
||
elif mstate == 'cancel' and s!=0 and s!=1:
|
||
s = 2
|
||
elif (mstate == 'draft' or mstate == 'proforma') and s!=0 and s!=1:
|
||
s = 3
|
||
if s==4:
|
||
for mline in partner_data.member_lines:
|
||
if mline.date_from < today and mline.date_to < today and mline.date_from <= mline.date_to and (mline.account_invoice_line and mline.account_invoice_line.invoice_id.state) == 'paid':
|
||
s = 5
|
||
else:
|
||
s = 6
|
||
if s==0:
|
||
res[id] = 'paid'
|
||
elif s==1:
|
||
res[id] = 'invoiced'
|
||
elif s==2:
|
||
res[id] = 'canceled'
|
||
elif s==3:
|
||
res[id] = 'waiting'
|
||
elif s==5:
|
||
res[id] = 'old'
|
||
elif s==6:
|
||
res[id] = 'none'
|
||
if partner_data.free_member and s!=0:
|
||
res[id] = 'free'
|
||
if partner_data.associate_member:
|
||
res_state = self._membership_state(cr, uid, [partner_data.associate_member.id], name, args, context=context)
|
||
res[id] = res_state[partner_data.associate_member.id]
|
||
return res
|
||
|
||
def _membership_date(self, cr, uid, ids, name, args, context=None):
|
||
"""Return date of membership"""
|
||
name = name[0]
|
||
res = {}
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
for partner in self.browse(cr, uid, ids, context=context):
|
||
if partner.associate_member:
|
||
partner_id = partner.associate_member.id
|
||
else:
|
||
partner_id = partner.id
|
||
res[partner.id] = {
|
||
'membership_start': False,
|
||
'membership_stop': False,
|
||
'membership_cancel': False
|
||
}
|
||
if name == 'membership_start':
|
||
line_id = member_line_obj.search(cr, uid, [('partner', '=', partner_id),('date_cancel','=',False)],
|
||
limit=1, order='date_from', context=context)
|
||
if line_id:
|
||
res[partner.id]['membership_start'] = member_line_obj.read(cr, uid, [line_id[0]],
|
||
['date_from'], context=context)[0]['date_from']
|
||
|
||
if name == 'membership_stop':
|
||
line_id1 = member_line_obj.search(cr, uid, [('partner', '=', partner_id),('date_cancel','=',False)],
|
||
limit=1, order='date_to desc', context=context)
|
||
if line_id1:
|
||
res[partner.id]['membership_stop'] = member_line_obj.read(cr, uid, [line_id1[0]],
|
||
['date_to'], context=context)[0]['date_to']
|
||
|
||
if name == 'membership_cancel':
|
||
if partner.membership_state == 'canceled':
|
||
line_id2 = member_line_obj.search(cr, uid, [('partner', '=', partner.id)], limit=1, order='date_cancel', context=context)
|
||
if line_id2:
|
||
res[partner.id]['membership_cancel'] = member_line_obj.read(cr, uid, [line_id2[0]], ['date_cancel'], context=context)[0]['date_cancel']
|
||
return res
|
||
|
||
def _get_partners(self, cr, uid, ids, context=None):
|
||
ids2 = ids
|
||
while ids2:
|
||
ids2 = self.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
|
||
ids += ids2
|
||
return ids
|
||
|
||
def __get_membership_state(self, *args, **kwargs):
|
||
return self._membership_state(*args, **kwargs)
|
||
|
||
_columns = {
|
||
'associate_member': fields.many2one('res.partner', 'Associate Member',help="A member with whom you want to associate your membership.It will consider the membership state of the associated member."),
|
||
'member_lines': fields.one2many('membership.membership_line', 'partner', 'Membership'),
|
||
'free_member': fields.boolean('Free Member', help = "Select if you want to give free membership."),
|
||
'membership_amount': fields.float(
|
||
'Membership Amount', digits=(16, 2),
|
||
help = 'The price negotiated by the partner'),
|
||
'membership_state': fields.function(
|
||
__get_membership_state,
|
||
string = 'Current Membership Status', type = 'selection',
|
||
selection = STATE,
|
||
store = {
|
||
'account.invoice': (_get_invoice_partner, ['state'], 10),
|
||
'membership.membership_line': (_get_partner_id, ['state'], 10),
|
||
'res.partner': (_get_partners, ['free_member', 'membership_state', 'associate_member'], 10)
|
||
}, help='It indicates the membership state.\n'
|
||
'-Non Member: A partner who has not applied for any membership.\n'
|
||
'-Cancelled Member: A member who has cancelled his membership.\n'
|
||
'-Old Member: A member whose membership date has expired.\n'
|
||
'-Waiting Member: A member who has applied for the membership and whose invoice is going to be created.\n'
|
||
'-Invoiced Member: A member whose invoice has been created.\n'
|
||
'-Paying member: A member who has paid the membership fee.'),
|
||
'membership_start': fields.function(
|
||
_membership_date, multi = 'membeship_start',
|
||
string = 'Membership Start Date', type = 'date',
|
||
store = {
|
||
'account.invoice': (_get_invoice_partner, ['state'], 10),
|
||
'membership.membership_line': (_get_partner_id, ['state'], 10, ),
|
||
'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
|
||
}, help="Date from which membership becomes active."),
|
||
'membership_stop': fields.function(
|
||
_membership_date,
|
||
string = 'Membership End Date', type='date', multi='membership_stop',
|
||
store = {
|
||
'account.invoice': (_get_invoice_partner, ['state'], 10),
|
||
'membership.membership_line': (_get_partner_id, ['state'], 10),
|
||
'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
|
||
}, help="Date until which membership remains active."),
|
||
'membership_cancel': fields.function(
|
||
_membership_date,
|
||
string = 'Cancel Membership Date', type='date', multi='membership_cancel',
|
||
store = {
|
||
'account.invoice': (_get_invoice_partner, ['state'], 11),
|
||
'membership.membership_line': (_get_partner_id, ['state'], 10),
|
||
'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
|
||
}, help="Date on which membership has been cancelled"),
|
||
}
|
||
_defaults = {
|
||
'free_member': False,
|
||
'membership_cancel': False,
|
||
}
|
||
|
||
def _check_recursion(self, cr, uid, ids, context=None):
|
||
"""Check Recursive for Associated Members.
|
||
"""
|
||
level = 100
|
||
while len(ids):
|
||
cr.execute('SELECT DISTINCT associate_member FROM res_partner WHERE id IN %s', (tuple(ids),))
|
||
ids = filter(None, map(lambda x:x[0], cr.fetchall()))
|
||
if not level:
|
||
return False
|
||
level -= 1
|
||
return True
|
||
|
||
_constraints = [
|
||
(_check_recursion, 'Error ! You cannot create recursive associated members.', ['associate_member'])
|
||
]
|
||
|
||
def create_membership_invoice(self, cr, uid, ids, product_id=None, datas=None, context=None):
|
||
""" Create Customer Invoice of Membership for partners.
|
||
@param datas: datas has dictionary value which consist Id of Membership product and Cost Amount of Membership.
|
||
datas = {'membership_product_id': None, 'amount': None}
|
||
"""
|
||
invoice_obj = self.pool.get('account.invoice')
|
||
invoice_line_obj = self.pool.get('account.invoice.line')
|
||
invoice_tax_obj = self.pool.get('account.invoice.tax')
|
||
product_id = product_id or datas.get('membership_product_id', False)
|
||
amount = datas.get('amount', 0.0)
|
||
invoice_list = []
|
||
if type(ids) in (int, long,):
|
||
ids = [ids]
|
||
for partner in self.browse(cr, uid, ids, context=context):
|
||
account_id = partner.property_account_receivable and partner.property_account_receivable.id or False
|
||
fpos_id = partner.property_account_position and partner.property_account_position.id or False
|
||
addr = self.address_get(cr, uid, [partner.id], ['invoice'])
|
||
if partner.free_member:
|
||
raise UserError(_("Partner is a free Member."))
|
||
if not addr.get('invoice', False):
|
||
raise UserError(_("Partner doesn't have an address to make the invoice."))
|
||
quantity = 1
|
||
line_value = {
|
||
'product_id': product_id,
|
||
}
|
||
|
||
line_dict = invoice_line_obj.product_id_change(cr, uid, {},
|
||
product_id, False, quantity, '', 'out_invoice', partner.id, fpos_id, price_unit=amount, context=context)
|
||
line_value.update(line_dict['value'])
|
||
line_value['price_unit'] = amount
|
||
if line_value.get('invoice_line_tax_id', False):
|
||
tax_tab = [(6, 0, line_value['invoice_line_tax_id'])]
|
||
line_value['invoice_line_tax_id'] = tax_tab
|
||
|
||
invoice_id = invoice_obj.create(cr, uid, {
|
||
'partner_id': partner.id,
|
||
'account_id': account_id,
|
||
'fiscal_position': fpos_id or False
|
||
}, context=context)
|
||
line_value['invoice_id'] = invoice_id
|
||
invoice_line_id = invoice_line_obj.create(cr, uid, line_value, context=context)
|
||
invoice_obj.write(cr, uid, invoice_id, {'invoice_line': [(6, 0, [invoice_line_id])]}, context=context)
|
||
invoice_list.append(invoice_id)
|
||
if line_value['invoice_line_tax_id']:
|
||
tax_value = invoice_tax_obj.compute(cr, uid, invoice_id).values()
|
||
for tax in tax_value:
|
||
invoice_tax_obj.create(cr, uid, tax, context=context)
|
||
#recompute the membership_state of those partners
|
||
self.pool.get('res.partner').write(cr, uid, ids, {})
|
||
return invoice_list
|
||
|
||
|
||
class Product(osv.osv):
|
||
|
||
def fields_view_get(self, cr, user, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
|
||
model_obj = self.pool.get('ir.model.data')
|
||
if context is None:
|
||
context = {}
|
||
|
||
if ('product' in context) and (context['product']=='membership_product'):
|
||
model_data_ids_form = model_obj.search(cr, user, [('model','=','ir.ui.view'), ('name', 'in', ['membership_products_form', 'membership_products_tree'])], context=context)
|
||
resource_id_form = model_obj.read(cr, user, model_data_ids_form, fields=['res_id', 'name'], context=context)
|
||
dict_model = {}
|
||
for i in resource_id_form:
|
||
dict_model[i['name']] = i['res_id']
|
||
if view_type == 'form':
|
||
view_id = dict_model['membership_products_form']
|
||
else:
|
||
view_id = dict_model['membership_products_tree']
|
||
return super(Product,self).fields_view_get(cr, user, view_id, view_type, context, toolbar, submenu)
|
||
|
||
'''Product'''
|
||
_inherit = 'product.template'
|
||
_columns = {
|
||
'membership': fields.boolean('Membership', help='Check if the product is eligible for membership.'),
|
||
'membership_date_from': fields.date('Membership Start Date', help='Date from which membership becomes active.'),
|
||
'membership_date_to': fields.date('Membership End Date', help='Date until which membership remains active.'),
|
||
}
|
||
|
||
_sql_constraints = [('membership_date_greater','check(membership_date_to >= membership_date_from)','Error ! Ending Date cannot be set before Beginning Date.')]
|
||
_defaults = {
|
||
'membership': False,
|
||
}
|
||
|
||
|
||
|
||
class Invoice(osv.osv):
|
||
'''Invoice'''
|
||
_inherit = 'account.invoice'
|
||
|
||
def action_cancel(self, cr, uid, ids, context=None):
|
||
'''Create a 'date_cancel' on the membership_line object'''
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
today = time.strftime('%Y-%m-%d')
|
||
for invoice in self.browse(cr, uid, ids, context=context):
|
||
mlines = member_line_obj.search(cr, uid,
|
||
[('account_invoice_line', 'in',
|
||
[l.id for l in invoice.invoice_line])])
|
||
member_line_obj.write(cr, uid, mlines, {'date_cancel': today})
|
||
return super(Invoice, self).action_cancel(cr, uid, ids, context=context)
|
||
|
||
|
||
class account_invoice_line(osv.osv):
|
||
_inherit='account.invoice.line'
|
||
|
||
def write(self, cr, uid, ids, vals, context=None):
|
||
"""Overrides orm write method
|
||
"""
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
res = super(account_invoice_line, self).write(cr, uid, ids, vals, context=context)
|
||
for line in self.browse(cr, uid, ids, context=context):
|
||
if line.invoice_id.type == 'out_invoice':
|
||
ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', line.id)], context=context)
|
||
if line.product_id and line.product_id.membership and not ml_ids:
|
||
# Product line has changed to a membership product
|
||
date_from = line.product_id.membership_date_from
|
||
date_to = line.product_id.membership_date_to
|
||
if line.invoice_id.date_invoice > date_from and line.invoice_id.date_invoice < date_to:
|
||
date_from = line.invoice_id.date_invoice
|
||
member_line_obj.create(cr, uid, {
|
||
'partner': line.invoice_id.partner_id.id,
|
||
'membership_id': line.product_id.id,
|
||
'member_price': line.price_unit,
|
||
'date': time.strftime('%Y-%m-%d'),
|
||
'date_from': date_from,
|
||
'date_to': date_to,
|
||
'account_invoice_line': line.id,
|
||
}, context=context)
|
||
if line.product_id and not line.product_id.membership and ml_ids:
|
||
# Product line has changed to a non membership product
|
||
member_line_obj.unlink(cr, uid, ml_ids, context=context)
|
||
return res
|
||
|
||
def unlink(self, cr, uid, ids, context=None):
|
||
"""Remove Membership Line Record for Account Invoice Line
|
||
"""
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
for id in ids:
|
||
ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', id)], context=context)
|
||
member_line_obj.unlink(cr, uid, ml_ids, context=context)
|
||
return super(account_invoice_line, self).unlink(cr, uid, ids, context=context)
|
||
|
||
def create(self, cr, uid, vals, context=None):
|
||
"""Overrides orm create method
|
||
"""
|
||
member_line_obj = self.pool.get('membership.membership_line')
|
||
result = super(account_invoice_line, self).create(cr, uid, vals, context=context)
|
||
line = self.browse(cr, uid, result, context=context)
|
||
if line.invoice_id.type == 'out_invoice':
|
||
ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', line.id)], context=context)
|
||
if line.product_id and line.product_id.membership and not ml_ids:
|
||
# Product line is a membership product
|
||
date_from = line.product_id.membership_date_from
|
||
date_to = line.product_id.membership_date_to
|
||
if line.invoice_id.date_invoice > date_from and line.invoice_id.date_invoice < date_to:
|
||
date_from = line.invoice_id.date_invoice
|
||
member_line_obj.create(cr, uid, {
|
||
'partner': line.invoice_id.partner_id and line.invoice_id.partner_id.id or False,
|
||
'membership_id': line.product_id.id,
|
||
'member_price': line.price_unit,
|
||
'date': time.strftime('%Y-%m-%d'),
|
||
'date_from': date_from,
|
||
'date_to': date_to,
|
||
'account_invoice_line': line.id,
|
||
}, context=context)
|
||
return result
|