Separe PlayCtrl in gui and functionality classes

Previously playCtrl had part of gui handling, and functionality due to
untoggle the play button when playback stops.

Author: WesleyMPG

atbswp/control.py

@@ -353,6 +353,116 @@ def update_timer(self, event):
             self.timer = self.countdown_dialog.update_ui()
 
 
+class PlayControl:
+    """Control class for playback functionality"""
+
+    def __init__(self, count=1, infinite=False):
+        self.count = count
+        self.infinite = infinite
+        self._stop_locks = [True, True]  # [once, many]
+        self._current_count = 0
+        self._play_once_thread = None
+        self._play_many_thread = None
+
+    def _play_once(self, capture, append_function=None):
+        for line in capture:
+            if self._stop_locks[0]: break
+            exec(line)
+        self._stop_locks[0] = True
+        if append_function:
+            append_function()
+
+    def _play_many(self, capture, append_function=None):
+        self.current_count = 0
+        while (self.current_count < self.count or self.infinite)\
+                and not self._stop_locks[1]:
+            self._stop_locks[0] = False
+            self._play_once_thread = self._start_thread(self._play_once, capture)
+            self._play_once_thread.join()
+            self.current_count += 1
+        self._stop_locks = [True, True]
+        if append_function:
+            append_function()
+
+    def _start_thread(self, target, *args, daemon=False):
+        thread = Thread(target=target, args=(*args,))
+        thread.daemon = daemon
+        thread.start()
+        return thread
+
+    def set_config(self, count, infinite):
+        """Set the configuration to the next playback"""
+        self.count = count
+        self.infinite = infinite
+
+    def get_current_count(self):
+        """Returns the current playback time"""
+        return current_count
+
+    def is_stoped(self):
+        return self._stop_locks[0] and self._stop_locks[1]
+
+    def play(self, capture, append_function=None):
+        """Starts playback according with configs setted
+
+            :param capture: list of commands to be executed
+            :param append_function: a function to be executed after
+             a playback
+        """
+        if self.infinite or self.count > 1:
+            self._stop_locks = [False, False]
+            self._start_thread(self._play_many, capture,
+                               append_function)
+        else:
+            self._stop_locks[0] = False
+            self._start_thread(self._play_once, capture,
+                               append_function)
+
+    def stop(self):
+        self._stop_locks = [True, True]
+
+
+class PlayInterface:
+    def __init__(self):
+        self.play_ctrl = PlayControl()
+        self.capture = None
+        self._config_was_updated = False
+        self._toggle_button = None
+
+    def _load_capture_file(self):
+        if TMP_PATH is None or not os.path.isfile(TMP_PATH):
+            wx.LogError("No capture loaded")
+            self._toggle_button.Value = False
+            return False
+        with open(TMP_PATH, 'r') as f:
+            self.capture = f.readlines()
+            return True
+
+    def _set_config(self):
+        if self._config_was_updated: return
+        count = settings.CONFIG.getint('DEFAULT', 'Repeat Count')
+        infinite = settings.CONFIG.getboolean('DEFAULT', 'Infinite Playback')
+        self.play_ctrl.set_config(count, infinite)
+        self._config_was_updated = False
+
+    def _stop(self):
+        self._toggle_button.Value = False
+        self.play_ctrl.stop()
+        self._config_was_updated = False
+        settings.save_config()
+
+    def action(self, event):
+        self._toggle_button = event.GetEventObject()
+        self._toggle_button.Parent.panel.SetFocus()
+        if self._toggle_button.Value:
+            if self.play_ctrl.is_stoped():
+                if not self._load_capture_file(): return
+                self._set_config()
+                self.play_ctrl.play(self.capture, self._stop)
+        else:
+            self._stop()
+
+
 class PlayCtrl:
     """Control class for the play button."""
 
@@ -532,3 +642,5 @@ def end(self):
 
     def ended(self):
         return self._end.isSet()
+
+

atbswp/gui.py

@@ -201,7 +201,7 @@ def __add_bindings(self):
         self.Bind(wx.EVT_TOGGLEBUTTON, self.rbc.action, self.record_button)
 
         # play_button_ctrl
-        self.pbc = control.PlayCtrl()
+        self.pbc = control.PlayInterface()
         self.Bind(wx.EVT_TOGGLEBUTTON, self.pbc.action, self.play_button)
 
         # compile_button_ctrl