Launchpad automatic translations update.
[odoo/odoo.git] / addons / membership / membership.py
1 # -*- coding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
6 #
7 #    This program is free software: you can redistribute it and/or modify
8 #    it under the terms of the GNU Affero General Public License as
9 #    published by the Free Software Foundation, either version 3 of the
10 #    License, or (at your option) any later version.
11 #
12 #    This program is distributed in the hope that it will be useful,
13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 #    GNU Affero General Public License for more details.
16 #
17 #    You should have received a copy of the GNU Affero General Public License
18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 #
20 ##############################################################################
21
22 import time
23
24 from osv import fields, osv
25 import decimal_precision as dp
26 from tools.translate import _
27
28 STATE = [
29     ('none', 'Non Member'),
30     ('canceled', 'Cancelled Member'),
31     ('old', 'Old Member'),
32     ('waiting', 'Waiting Member'),
33     ('invoiced', 'Invoiced Member'),
34     ('free', 'Free Member'),
35     ('paid', 'Paid Member'),
36 ]
37
38 STATE_PRIOR = {
39     'none': 0,
40     'canceled': 1,
41     'old': 2,
42     'waiting': 3,
43     'invoiced': 4,
44     'free': 6,
45     'paid': 7
46 }
47
48 class membership_line(osv.osv):
49     '''Member line'''
50
51     def _get_partners(self, cr, uid, ids, context=None):
52         list_membership_line = []
53         member_line_obj = self.pool.get('membership.membership_line')
54         for partner in self.pool.get('res.partner').browse(cr, uid, ids, context=context):
55             if partner.member_lines:
56                 list_membership_line += member_line_obj.search(cr, uid, [('id', 'in', [ l.id for l in partner.member_lines])], context=context)
57         return list_membership_line
58
59     def _get_membership_lines(self, cr, uid, ids, context=None):
60         list_membership_line = []
61         member_line_obj = self.pool.get('membership.membership_line')
62         for invoice in self.pool.get('account.invoice').browse(cr, uid, ids, context=context):
63             if invoice.invoice_line:
64                 list_membership_line += member_line_obj.search(cr, uid, [('account_invoice_line', 'in', [ l.id for l in invoice.invoice_line])], context=context)
65         return list_membership_line
66
67     def _check_membership_date(self, cr, uid, ids, context=None):
68         """Check if membership product is not in the past
69         @param self: The object pointer
70         @param cr: the current row, from the database cursor,
71         @param uid: the current user’s ID for security checks,
72         @param ids: List of Membership Line IDs
73         @param context: A standard dictionary for contextual values
74         """
75
76         cr.execute('''
77          SELECT MIN(ml.date_to - ai.date_invoice)
78              FROM membership_membership_line ml
79              JOIN account_invoice_line ail ON (
80                 ml.account_invoice_line = ail.id
81                 )
82             JOIN account_invoice ai ON (
83             ai.id = ail.invoice_id)
84             WHERE ml.id IN %s''', (tuple(ids),))
85         res = cr.fetchall()
86         for r in res:
87             if r[0] and r[0] < 0:
88                 return False
89         return True
90
91     def _state(self, cr, uid, ids, name, args, context=None):
92         """Compute the state lines
93         @param self: The object pointer
94         @param cr: the current row, from the database cursor,
95         @param uid: the current user’s ID for security checks,
96         @param ids: List of Membership Line IDs
97         @param name: Field Name
98         @param context: A standard dictionary for contextual values
99         @param return: Dictionary of state Value
100         """
101         res = {}
102         inv_obj = self.pool.get('account.invoice')
103         for line in self.browse(cr, uid, ids, context=context):
104             cr.execute('''
105             SELECT i.state, i.id FROM
106             account_invoice i
107             WHERE
108             i.id = (
109                 SELECT l.invoice_id FROM
110                 account_invoice_line l WHERE
111                 l.id = (
112                     SELECT  ml.account_invoice_line FROM
113                     membership_membership_line ml WHERE
114                     ml.id = %s
115                     )
116                 )
117             ''', (line.id,))
118             fetched = cr.fetchone()
119             if not fetched:
120                 res[line.id] = 'canceled'
121                 continue
122             istate = fetched[0]
123             state = 'none'
124             if (istate == 'draft') | (istate == 'proforma'):
125                 state = 'waiting'
126             elif istate == 'open':
127                 state = 'invoiced'
128             elif istate == 'paid':
129                 state = 'paid'
130                 inv = inv_obj.browse(cr, uid, fetched[1], context=context)
131                 for payment in inv.payment_ids:
132                     if payment.invoice and payment.invoice.type == 'out_refund':
133                         state = 'canceled'
134             elif istate == 'cancel':
135                 state = 'canceled'
136             res[line.id] = state
137         return res
138
139
140     _description = __doc__
141     _name = 'membership.membership_line'
142     _columns = {
143         'partner': fields.many2one('res.partner', 'Partner', ondelete='cascade', select=1),
144         'membership_id': fields.many2one('product.product', string="Membership", required=True),
145         'date_from': fields.date('From', readonly=True),
146         'date_to': fields.date('To', readonly=True),
147         'date_cancel': fields.date('Cancel date'),
148         'date': fields.date('Join Date', help="Date on which member has joined the membership"),
149         'member_price': fields.float('Membership Fee', digits_compute= dp.get_precision('Product Price'), required=True, help='Amount for the membership'),
150         'account_invoice_line': fields.many2one('account.invoice.line', 'Account Invoice line', readonly=True),
151         'account_invoice_id': fields.related('account_invoice_line', 'invoice_id', type='many2one', relation='account.invoice', string='Invoice', readonly=True),
152         'state': fields.function(_state,
153                         string='Membership Status', type='selection',
154                         selection=STATE, store = {
155                         'account.invoice': (_get_membership_lines, ['state'], 10),
156                         'res.partner': (_get_partners, ['membership_state'], 12),
157                         }, help="""It indicates the membership status.
158                         -Non Member: A member who has not applied for any membership.
159                         -Cancelled Member: A member who has cancelled his membership.
160                         -Old Member: A member whose membership date has expired.
161                         -Waiting Member: A member who has applied for the membership and whose invoice is going to be created.
162                         -Invoiced Member: A member whose invoice has been created.
163                         -Paid Member: A member who has paid the membership amount."""),
164         'company_id': fields.related('account_invoice_line', 'invoice_id', 'company_id', type="many2one", relation="res.company", string="Company", readonly=True, store=True)
165     }
166     _rec_name = 'partner'
167     _order = 'id desc'
168     _constraints = [
169         (_check_membership_date, 'Error, this membership product is out of date', [])
170     ]
171
172 membership_line()
173
174
175 class Partner(osv.osv):
176     '''Partner'''
177     _inherit = 'res.partner'
178
179     def _get_partner_id(self, cr, uid, ids, context=None):
180         member_line_obj = self.pool.get('membership.membership_line')
181         res_obj =  self.pool.get('res.partner')
182         data_inv = member_line_obj.browse(cr, uid, ids, context=context)
183         list_partner = []
184         for data in data_inv:
185             list_partner.append(data.partner.id)
186         ids2 = list_partner
187         while ids2:
188             ids2 = res_obj.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
189             list_partner += ids2
190         return list_partner
191
192     def _get_invoice_partner(self, cr, uid, ids, context=None):
193         inv_obj = self.pool.get('account.invoice')
194         res_obj = self.pool.get('res.partner')
195         data_inv = inv_obj.browse(cr, uid, ids, context=context)
196         list_partner = []
197         for data in data_inv:
198             list_partner.append(data.partner_id.id)
199         ids2 = list_partner
200         while ids2:
201             ids2 = res_obj.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
202             list_partner += ids2
203         return list_partner
204
205     def _membership_state(self, cr, uid, ids, name, args, context=None):
206         """This Function return Membership State For Given Partner.
207         @param self: The object pointer
208         @param cr: the current row, from the database cursor,
209         @param uid: the current user’s ID for security checks,
210         @param ids: List of Partner IDs
211         @param name: Field Name
212         @param context: A standard dictionary for contextual values
213         @param return: Dictionary of Membership state Value
214         """
215         res = {}
216         for id in ids:
217             res[id] = 'none'
218         today = time.strftime('%Y-%m-%d')
219         for id in ids:
220             partner_data = self.browse(cr, uid, id, context=context)
221             if partner_data.membership_cancel and today > partner_data.membership_cancel:
222                 res[id] = 'canceled'
223                 continue
224             if partner_data.membership_stop and today > partner_data.membership_stop:
225                 res[id] = 'old'
226                 continue
227             s = 4
228             if partner_data.member_lines:
229                 for mline in partner_data.member_lines:
230                     if mline.date_to >= today:
231                         if mline.account_invoice_line and mline.account_invoice_line.invoice_id:
232                             mstate = mline.account_invoice_line.invoice_id.state
233                             if mstate == 'paid':
234                                 s = 0
235                                 inv = mline.account_invoice_line.invoice_id
236                                 for payment in inv.payment_ids:
237                                     if payment.invoice.type == 'out_refund':
238                                         s = 2
239                                 break
240                             elif mstate == 'open' and s!=0:
241                                 s = 1
242                             elif mstate == 'cancel' and s!=0 and s!=1:
243                                 s = 2
244                             elif  (mstate == 'draft' or mstate == 'proforma') and s!=0 and s!=1:
245                                 s = 3
246                 if s==4:
247                     for mline in partner_data.member_lines:
248                         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':
249                             s = 5
250                         else:
251                             s = 6
252                 if s==0:
253                     res[id] = 'paid'
254                 elif s==1:
255                     res[id] = 'invoiced'
256                 elif s==2:
257                     res[id] = 'canceled'
258                 elif s==3:
259                     res[id] = 'waiting'
260                 elif s==5:
261                     res[id] = 'old'
262                 elif s==6:
263                     res[id] = 'none'
264             if partner_data.free_member and s!=0:
265                 res[id] = 'free'
266             if partner_data.associate_member:
267                 res_state = self._membership_state(cr, uid, [partner_data.associate_member.id], name, args, context=context)
268                 res[id] = res_state[partner_data.associate_member.id]
269         return res
270
271     def _membership_date(self, cr, uid, ids, name, args, context=None):
272         """Return  date of membership"""
273         name = name[0]
274         res = {}
275         member_line_obj = self.pool.get('membership.membership_line')
276         for partner in self.browse(cr, uid, ids, context=context):
277             if partner.associate_member:
278                  partner_id = partner.associate_member.id
279             else:
280                  partner_id = partner.id
281             res[partner.id] = {
282                  'membership_start': False,
283                  'membership_stop': False,
284                  'membership_cancel': False
285             }
286             if name == 'membership_start':
287                 line_id = member_line_obj.search(cr, uid, [('partner', '=', partner_id),('date_cancel','=',False)],
288                             limit=1, order='date_from', context=context)
289                 if line_id:
290                         res[partner.id]['membership_start'] = member_line_obj.read(cr, uid, line_id[0],
291                                 ['date_from'], context=context)['date_from']
292
293             if name == 'membership_stop':
294                 line_id1 = member_line_obj.search(cr, uid, [('partner', '=', partner_id),('date_cancel','=',False)],
295                             limit=1, order='date_to desc', context=context)
296                 if line_id1:
297                       res[partner.id]['membership_stop'] = member_line_obj.read(cr, uid, line_id1[0],
298                                 ['date_to'], context=context)['date_to']
299
300             if name == 'membership_cancel':
301                 if partner.membership_state == 'canceled':
302                     line_id2 = member_line_obj.search(cr, uid, [('partner', '=', partner.id)], limit=1, order='date_cancel', context=context)
303                     if line_id2:
304                         res[partner.id]['membership_cancel'] = member_line_obj.read(cr, uid, line_id2[0], ['date_cancel'], context=context)['date_cancel']
305         return res
306
307     def _get_partners(self, cr, uid, ids, context=None):
308         ids2 = ids
309         while ids2:
310             ids2 = self.search(cr, uid, [('associate_member', 'in', ids2)], context=context)
311             ids += ids2
312         return ids
313
314     def __get_membership_state(self, *args, **kwargs):
315         return self._membership_state(*args, **kwargs)
316
317     _columns = {
318         '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."),
319         'member_lines': fields.one2many('membership.membership_line', 'partner', 'Membership'),
320         'free_member': fields.boolean('Free Member', help = "Select if you want to give free membership."),
321         'membership_amount': fields.float(
322                     'Membership Amount', digits=(16, 2),
323                     help = 'The price negotiated by the partner'),
324         'membership_state': fields.function(
325                     __get_membership_state,
326                     string = 'Current Membership Status', type = 'selection',
327                     selection = STATE,
328                     store = {
329                         'account.invoice': (_get_invoice_partner, ['state'], 10),
330                         'membership.membership_line': (_get_partner_id, ['state'], 10),
331                         'res.partner': (_get_partners, ['free_member', 'membership_state', 'associate_member'], 10)
332                     }, help="""It indicates the membership state.
333                     -Non Member: A partner who has not applied for any membership.
334                     -Cancelled Member: A member who has cancelled his membership.
335                     -Old Member: A member whose membership date has expired.
336                     -Waiting Member: A member who has applied for the membership and whose invoice is going to be created.
337                     -Invoiced Member: A member whose invoice has been created.
338                     -Paying member: A member who has paid the membership fee."""),
339         'membership_start': fields.function(
340                     _membership_date, multi = 'membeship_start',
341                     string = 'Membership Start Date', type = 'date',
342                     store = {
343                         'account.invoice': (_get_invoice_partner, ['state'], 10),
344                         'membership.membership_line': (_get_partner_id, ['state'], 10, ),
345                         'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
346                     }, help="Date from which membership becomes active."),
347         'membership_stop': fields.function(
348                     _membership_date,
349                     string = 'Membership End Date', type='date', multi='membership_stop',
350                     store = {
351                         'account.invoice': (_get_invoice_partner, ['state'], 10),
352                         'membership.membership_line': (_get_partner_id, ['state'], 10),
353                         'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
354                     }, help="Date until which membership remains active."),
355         'membership_cancel': fields.function(
356                     _membership_date,
357                     string = 'Cancel Membership Date', type='date', multi='membership_cancel',
358                     store = {
359                         'account.invoice': (_get_invoice_partner, ['state'], 11),
360                         'membership.membership_line': (_get_partner_id, ['state'], 10),
361                         'res.partner': (lambda self, cr, uid, ids, c={}: ids, ['free_member'], 10)
362                     }, help="Date on which membership has been cancelled"),
363     }
364     _defaults = {
365         'free_member': False,
366         'membership_cancel': False,
367     }
368
369     def _check_recursion(self, cr, uid, ids, context=None):
370         """Check  Recursive  for Associated Members.
371         """
372         level = 100
373         while len(ids):
374             cr.execute('SELECT DISTINCT associate_member FROM res_partner WHERE id IN %s', (tuple(ids),))
375             ids = filter(None, map(lambda x:x[0], cr.fetchall()))
376             if not level:
377                 return False
378             level -= 1
379         return True
380
381     _constraints = [
382         (_check_recursion, 'Error ! You cannot create recursive associated members.', ['associate_member'])
383     ]
384
385     def copy(self, cr, uid, id, default=None, context=None):
386         if default is None:
387             default = {}
388         default = default.copy()
389         default['member_lines'] = []
390         return super(Partner, self).copy(cr, uid, id, default, context=context)
391
392     def create_membership_invoice(self, cr, uid, ids, product_id=None, datas=None, context=None):
393         """ Create Customer Invoice of Membership for partners.
394         @param datas: datas has dictionary value which consist Id of Membership product and Cost Amount of Membership.
395                       datas = {'membership_product_id': None, 'amount': None}
396         """
397         invoice_obj = self.pool.get('account.invoice')
398         invoice_line_obj = self.pool.get('account.invoice.line')
399         invoice_tax_obj = self.pool.get('account.invoice.tax')
400         product_id = product_id or datas.get('membership_product_id', False)
401         amount = datas.get('amount', 0.0)
402         invoice_list = []
403         if type(ids) in (int, long,):
404             ids = [ids]
405         for partner in self.browse(cr, uid, ids, context=context):
406             account_id = partner.property_account_receivable and partner.property_account_receivable.id or False
407             fpos_id = partner.property_account_position and partner.property_account_position.id or False
408             addr = self.address_get(cr, uid, [partner.id], ['invoice'])
409             if partner.free_member:
410                 raise osv.except_osv(_('Error!'),
411                         _("Partner is a free Member."))
412             if not addr.get('invoice', False):
413                 raise osv.except_osv(_('Error!'),
414                         _("Partner doesn't have an address to make the invoice."))
415             quantity = 1
416             line_value =  {
417                 'product_id': product_id,
418             }
419
420             line_dict = invoice_line_obj.product_id_change(cr, uid, {},
421                             product_id, False, quantity, '', 'out_invoice', partner.id, fpos_id, price_unit=amount, context=context)
422             line_value.update(line_dict['value'])
423             line_value['price_unit'] = amount
424             if line_value.get('invoice_line_tax_id', False):
425                 tax_tab = [(6, 0, line_value['invoice_line_tax_id'])]
426                 line_value['invoice_line_tax_id'] = tax_tab
427
428             invoice_id = invoice_obj.create(cr, uid, {
429                 'partner_id': partner.id,
430                 'account_id': account_id,
431                 'fiscal_position': fpos_id or False
432                 }, context=context)
433             line_value['invoice_id'] = invoice_id
434             invoice_line_id = invoice_line_obj.create(cr, uid, line_value, context=context)
435             invoice_obj.write(cr, uid, invoice_id, {'invoice_line': [(6, 0, [invoice_line_id])]}, context=context)
436             invoice_list.append(invoice_id)
437             if line_value['invoice_line_tax_id']:
438                 tax_value = invoice_tax_obj.compute(cr, uid, invoice_id).values()
439                 for tax in tax_value:
440                        invoice_tax_obj.create(cr, uid, tax, context=context)
441         #recompute the membership_state of those partners
442         self.pool.get('res.partner').write(cr, uid, ids, {})
443         return invoice_list
444
445 Partner()
446
447 class Product(osv.osv):
448
449     def fields_view_get(self, cr, user, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
450         model_obj = self.pool.get('ir.model.data')
451         if context is None:
452             context = {}
453
454         if ('product' in context) and (context['product']=='membership_product'):
455             model_data_ids_form = model_obj.search(cr, user, [('model','=','ir.ui.view'), ('name', 'in', ['membership_products_form', 'membership_products_tree'])], context=context)
456             resource_id_form = model_obj.read(cr, user, model_data_ids_form, fields=['res_id', 'name'], context=context)
457             dict_model = {}
458             for i in resource_id_form:
459                 dict_model[i['name']] = i['res_id']
460             if view_type == 'form':
461                 view_id = dict_model['membership_products_form']
462             else:
463                 view_id = dict_model['membership_products_tree']
464         return super(Product,self).fields_view_get(cr, user, view_id, view_type, context, toolbar, submenu)
465
466     '''Product'''
467     _inherit = 'product.product'
468     _columns = {
469         'membership': fields.boolean('Membership', help='Check if the product is eligible for membership.'),
470         'membership_date_from': fields.date('Date from', help='Date from which membership becomes active.'),
471         'membership_date_to': fields.date('Date to', help='Date until which membership remains active.'),
472     }
473
474     _sql_constraints = [('membership_date_greater','check(membership_date_to >= membership_date_from)','Error ! Ending Date cannot be set before Beginning Date.')]
475     _defaults = {
476         'membership': False,
477     }
478
479 Product()
480
481
482 class Invoice(osv.osv):
483     '''Invoice'''
484     _inherit = 'account.invoice'
485
486     def action_cancel(self, cr, uid, ids, *args):
487         '''Create a 'date_cancel' on the membership_line object'''
488         member_line_obj = self.pool.get('membership.membership_line')
489         today = time.strftime('%Y-%m-%d')
490         for invoice in self.browse(cr, uid, ids):
491             mlines = member_line_obj.search(cr, uid,
492                     [('account_invoice_line', 'in',
493                         [l.id for l in invoice.invoice_line])])
494             member_line_obj.write(cr, uid, mlines, {'date_cancel': today})
495         return super(Invoice, self).action_cancel(cr, uid, ids)
496
497 Invoice()
498
499 class account_invoice_line(osv.osv):
500     _inherit='account.invoice.line'
501
502     def write(self, cr, uid, ids, vals, context=None):
503         """Overrides orm write method
504         """
505         member_line_obj = self.pool.get('membership.membership_line')
506         res = super(account_invoice_line, self).write(cr, uid, ids, vals, context=context)
507         for line in self.browse(cr, uid, ids, context=context):
508             if line.invoice_id.type == 'out_invoice':
509                 ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', line.id)], context=context)
510                 if line.product_id and line.product_id.membership and not ml_ids:
511                     # Product line has changed to a membership product
512                     date_from = line.product_id.membership_date_from
513                     date_to = line.product_id.membership_date_to
514                     if line.invoice_id.date_invoice > date_from and line.invoice_id.date_invoice < date_to:
515                         date_from = line.invoice_id.date_invoice
516                     member_line_obj.create(cr, uid, {
517                                     'partner': line.invoice_id.partner_id.id,
518                                     'membership_id': line.product_id.id,
519                                     'member_price': line.price_unit,
520                                     'date': time.strftime('%Y-%m-%d'),
521                                     'date_from': date_from,
522                                     'date_to': date_to,
523                                     'account_invoice_line': line.id,
524                                     }, context=context)
525                 if line.product_id and not line.product_id.membership and ml_ids:
526                     # Product line has changed to a non membership product
527                     member_line_obj.unlink(cr, uid, ml_ids, context=context)
528         return res
529
530     def unlink(self, cr, uid, ids, context=None):
531         """Remove Membership Line Record for Account Invoice Line
532         """
533         member_line_obj = self.pool.get('membership.membership_line')
534         for id in ids:
535             ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', id)], context=context)
536             member_line_obj.unlink(cr, uid, ml_ids, context=context)
537         return super(account_invoice_line, self).unlink(cr, uid, ids, context=context)
538
539     def create(self, cr, uid, vals, context=None):
540         """Overrides orm create method
541         """
542         member_line_obj = self.pool.get('membership.membership_line')
543         result = super(account_invoice_line, self).create(cr, uid, vals, context=context)
544         line = self.browse(cr, uid, result, context=context)
545         if line.invoice_id.type == 'out_invoice':
546             ml_ids = member_line_obj.search(cr, uid, [('account_invoice_line', '=', line.id)], context=context)
547             if line.product_id and line.product_id.membership and not ml_ids:
548                 # Product line is a membership product
549                 date_from = line.product_id.membership_date_from
550                 date_to = line.product_id.membership_date_to
551                 if line.invoice_id.date_invoice > date_from and line.invoice_id.date_invoice < date_to:
552                     date_from = line.invoice_id.date_invoice
553                 member_line_obj.create(cr, uid, {
554                             'partner': line.invoice_id.partner_id and line.invoice_id.partner_id.id or False,
555                             'membership_id': line.product_id.id,
556                             'member_price': line.price_unit,
557                             'date': time.strftime('%Y-%m-%d'),
558                             'date_from': date_from,
559                             'date_to': date_to,
560                             'account_invoice_line': line.id,
561                         }, context=context)
562         return result
563
564 account_invoice_line()
565
566 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: