From 43b5d7475e49af1fc194b8ced101e6b2e9047ac8 Mon Sep 17 00:00:00 2001
From: Guewen Baconnier <guewen.baconnier@camptocamp.com>
Date: Mon, 15 Dec 2014 14:29:59 +0100
Subject: [PATCH] Add a button on the "Run" to open the generated lines

---
 account_credit_control/line.py      |  3 +++
 account_credit_control/run.py       | 19 ++++++++++++++++++-
 account_credit_control/run_view.xml |  4 ++++
 3 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/account_credit_control/line.py b/account_credit_control/line.py
index abbfbccfd..326778870 100644
--- a/account_credit_control/line.py
+++ b/account_credit_control/line.py
@@ -137,6 +137,9 @@ class CreditControlLine(models.Model):
 
     manually_overridden = fields.Boolean(string='Manually overridden')
 
+    run_id = fields.Many2one(comodel_name='credit.control.run',
+                             string='Source')
+
     @api.model
     def _prepare_from_move_line(self, move_line, level, controlling_date,
                                 open_amount):
diff --git a/account_credit_control/run.py b/account_credit_control/run.py
index 417af2839..c6afb1d1e 100644
--- a/account_credit_control/run.py
+++ b/account_credit_control/run.py
@@ -56,6 +56,12 @@ class CreditControlRun(models.Model):
                              required=True,
                              readonly=True,
                              default='draft')
+
+    line_ids = fields.One2many(
+        comodel_name='credit.control.line',
+        inverse_name='run_id',
+        string='Generated lines')
+
     manual_ids = fields.Many2many(
         'account.move.line',
         rel="credit_runreject_rel",
@@ -134,7 +140,8 @@ class CreditControlRun(models.Model):
 
         vals = {'state': 'done',
                 'report': report,
-                'manual_ids': [(6, 0, manually_managed_lines)]}
+                'manual_ids': [(6, 0, manually_managed_lines.ids)],
+                'line_ids': [(6, 0, generated.ids)]}
         self.write(vals)
         return generated
 
@@ -156,3 +163,13 @@ class CreditControlRun(models.Model):
 
         self._generate_credit_lines()
         return True
+
+    @api.multi
+    def open_credit_lines(self):
+        """ Open the generated lines """
+        self.ensure_one()
+        action_name = 'account_credit_control.credit_control_line_action'
+        action = self.env.ref(action_name)
+        action = action.read()[0]
+        action['domain'] = [('id', 'in', self.line_ids.ids)]
+        return action
diff --git a/account_credit_control/run_view.xml b/account_credit_control/run_view.xml
index 9eee16539..d4bcf0ca9 100644
--- a/account_credit_control/run_view.xml
+++ b/account_credit_control/run_view.xml
@@ -23,6 +23,10 @@
               class="oe_highlight"
               type="object" icon="gtk-execute"
               attrs="{'invisible': [('state', '!=', 'draft')]}"/>
+            <button name="open_credit_lines"
+              string="Open Credit Control Lines"
+              type="object"
+              attrs="{'invisible': [('state', '=', 'draft')]}"/>
             <field name="state" widget="statusbar"
               statusbar_visible="draft,done"
               statusbar_colors='{}'/>