1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # Remembership module for OpenERP, Overload membership module
5 # Copyright (C) 2012 L'Heureux Cyclage (<http://www.heureux-cyclage.org>) Ludovic CHEVALIER
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 ##############################################################################
25 from osv
import fields
27 from datetime
import datetime
, date
28 from dateutil
.relativedelta
import relativedelta
31 class Partner(osv
.osv
):
32 _inherit
= 'res.partner'
34 # XXX: copied from openerp/addons/membership/membership.py
35 def _membership_date(self
, cr
, uid
, ids
, name
, args
, context
=None):
36 """Return date of membership"""
39 member_line_obj
= self
.pool
.get('membership.membership_line')
40 for partner
in self
.browse(cr
, uid
, ids
, context
=context
):
41 # XXX: commented part from openerp/addons/membership/membership.py
42 #if partner.associate_member:
43 # partner_id = partner.associate_member.id
45 # NOTE: hmm, ça devrait être toujours partner.id
46 partner_id
= partner
.id
48 'membership_start': False,
49 'membership_stop': False,
50 'membership_cancel': False
52 if name
== 'membership_start':
53 line_id
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner_id
),('date_cancel','=',False)],
54 limit
=1, order
='date_from', context
=context
)
56 res
[partner
.id]['membership_start'] = member_line_obj
.read(cr
, uid
, line_id
[0],
57 ['date_from'], context
=context
)['date_from']
59 if name
== 'membership_stop':
60 line_id1
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner_id
),('date_cancel','=',False)],
61 limit
=1, order
='date_to desc', context
=context
)
63 res
[partner
.id]['membership_stop'] = member_line_obj
.read(cr
, uid
, line_id1
[0],
64 ['date_to'], context
=context
)['date_to']
66 if name
== 'membership_cancel':
67 if partner
.membership_state
== 'canceled':
68 line_id2
= member_line_obj
.search(cr
, uid
, [('partner', '=', partner
.id)], limit
=1, order
='date_cancel', context
=context
)
70 res
[partner
.id]['membership_cancel'] = member_line_obj
.read(cr
, uid
, line_id2
[0], ['date_cancel'], context
=context
)['date_cancel']
72 # XXX: copied from openerp/addons/membership/membership.py
73 def _get_partner_id(self
, cr
, uid
, ids
, context
=None):
74 member_line_obj
= self
.pool
.get('membership.membership_line')
75 res_obj
= self
.pool
.get('res.partner')
76 data_inv
= member_line_obj
.browse(cr
, uid
, ids
, context
=context
)
79 list_partner
.append(data
.partner
.id)
82 ids2
= res_obj
.search(cr
, uid
, [('associate_member', 'in', ids2
)], context
=context
)
85 # XXX: copied from openerp/addons/membership/membership.py
86 def _get_invoice_partner(self
, cr
, uid
, ids
, context
=None):
87 inv_obj
= self
.pool
.get('account.invoice')
88 res_obj
= self
.pool
.get('res.partner')
89 data_inv
= inv_obj
.browse(cr
, uid
, ids
, context
=context
)
92 list_partner
.append(data
.partner_id
.id)
95 ids2
= res_obj
.search(cr
, uid
, [('associate_member', 'in', ids2
)], context
=context
)
100 'member_ident': fields
.char('Member identifier', size
=64),
101 'associate_members': fields
.one2many('res.partner', 'associate_member', 'Associate members', help='Members who are associated to this partner.'),
102 # XXX: copied from openerp/addons/membership/membership.py
103 'membership_start': fields
.function(
104 _membership_date
, multi
= 'membership_start',
105 string
= 'Membership Start Date', type = 'date',
107 'account.invoice': (_get_invoice_partner
, ['state'], 10),
108 'membership.membership_line': (_get_partner_id
, ['state'], 10, ),
109 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
110 }, help="Date from which membership becomes active."),
111 # XXX: copied from openerp/addons/membership/membership.py
112 'membership_stop': fields
.function(
114 string
= 'Membership End Date', type='date', multi
='membership_stop',
116 'account.invoice': (_get_invoice_partner
, ['state'], 10),
117 'membership.membership_line': (_get_partner_id
, ['state'], 10),
118 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
119 }, help="Date until which membership remains active."),
120 # XXX: copied from openerp/addons/membership/membership.py
121 'membership_cancel': fields
.function(
123 string
= 'Cancel Membership Date', type='date', multi
='membership_cancel',
125 'account.invoice': (_get_invoice_partner
, ['state'], 11),
126 'membership.membership_line': (_get_partner_id
, ['state'], 10),
127 'res.partner': (lambda self
, cr
, uid
, ids
, c
={}: ids
, ['free_member'], 10)
128 }, help="Date on which membership has been cancelled"),
132 ('member_ident_uniq', 'unique(member_ident, company_id)', 'The member identifier must be unique !'),
135 def create_membership_invoice(self
, cr
, uid
, ids
, product_id
=None, datas
=None, context
=None):
136 # NOTE: use context to supply date_from to account_invoice_line.create
137 ctx
= context
and context
.copy() or {}
138 ctx
['date_from'] = datas
.get('date_from', None)
139 return super(Partner
, self
).create_membership_invoice(cr
, uid
, ids
, product_id
=product_id
, datas
=datas
, context
=ctx
)
144 class Product(osv
.osv
):
145 _inherit
= 'product.product'
148 'membership_grouped': fields
.boolean('Grouped membership product', help='Check if it\'s a grouped membership product.'),
149 'membership_date2date': fields
.boolean('Date to date membership product', required
=False, help='Check if it\'s a date to date membership product.'),
155 class account_invoice_line(osv
.osv
):
156 _inherit
= 'account.invoice.line'
158 def write(self
, cr
, uid
, ids
, vals
, context
=None):
159 member_line_obj
= self
.pool
.get('membership.membership_line')
160 res
= super(account_invoice_line
, self
).write(cr
, uid
, ids
, vals
, context
=context
)
161 for line
in self
.browse(cr
, uid
, ids
, context
=context
):
162 if line
.invoice_id
.type == 'out_invoice':
163 ml_ids
= member_line_obj
.search(cr
, uid
, [('account_invoice_line', '=', line
.id)], context
=context
)
164 if line
.product_id
and line
.product_id
.membership
:
165 member_line_id
= member_line_obj
.search(cr
, uid
166 , [('partner' , '=', line
.invoice_id
.partner_id
.id)
167 ,('account_invoice_line', '=', line
.id)]
170 for member_line
in member_line_obj
.browse(cr
, uid
, member_line_id
, context
=context
):
171 # NOTE: get member_line created in membership.account_invoice_line.create
172 date_from
= member_line
.date_from
173 date_to
= member_line
.date_to
174 if line
.product_id
.membership_date2date
:
175 date_from
= context
['date_from'] and datetime
.strptime(context
['date_from'], "%Y-%m-%d") or date
.today()
176 date_to
= date_from
+ relativedelta(months
= +12) # TODO: parameterize this delta?
177 date_from
= date_from
.strftime("%Y-%m-%d")
178 date_to
= date_to
.strftime("%Y-%m-%d")
179 member_line_obj
.write(cr
, uid
, member_line
.id
180 , {'date_from': date_from
184 if line
.product_id
.membership_grouped
:
185 if line
.invoice_id
.partner_id
.associate_members
:
186 associate_member_line_ids
= member_line_obj
.search(cr
, uid
187 , [ ('account_invoice_line', '=', line
.id)
188 , ('partner', '!=', line
.invoice_id
.partner_id
.id)
191 for associate_member_line
in member_line_obj
.browse(cr
, uid
, associate_member_line_ids
, context
=context
):
192 member_line_obj
.write(cr
, uid
, associate_member_line
.id
193 , {'date_from': date_from
198 print("DEV: mettre une contrainte pour l'objet\
199 membership.membership_line interdisant les\
200 adhésions groupées reliées à des partenaires\
201 sans membres associés")
203 associate_member_line_ids
= member_line_obj
.search(cr
, uid
204 , [ ('account_invoice_line', '=', line
.id)
205 , ('partner', '!=', line
.invoice_id
.partner_id
.id)
208 member_line_obj
.unlink(cr
, uid
, associate_member_line_ids
, context
=context
)
210 #Define member ident if it's necessary
211 partners
= [line
.invoice_id
.partner_id
]
212 if line
.invoice_id
.partner_id
.associate_members
:
213 partners
.extend(line
.invoice_id
.partner_id
.associate_members
)
215 if not i
.member_ident
:
216 mbr_id
= self
.pool
.get('ir.sequence').get(cr
, uid
, 'member_ident')
217 self
.pool
.get('res.partner').write(cr
, uid
, i
.id, {'member_ident': mbr_id
})
218 if line
.product_id
and not line
.product_id
.membership
and ml_ids
:
219 # Product line has changed to a non membership product
220 member_line_obj
.unlink(cr
, uid
, ml_ids
, context
=context
)
223 def create(self
, cr
, uid
, vals
, context
=None):
224 member_line_obj
= self
.pool
.get('membership.membership_line')
225 res
= super(account_invoice_line
, self
).create(cr
, uid
, vals
, context
=context
)
226 line
= self
.browse(cr
, uid
, res
, context
=context
)
228 if line
.invoice_id
.type == 'out_invoice' and line
.product_id
and line
.product_id
.membership
:
229 member_line_id
= member_line_obj
.search(cr
, uid
230 , [('partner' , '=', line
.invoice_id
.partner_id
.id)
231 ,('account_invoice_line', '=', line
.id)]
234 for member_line
in member_line_obj
.browse(cr
, uid
, member_line_id
, context
=context
):
235 # NOTE: get member_line created in membership.account_invoice_line.create
236 date_from
= member_line
.date_from
237 date_to
= member_line
.date_to
238 if line
.product_id
.membership_date2date
:
239 date_from
= context
['date_from'] and datetime
.strptime(context
['date_from'], "%Y-%m-%d") or date
.today()
240 date_to
= date_from
+ relativedelta(months
= +12) # TODO: parameterize this delta?
241 date_from
= date_from
.strftime("%Y-%m-%d")
242 date_to
= date_to
.strftime("%Y-%m-%d")
243 member_line_obj
.write(cr
, uid
, member_line
.id
244 , {'date_from': date_from
248 partners
= [line
.invoice_id
.partner_id
]
249 if line
.product_id
.membership_grouped
and line
.invoice_id
.partner_id
.associate_members
:
250 partners
.extend(line
.invoice_id
.partner_id
.associate_members
)
251 #Adding membership lines just for associate partners
252 for associate_member
in line
.invoice_id
.partner_id
.associate_members
:
253 member_line_obj
.create(cr
, uid
, {
254 'partner': associate_member
.id,
255 'membership_id': line
.product_id
.id,
256 'member_price': line
.price_unit
,
257 'date': time
.strftime('%Y-%m-%d'),
258 'date_from': date_from
,
260 'account_invoice_line': line
.id,
262 #Define member ident if it's necessary
264 if not i
.member_ident
:
265 mbr_id
= self
.pool
.get('ir.sequence').get(cr
, uid
, 'member_ident')
266 self
.pool
.get('res.partner').write(cr
, uid
, i
.id, {'member_ident': mbr_id
})
269 account_invoice_line()
272 class pos_make_payment(osv
.osv_memory
):
273 _inherit
= 'pos.make.payment'
275 def check(self
, cr
, uid
, ids
, context
=None):
277 Auto invoice orders. Temporarly solution before make partner members just with orders, without invoice.
279 res
= super(pos_make_payment
, self
).check(cr
, uid
, ids
, context
=context
)
281 order_obj
= self
.pool
.get('pos.order')
282 order_obj
.action_invoice(cr
, uid
, ids
, context
=context
)
288 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: