Add unmark as next, played to context menu

Fixes #166
Fixex #138

app/musicmuster.py

@@ -427,6 +427,13 @@ class Window(QMainWindow, Ui_MainWindow):
                     btn.setStyleSheet("background-color: " + colour + ";\n")
                     btn.pgb.setValue(0)
 
+    def clear_next(self) -> None:
+        """
+        Clear next track
+        """
+
+        self.next_track = PlaylistTrack()
+
     def clear_selection(self) -> None:
         """ Clear selected row"""
 
@@ -1183,7 +1190,7 @@ class Window(QMainWindow, Ui_MainWindow):
             # stop_playing() above has called end_of_track_actions()
             # which will have populated self.previous_track
             self.current_track = self.next_track
-            self.next_track = PlaylistTrack()
+            self.clear_next()
 
             if not self.current_track.track_id:
                 log.debug("musicmuster.play_next(): no id for next track")
@@ -1490,7 +1497,7 @@ class Window(QMainWindow, Ui_MainWindow):
                                 QColor(Config.COLOUR_NORMAL_TAB))
 
             # Discard now-incorrect next_track PlaylistTrack
-            self.next_track = PlaylistTrack()
+            self.clear_next()
 
         self.next_track.set_plr(session, plr, playlist_tab)
         if self.next_track.playlist_tab:

app/playlists.py

@@ -265,18 +265,38 @@ class PlaylistTab(QTableWidget):
             self.menu = QMenu(self)
             item = self.itemAt(event.pos())
             if item is not None:
+                with Session() as session:
                     row_number = item.row()
-                track_id = self._get_row_track_id(row_number)
+                    plr_id = self._get_playlistrow_id(row_number)
+                    plr = session.get(PlaylistRows, plr_id)
+                    track_id = plr.track_id
                     track_row = track_id > 0
                     header_row = not track_row
                     if track_row:
                         current = (
                             row_number == self._get_current_track_row_number()
                         )
-                    next_row = row_number == self._get_next_track_row_number()
+                        next_row = (
+                            row_number == self._get_next_track_row_number())
                     else:
                         current = next_row = False
 
+                    # Mark unplayed / un-next
+                    sep = False
+                    if plr.played:
+                        sep = True
+                        act_unplay = self.menu.addAction("Mark unplayed")
+                        act_unplay.triggered.connect(
+                            lambda: self._mark_unplayed(plr))
+                    if next_row:
+                        sep = True
+                        act_unnext = self.menu.addAction(
+                            "Unmark as next track")
+                        act_unnext.triggered.connect(
+                            lambda: self._mark_unnext(row_number))
+                    if sep:
+                        self.menu.addSeparator()
+
                     # Cut/paste
                     act_cut = self.menu.addAction(
                         "Mark for moving")
@@ -313,7 +333,6 @@ class PlaylistTab(QTableWidget):
                         # Set next
                         if not current and not next_row:
                             act_setnext = self.menu.addAction("Set next")
-                        with Session() as session:
                             act_setnext.triggered.connect(
                                 lambda: self._set_next(session, row_number))
 
@@ -347,7 +366,8 @@ class PlaylistTab(QTableWidget):
                             self.menu.addSeparator()
 
                             # Remove track
-                        act_remove_track = self.menu.addAction('Remove track')
+                            act_remove_track = self.menu.addAction(
+                                'Remove track')
                             act_remove_track.triggered.connect(
                                 lambda: self._remove_track(row_number)
                             )
@@ -367,7 +387,8 @@ class PlaylistTab(QTableWidget):
 
                     if not current and not next_row:
                         act_move = self.menu.addAction('Move to playlist...')
-                    act_move.triggered.connect(self.musicmuster.move_selected)
+                        act_move.triggered.connect(
+                            self.musicmuster.move_selected)
                         self.menu.addSeparator()
 
         return super(PlaylistTab, self).eventFilter(source, event)
@@ -1569,6 +1590,27 @@ class PlaylistTab(QTableWidget):
                 and pos.y() >= rect.center().y()  # noqa W503
         )
 
+    def _mark_unnext(self, row: int) -> None:
+        """
+        Unmark passed row as next track
+        """
+
+        self.musicmuster.clear_next()
+        with Session() as session:
+            self.update_display(session)
+        self.musicmuster.update_headers()
+
+    def _mark_unplayed(self, plr: PlaylistRows) -> None:
+        """
+        Mark passed row as unplayed in this playlist
+        """
+
+        with Session() as session:
+            session.add(plr)
+            plr.played = False
+            session.flush()
+            self.update_display(session)
+
     def _move_row(self, session: scoped_session, plr: PlaylistRows,
                   new_row_number: int) -> None:
         """Move playlist row to new_row_number using parent copy/paste"""