Set track end time when setting next track. Fixes #33

app/playlists.py

@@ -311,6 +311,15 @@ class PlaylistTab(QTableWidget):
 
         return result
 
+    def get_selected_title(self):
+        "Return title of selected row or None"
+
+        if self.selectionModel().hasSelection():
+            row = self.currentRow()
+            return self.item(row, self.COL_TITLE).text()
+        else:
+            return None
+
     def remove_rows(self, rows):
         "Remove rows passed in rows list"
 
@@ -323,15 +332,6 @@ class PlaylistTab(QTableWidget):
 
         self._repaint()
 
-    def get_selected_title(self):
-        "Return title of selected row or None"
-
-        if self.selectionModel().hasSelection():
-            row = self.currentRow()
-            return self.item(row, self.COL_TITLE).text()
-        else:
-            return None
-
     def play_started(self):
         """
         Update current track to be what was next, and determine next track.
@@ -834,6 +834,8 @@ class PlaylistTab(QTableWidget):
                     self._set_row_start_time(row, start_time)
                     next_start_time = self._calculate_next_start_time(
                         session, row, start_time)
+                    # Set end time
+                    self._set_row_end_time(row, next_start_time)
                     # Set colour
                     self._set_row_colour(
                         row, QColor(Config.COLOUR_NEXT_PLAYLIST))