[PYTHON] ~fix update_membership_state function

[burette/remembership.git] / remembership.py

diff --git a/remembership.py b/remembership.py
index 7a03490..c8d8b46 100644 (file)
--- a/remembership.py
+++ b/remembership.py
@@ -2,7 +2,7 @@
 ##############################################################################
 #
 #    Remembership module for OpenERP, Membership module enhancement and bug fixes
-#    Copyright (C) 2012-2013 L'Heureux Cyclage (<http://www.heureux-cyclage.org>)
+#    Copyright (C) 2012-2018 L'Heureux Cyclage (<http://www.heureux-cyclage.org>)
 #
 #    This file is a part of Remembership
 #
@@ -21,7 +21,8 @@
 #
 ##############################################################################
 
-from openerp.osv import fields, osv
+from openerp.osv import osv
+from openerp.osv import fields
 import time
 from datetime import datetime, date
 from dateutil.relativedelta import relativedelta
@@ -212,13 +213,21 @@ class Partner(osv.osv):
         partner_obj = self.pool.get('res.partner')
         today = date.today()
         ids = partner_obj.search(cr, uid, [('membership_state', '=', 'paid'), ('membership_stop', '<', today)], context=None)
-        self.write(cr, uid, ids, {'membership_state': 'old'}, context=None)
+        for id in ids:
+            self.write(cr, uid, id, {'membership_state': 'old'}, context=None)
+
+    def copy(self, cr, uid, id, default=None, context=None):
+        if default is None:
+            default = {}
+        default['member_ident'] = None
+        return super(Partner, self).copy(cr, uid, id, default, context)
+
 
 Partner()
 
 
 class Product(osv.osv):
-    _inherit = 'product.template'
+    _inherit = 'product.product'
 
     _columns = {
         'membership_grouped': fields.boolean('Grouped membership product', help='Check if it\'s a grouped membership product.'),