Add function get_track to find track by its title

UmSenhorQualquer · Apr 15, 2019 · 05f3b96 · 05f3b96
1 parent 60146fe
commit 05f3b96
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/pyforms_gui/controls/control_event_timeline/control_eventtimeline.py b/pyforms_gui/controls/control_event_timeline/control_eventtimeline.py
@@ -246,6 +246,14 @@ def add_track(self, title='', color=None):
         """
         return self._time.add_track(title=title, color=color)
 
+    def get_track(self, title):
+        """
+        Get a track by its title
+        :param str title: Title of the track.
+        :return: Return the track with the matching title.
+        """
+        return self._time.get_track(title)
+
 
     def rename_graph(self, graph_index, newname):
         """

diff --git a/pyforms_gui/controls/control_event_timeline/timeline_widget.py b/pyforms_gui/controls/control_event_timeline/timeline_widget.py
@@ -136,6 +136,17 @@ def add_track(self, title='', color=None):
         self.setMinimumHeight(Track.which_top(len(self._tracks)))
         return t
 
+    def get_track(self, title):
+        """
+        Get a track by its title
+        :param str title: Title of the track.
+        :return: Return the track with the matching title.
+        """
+        for track in self.tracks:
+            if track.title==title:
+                return track
+        return None
+
     def add_event(self, begin, end, title='', track=None, lock=False, color=None, row=0):
         """
         Add a new event to the timeline.