1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # Remembership module for OpenERP, Membership module enhancement and bug fixes
5 # Copyright (C) 2012-2013 L'Heureux Cyclage (<http://www.heureux-cyclage.org>)
7 # This file is a part of Remembership
9 # Remembership is free software: you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation, either version 3 of the License, or
12 # (at your option) any later version.
14 # Remembership is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 # GNU General Public License for more details.
19 # You should have received a copy of the GNU General Public License
20 # along with this program. If not, see <http://www.gnu.org/licenses/>.
22 ##############################################################################
24 from openerp
.osv
import osv
25 from openerp
.osv
import fields
27 from datetime
import datetime
, date
28 from dateutil
.relativedelta
import relativedelta
29 from openerp
.tools
.translate
import _
32 ('none', 'Non Member'),
33 ('canceled', 'Cancelled Member'),
34 ('old', 'Old Member'),
35 ('waiting', 'Waiting Member'),
36 ('invoiced', 'Invoiced Member'),
37 ('free', 'Free Member'),
38 ('paid', 'Paid Member'),
41 class Partner(osv
.osv
):
42 _inherit
= 'res.partner'
44 # XXX: copied from openerp/addons/membership/membership.py
45 def _membership_date(self
, cr
, uid
, ids
, name
, args
, context
=None):
46 """Return date of membership"""
49 member_line_obj
= self
.pool
.get('membership.membership_line')
50 print ("DEV: [remembership] [_membership_date]")
51 for partner
in self
.browse(cr
, uid
, ids
, context
=context
):
52 # XXX: commented part from openerp/addons/membership/membership.py
53 #if partner.associate_member:
54 # partner_id = partner.associate_member.id
56 # NOTE: hmm, ça devrait être toujours partner.id
57 partner_id
= partner
.id
59 'membership_start': False,
60 'membership_stop': False,
61 'membership_cancel': False
63 if name
== 'membership_start':
64 line_id
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner_id
),('date_cancel','=',False)],
65 limit
=1, order
='date_from', context
=context
)
67 res
[partner
.id]['membership_start'] = member_line_obj
.read(cr
, uid
, line_id
[0],
68 ['date_from'], context
=context
)['date_from']
69 if partner
.associate_member
:
70 print ("DEV: [_membership_date] [associate] date_from: %s" % str(res
[partner
.id]['membership_start']))
72 print ("DEV: [_membership_date] date_from: %s" % str(res
[partner
.id]['membership_start']))
74 if name
== 'membership_stop':
75 line_id1
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner_id
),('date_cancel','=',False)],
76 limit
=1, order
='date_to desc', context
=context
)
78 res
[partner
.id]['membership_stop'] = member_line_obj
.read(cr
, uid
, line_id1
[0],
79 ['date_to'], context
=context
)['date_to']
80 if partner
.associate_member
:
81 print ("DEV: [_membership_date] [associate] date_to: %s" % str(res
[partner
.id]['membership_stop']))
83 print ("DEV: [_membership_date] date_to: %s" % str(res
[partner
.id]['membership_stop']))
85 if name
== 'membership_cancel':
86 if partner
.membership_state
== 'canceled':
87 line_id2
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner
.id)], limit
=1, order
='date_cancel', context
=context
)
89 res
[partner
.id]['membership_cancel'] = member_line_obj
.read(cr
, uid
, line_id2
[0], ['date_cancel'], context
=context
)['date_cancel']
91 # XXX: copied from openerp/addons/membership/membership.py
92 def _get_partner_id(self
, cr
, uid
, ids
, context
=None):
93 member_line_obj
= self
.pool
.get('membership.membership_line')
94 res_obj
= self
.pool
.get('res.partner')
95 data_inv
= member_line_obj
.browse(cr
, uid
, ids
, context
=context
)
98 list_partner
.append(data
.partner
.id)
101 ids2
= res_obj
.search(cr
, uid
, [('associate_member', 'in', ids2
)], context
=context
)
104 # XXX: copied from openerp/addons/membership/membership.py
105 def _get_invoice_partner(self
, cr
, uid
, ids
, context
=None):
106 inv_obj
= self
.pool
.get('account.invoice')
107 res_obj
= self
.pool
.get('res.partner')
108 data_inv
= inv_obj
.browse(cr
, uid
, ids
, context
=context
)
110 for data
in data_inv
:
111 list_partner
.append(data
.partner_id
.id)
114 ids2
= res_obj
.search(cr
, uid
, [('associate_member', 'in', ids2
)], context
=context
)
119 def __get_membership_state(self
, cr
, uid
, ids
, name
, args
, context
=None):
120 res
= super(Partner
, self
).__get
_membership
_state
(cr
, uid
, ids
, name
, args
, context
)
121 today
= time
.strftime('%Y-%m-%d')
123 # special case when the partner is imported from a file,
124 # in this case is has a valid membership line but
125 # no payment information associated to it, consider it valid
127 if res
[id] == 'none':
129 partner_data
= self
.browse(cr
, uid
, id, context
=context
)
131 no_payment_info_at_all
= True
134 for mline
in partner_data
.member_lines
:
135 if mline
.date_to
>= today
:
138 if mline
.account_invoice_line
:
139 no_payment_info_at_all
= False
141 if no_payment_info_at_all
and valid_line
:
147 def _get_partners(self
, *args
, **kwargs
):
148 res
= super(Partner
, self
.pool
.get('res.partner'))._get
_partners
(*args
, **kwargs
)
153 'member_ident': fields
.char('Member identifier', size
=64),
154 'associate_members': fields
.one2many('res.partner', 'associate_member', 'Associate members', help='Members who are associated to this partner.'),
155 # XXX: copied from openerp/addons/membership/membership.py
156 'membership_state': fields
.function(
157 __get_membership_state
,
158 string
= 'Current Membership Status', type = 'selection',
161 'account.invoice': (_get_invoice_partner
, ['state'], 20),
162 'membership.membership_line': (_get_partner_id
, ['state'], 20),
163 'res.partner': (_get_partners
, ['free_member', 'membership_state', 'associate_member'], 20)
164 }, help="""It indicates the membership state.
165 -Non Member: A partner who has not applied for any membership.
166 -Cancelled Member: A member who has cancelled his membership.
167 -Old Member: A member whose membership date has expired.
168 -Waiting Member: A member who has applied for the membership and whose invoice is going to be created.
169 -Invoiced Member: A member whose invoice has been created.
170 -Paying member: A member who has paid the membership fee."""),
171 # XXX: copied from openerp/addons/membership/membership.py
172 'membership_start': fields
.function(
173 _membership_date
, multi
= 'membership_start',
174 string
= 'Membership Start Date', type = 'date',
176 'account.invoice': (_get_invoice_partner
, ['state'], 10),
177 'membership.membership_line': (_get_partner_id
, ['state'], 10, ),
178 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
179 }, help="Date from which membership becomes active."),
180 # XXX: copied from openerp/addons/membership/membership.py
181 'membership_stop': fields
.function(
183 string
= 'Membership End Date', type='date', multi
='membership_stop',
185 'account.invoice': (_get_invoice_partner
, ['state'], 10),
186 'membership.membership_line': (_get_partner_id
, ['state'], 10),
187 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
188 }, help="Date until which membership remains active."),
189 # XXX: copied from openerp/addons/membership/membership.py
190 'membership_cancel': fields
.function(
192 string
= 'Cancel Membership Date', type='date', multi
='membership_cancel',
194 'account.invoice': (_get_invoice_partner
, ['state'], 11),
195 'membership.membership_line': (_get_partner_id
, ['state'], 10),
196 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
197 }, help="Date on which membership has been cancelled"),
201 ('member_ident_uniq', 'unique(member_ident, company_id)', 'The member identifier must be unique !'),
204 def create_membership_invoice(self
, cr
, uid
, ids
, product_id
=None, datas
=None, context
=None):
205 # NOTE: use context to supply date_from to account_invoice_line.create
206 context
= context
and context
.copy() or {}
207 context
.update({'date_from': datas
.get('date_from', None)});
208 # XXX: copied from openerp/addons/membership/membership.py to fix account_invoice_line creation
209 print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s" % str(ids
))
210 invoice_obj
= self
.pool
.get('account.invoice')
211 invoice_line_obj
= self
.pool
.get('account.invoice.line')
212 invoice_tax_obj
= self
.pool
.get('account.invoice.tax')
213 product_id
= product_id
or datas
.get('membership_product_id', False)
214 amount
= datas
.get('amount', 0.0)
216 if type(ids
) in (int, long,):
218 for partner
in self
.browse(cr
, uid
, ids
, context
=context
):
219 print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s partner=%s" % (str(ids
),str(partner
.id)))
220 account_id
= partner
.property_account_receivable
and partner
.property_account_receivable
.id or False
221 fpos_id
= partner
.property_account_position
and partner
.property_account_position
.id or False
222 addr
= self
.address_get(cr
, uid
, [partner
.id], ['invoice'])
223 if partner
.free_member
:
224 raise osv
.except_osv(_('Error!'),
225 _("Partner is a free Member."))
226 if not addr
.get('invoice', False):
227 raise osv
.except_osv(_('Error!'),
228 _("Partner doesn't have an address to make the invoice."))
231 'product_id': product_id
,
234 print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s partner=%s [invoice_line_obj] [product_id_change]" % (str(ids
),str(partner
.id)))
235 line_dict
= invoice_line_obj
.product_id_change(cr
, uid
, {},
236 product_id
, False, quantity
, '', 'out_invoice', partner
.id, fpos_id
, price_unit
=amount
, context
=context
)
237 line_value
.update(line_dict
['value'])
238 line_value
['price_unit'] = amount
239 if line_value
.get('invoice_line_tax_id', False):
240 tax_tab
= [(6, 0, line_value
['invoice_line_tax_id'])]
241 line_value
['invoice_line_tax_id'] = tax_tab
243 print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s partner=%s [invoice_obj] [create] account=%s" % (str(ids
),str(partner
.id),str(account_id
)))
244 # NOTE: let account_invoice_obj.create() call account_invoice_line.create()
245 # to have a correct membership_state: becauce now fields.function 'store' watchdogs
246 # are triggered after account_invoice_line.create().
247 invoice_id
= invoice_obj
.create(cr
, uid
, {
248 'partner_id': partner
.id,
249 'account_id': account_id
,
250 'fiscal_position': fpos_id
or False,
251 'invoice_line':[(0, 0, line_value
)]
253 #line_value['invoice_id'] = invoice_id
254 #print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s partner=%s [invoice_line_obj] [create]" % (str(ids),str(partner.id)))
255 #invoice_line_id = invoice_line_obj.create(cr, uid, line_value, context=context)
256 #invoice_obj.write(cr, uid, invoice_id, {'invoice_line': line_value}, context=context)
257 print ("DEV: [remembership] [partner] [create_membership_invoice] : invoice_list.append(invoice_id) : pre")
258 invoice_list
.append(invoice_id
)
259 print ("DEV: [remembership] [partner] [create_membership_invoice] : invoice_list.append(invoice_id) : post")
260 if line_value
['invoice_line_tax_id']:
261 tax_value
= invoice_tax_obj
.compute(cr
, uid
, invoice_id
).values()
262 for tax
in tax_value
:
263 invoice_tax_obj
.create(cr
, uid
, tax
, context
=context
)
264 # NOTE: this wasn't working, and is not needed now.
265 #recompute the membership_state of those partners
266 #print ("DEV: [remembership] [partner] [create_membership_invoice] ids=%s [recompute states]" % str(ids))
267 #self.pool.get('res.partner').write(cr, uid, ids, {}, context=context)
268 print ("DEV: [remembership] [partner] [create_membership_invoice] return=%s" % str(invoice_list
))
271 def update_membership_status(self
, cr
, uid
):
272 """Action for updating old members status with scheduled cron job"""
273 partner_obj
= self
.pool
.get('res.partner')
275 ids
= partner_obj
.search(cr
, uid
, [('membership_state', '=', 'paid'), ('membership_stop', '<', today
)], context
=None)
276 self
.write(cr
, uid
, ids
, {'membership_state': 'old'}, context
=None)
281 class Product(osv
.osv
):
282 _inherit
= 'product.product'
285 'membership_grouped': fields
.boolean('Grouped membership product', help='Check if it\'s a grouped membership product.'),
286 'membership_date2date': fields
.boolean('Date to date membership product', required
=False, help='Check if it\'s a date to date membership product.'),
292 class account_invoice_line(osv
.osv
):
293 _inherit
= 'account.invoice.line'
295 def write(self
, cr
, uid
, ids
, vals
, context
=None):
296 member_line_obj
= self
.pool
.get('membership.membership_line')
297 res
= super(account_invoice_line
, self
).write(cr
, uid
, ids
, vals
, context
=context
)
298 for line
in self
.browse(cr
, uid
, ids
, context
=context
):
299 if line
.invoice_id
.type == 'out_invoice':
300 ml_ids
= member_line_obj
.search(cr
, uid
, [('account_invoice_line', '=', line
.id)], context
=context
)
301 if line
.product_id
and line
.product_id
.membership
:
302 member_line_id
= member_line_obj
.search(cr
, uid
303 , [('partner' , '=', line
.invoice_id
.partner_id
.id)
304 ,('account_invoice_line', '=', line
.id)]
307 for member_line
in member_line_obj
.browse(cr
, uid
, member_line_id
, context
=context
):
308 # NOTE: get member_line created in membership.account_invoice_line.create
309 date_from
= member_line
.date_from
310 date_to
= member_line
.date_to
311 if line
.product_id
.membership_date2date
:
312 date_from
= ('date_from' in context
313 and context
.get('date_from')
314 and datetime
.strptime(context('date_from'), "%Y-%m-%d")
316 date_to
= date_from
+ relativedelta(months
=+12, days
=-1) # TODO: parameterize this delta?
317 date_from
= date_from
.strftime("%Y-%m-%d")
318 date_to
= date_to
.strftime("%Y-%m-%d")
319 print ("DEV: [remembership] [account_invoice_line] [write] date_from: %s" % str(date_from
))
320 print ("DEV: [remembership] [account_invoice_line] [write] date_to : %s" % str(date_to
))
321 member_line_obj
.write(cr
, uid
, member_line
.id
322 , {'date_from': date_from
326 if line
.product_id
.membership_grouped
:
327 if line
.invoice_id
.partner_id
.associate_members
:
328 associate_member_line_ids
= member_line_obj
.search(cr
, uid
329 , [ ('account_invoice_line', '=', line
.id)
330 , ('partner', '!=', line
.invoice_id
.partner_id
.id)
333 for associate_member_line
in member_line_obj
.browse(cr
, uid
, associate_member_line_ids
, context
=context
):
334 print ("DEV: [remembership] [account_invoice_line] [write] [associate] date_from: %s" % str(date_from
))
335 print ("DEV: [remembership] [account_invoice_line] [write] [associate] date_to : %s" % str(date_to
))
336 member_line_obj
.write(cr
, uid
, associate_member_line
.id
337 , {'date_from': date_from
342 print("DEV: mettre une contrainte pour l'objet\
343 membership.membership_line interdisant les\
344 adhésions groupées reliées à des partenaires\
345 sans membres associés")
347 associate_member_line_ids
= member_line_obj
.search(cr
, uid
348 , [ ('account_invoice_line', '=', line
.id)
349 , ('partner', '!=', line
.invoice_id
.partner_id
.id)
352 member_line_obj
.unlink(cr
, uid
, associate_member_line_ids
, context
=context
)
353 #Define member ident if it's necessary
354 partners
= [line
.invoice_id
.partner_id
]
355 if line
.invoice_id
.partner_id
.associate_members
:
356 partners
.extend(line
.invoice_id
.partner_id
.associate_members
)
358 if not i
.member_ident
:
359 mbr_id
= self
.pool
.get('ir.sequence').get(cr
, uid
, 'member_ident')
360 self
.pool
.get('res.partner').write(cr
, uid
, i
.id, {'member_ident': mbr_id
})
361 if line
.product_id
and not line
.product_id
.membership
and ml_ids
:
362 # Product line has changed to a non membership product
363 member_line_obj
.unlink(cr
, uid
, ml_ids
, context
=context
)
364 print ("DEV: [remembership] [account_invoice_line] [write] : return=%s" % str(res
))
367 def create(self
, cr
, uid
, vals
, context
=None):
368 member_line_obj
= self
.pool
.get('membership.membership_line')
369 res
= super(account_invoice_line
, self
).create(cr
, uid
, vals
, context
=context
)
370 print ("DEV: [remembership] [account_invoice_line] [create] : begin res=%s" % str(res
))
371 line
= self
.browse(cr
, uid
, res
, context
=context
)
373 if line
.invoice_id
.type == 'out_invoice' and line
.product_id
and line
.product_id
.membership
:
374 member_line_id
= member_line_obj
.search(cr
, uid
375 , [('partner' , '=', line
.invoice_id
.partner_id
.id)
376 ,('account_invoice_line', '=', line
.id)]
379 for member_line
in member_line_obj
.browse(cr
, uid
, member_line_id
, context
=context
):
380 # NOTE: get member_line created in membership.account_invoice_line.create
381 date_from
= member_line
.date_from
382 date_to
= member_line
.date_to
383 if line
.product_id
.membership_date2date
:
384 date_from
= ('date_from' in context
385 and context
.get('date_from')
386 and datetime
.strptime(context
.get('date_from'), "%Y-%m-%d")
388 print("#DEBUG: Date de début - %s" % str(date_from
))
389 date_to
= date_from
+ relativedelta(months
=+12, days
=-1) # TODO: parameterize this delta?
390 date_from
= date_from
.strftime("%Y-%m-%d")
391 date_to
= date_to
.strftime("%Y-%m-%d")
392 print ("DEV: [remembership] [account_invoice_line] [create] date_from: %s" % str(date_from
))
393 print ("DEV: [remembership] [account_invoice_line] [create] date_to : %s" % str(date_to
))
394 member_line_obj
.write(cr
, uid
, member_line
.id
395 , {'date_from': date_from
399 partners
= [line
.invoice_id
.partner_id
]
400 if line
.product_id
.membership_grouped
and line
.invoice_id
.partner_id
.associate_members
:
401 partners
.extend(line
.invoice_id
.partner_id
.associate_members
)
402 #Adding membership lines just for associate partners
403 for associate_member
in line
.invoice_id
.partner_id
.associate_members
:
404 print ("DEV: [remembership] [account_invoice_line] [create] [associate] date_from: %s" % str(date_from
))
405 print ("DEV: [remembership] [account_invoice_line] [create] [associate] date_to : %s" % str(date_to
))
406 member_line_obj
.create(cr
, uid
, {
407 'partner': associate_member
.id,
408 'membership_id': line
.product_id
.id,
409 'member_price': line
.price_unit
,
410 'date': time
.strftime('%Y-%m-%d'),
411 'date_from': date_from
,
413 'account_invoice_line': line
.id,
415 elif line
.product_id
.membership_grouped
and not line
.invoice_id
.partner_id
.associate_members
:
416 raise osv
.except_osv(_('Error!!!'), _('You try to invoice grouped membership product to a partner who hasn\'t associated partners.'))
417 #Define member ident if it's necessary
419 if not i
.member_ident
:
420 mbr_id
= self
.pool
.get('ir.sequence').get(cr
, uid
, 'member_ident')
421 self
.pool
.get('res.partner').write(cr
, uid
, i
.id, {'member_ident': mbr_id
})
422 print ("DEV: [remembership] [account_invoice_line] [create] : return=%s" % str(res
))
425 account_invoice_line()
426 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: