From f4ae3f7e2627d20f6319b51fbd7ee6f05b63657e Mon Sep 17 00:00:00 2001
From: Fabien Pinckaers <fp@tinyerp.com>
Date: Thu, 15 Jan 2009 17:57:02 +0100
Subject: [PATCH] modifs

bzr revid: fp@tinyerp.com-20090115165702-9qo212e0614c5ovo
---
 addons/hr_timesheet_sheet/hr_timesheet_sheet.py | 3 +++
 addons/membership/membership.py                 | 9 ++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/addons/hr_timesheet_sheet/hr_timesheet_sheet.py b/addons/hr_timesheet_sheet/hr_timesheet_sheet.py
index c0ab418fd67..83ce4f5168f 100644
--- a/addons/hr_timesheet_sheet/hr_timesheet_sheet.py
+++ b/addons/hr_timesheet_sheet/hr_timesheet_sheet.py
@@ -135,6 +135,9 @@ class hr_timesheet_sheet(osv.osv):
                 result[sheet_id] = emp.state
         return result
 
+    def copy(self, cr, uid, ids, *args, **argv):
+        raise osv.except_osv(_('Error !'), _('You can not duplicate a timesheet !'))
+
     def button_confirm(self, cr, uid, ids, context):
         for sheet in self.browse(cr, uid, ids, context):
             di = sheet.user_id.company_id.timesheet_max_difference
diff --git a/addons/membership/membership.py b/addons/membership/membership.py
index e1a4e2156f6..76633abb496 100644
--- a/addons/membership/membership.py
+++ b/addons/membership/membership.py
@@ -435,6 +435,13 @@ class Partner(osv.osv):
 
 
     _inherit = 'res.partner'
+    def _get_partners(self, cr, uid, ids, context={}):
+        ids2 = ids
+        while ids2:
+            ids2 = self.search(cr, uid, [('associate_member','in',ids2)], context=context)
+            ids+=ids2
+        return ids
+
     _columns = {
         'member_lines': fields.one2many('membership.membership_line', 'partner',
             'Membership'),
@@ -445,7 +452,7 @@ class Partner(osv.osv):
         'membership_state': fields.function(_membership_state, method=True, string='Current membership state',
             type='selection',selection=STATE,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)}),
+                                                    'res.partner':(_get_partners, ['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,