[FIX] Account: Compare_context added in wizard

[odoo/odoo.git] / addons / account / wizard / account_subscription_generate.py

diff --git a/addons/account/wizard/account_subscription_generate.py b/addons/account/wizard/account_subscription_generate.py
index 5c44a61..8efcd25 100644 (file)
--- a/addons/account/wizard/account_subscription_generate.py
+++ b/addons/account/wizard/account_subscription_generate.py
@@ -18,6 +18,7 @@
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 ##############################################################################
+
 import time
 
 from osv import fields, osv
@@ -28,17 +29,25 @@ class account_subscription_generate(osv.osv_memory):
     _description = "Subscription Compute"
     _columns = {
        'date': fields.date('Date', required=True),
-              }
+    }
     _defaults = {
         'date': lambda *a: time.strftime('%Y-%m-%d'),
-                }
-    def action_generate(self, cr, uid, ids, context={}):
-        for data in  self.read(cr, uid, ids,context=context):
-             cr.execute('select id from account_subscription_line where date<%s and move_id is null', (data['date'],))
-             ids = map(lambda x: x[0], cr.fetchall())
-             self.pool.get('account.subscription.line').move_create(cr, uid, ids)
-             return {}
+    }
+    def action_generate(self, cr, uid, ids, context=None):
+        mod_obj = self.pool.get('ir.model.data')
+        act_obj = self.pool.get('ir.actions.act_window')
+        moves_created=[]
+        for data in  self.read(cr, uid, ids, context=context):
+            cr.execute('select id from account_subscription_line where date<%s and move_id is null', (data['date'],))
+            line_ids = map(lambda x: x[0], cr.fetchall())
+            moves = self.pool.get('account.subscription.line').move_create(cr, uid, line_ids, context=context)
+            moves_created.extend(moves)
+        result = mod_obj.get_object_reference(cr, uid, 'account', 'action_move_line_form')
+        id = result and result[1] or False
+        result = act_obj.read(cr, uid, [id], context=context)[0]
+        result['domain'] = str([('id','in',moves_created)])
+        return result
 
 account_subscription_generate()
-# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
\ No newline at end of file