From 4f29b9ae0b518bcc41038d6d300429e5d656d8e0 Mon Sep 17 00:00:00 2001
From: Sebastian Koslowski <koslowski@kit.edu>
Date: Fri, 8 Jul 2016 15:18:58 +0200
Subject: grc: refactor: block state handling

---
 grc/gui/FlowGraph.py | 25 +++++--------------------
 1 file changed, 5 insertions(+), 20 deletions(-)

(limited to 'grc/gui/FlowGraph.py')

diff --git a/grc/gui/FlowGraph.py b/grc/gui/FlowGraph.py
index dc80a4c87e..87bd91d880 100644
--- a/grc/gui/FlowGraph.py
+++ b/grc/gui/FlowGraph.py
@@ -273,35 +273,20 @@ class FlowGraph(Element, _Flowgraph):
         """
         return any(sb.port_controller_modify(direction) for sb in self.get_selected_blocks())
 
-    def enable_selected(self, enable):
+    def change_state_selected(self, new_state):
         """
         Enable/disable the selected blocks.
 
         Args:
-            enable: true to enable
+            new_state: a block state
 
         Returns:
             true if changed
         """
         changed = False
-        for selected_block in self.get_selected_blocks():
-            if selected_block.set_enabled(enable):
-                changed = True
-        return changed
-
-    def bypass_selected(self):
-        """
-        Bypass the selected blocks.
-
-        Args:
-            None
-        Returns:
-            true if changed
-        """
-        changed = False
-        for selected_block in self.get_selected_blocks():
-            if selected_block.set_bypassed():
-                changed = True
+        for block in self.selected_blocks():
+            changed |= block.state != new_state
+            block.state = new_state
         return changed
 
     def move_selected(self, delta_coordinate):
-- 
cgit v1.2.3