Wobuzz/wobuzz/player/playlist.py

#!/usr/bin/python3

import os
from PyQt6.QtCore import Qt
from .track import Track


class Playlist:
    def __init__(self, app, title: str):
        self.app = app
        self.title = title  # playlist title
        self.sorting: list[Qt.SortOrder] | None = None  # Custom sort order if None
        self.tracks: list[Track] = []
        self.current_track_index = 0
        self.current_track: Track | None = None
        self.view = None

    def clear(self):
        self.sorting: list[Qt.SortOrder] | None = None
        self.tracks = []
        self.current_track_index = 0
        self.current_track = None

    def load_from_paths(self, paths):
        i = 0

        while i < len(paths):
            path = paths[i]

            if os.path.isfile(path):
                self.tracks.append(Track(self.app, path, cache=i==0))  # first track is cached

            i += 1

        # set current track to the first track if there is no currently playing track
        if self.current_track is None and self.has_tracks():
            self.current_track = self.tracks[0]

    def load_from_m3u(self, path):
        file = open(path, "r")
        m3u = file.read()
        file.close()

        lines = m3u.split("\n")  # m3u entries are separated by newlines
        lines = lines[:-1]  # remove last entry because it is just an empty string

        i = 0
        num_lines = len(lines)

        while i < num_lines:
            line = lines[i]

            if line.startswith("#") or line.startswith("http"):  # filter out comments, extended m3u and urls
                i += 1

                continue

            self.tracks.append(Track(self.app, line, cache=i==0))  # first track is cached

            i += 1

        # set current track to the first track if there is no currently playing track
        if self.current_track is None and self.has_tracks():
            self.current_track = self.tracks[0]

            #self.app.player.history.append_track(self.current_track)

    def load_from_wbz(self, path):
        pass

    def has_tracks(self):
        return len(self.tracks) > 0

    def on_first_track(self):
        return self.current_track_index == 0

    def on_last_track(self):  # if the current track is the last
        return self.current_track_index == len(self.tracks) - 1

    def next_track(self):
        self.current_track_index += 1
        self.current_track = self.tracks[self.current_track_index]

        if not self.current_track.cached:  # make sure the track is cached because else the player can't play it
            self.current_track.cache()

        return self.current_track.sound, self.current_track.duration

    def previous_track(self):
        if self.on_first_track():
            return self.current_track, self.current_track.duration

        self.current_track_index -= 1
        self.current_track = self.tracks[self.current_track_index]

        if not self.current_track.cached:  # make sure the track is cached because else the player can't play it
            self.current_track.cache()

        return self.current_track.sound, self.current_track.duration

    def set_track(self, track_index):
        self.current_track_index = track_index

        self.current_track = self.tracks[self.current_track_index]

        if not self.current_track.cached:
            self.current_track.cache()

        return self.current_track.sound, self.current_track.duration

    def save(self):
        wbz_data = ""

        for track in self.tracks:
            wbz_data += f"{track.path}\n"

        wbz = open(
            os.path.expanduser(
                f"{self.app.settings.library_path}/playlists/{self.title.replace(" ", "_")}.wbz.m3u"
            ),
            "w"
        )
        wbz.write(wbz_data)
        wbz.close()

    def delete(self):
        path = f"{self.app.settings.library_path}/playlists/{self.title.replace(" ", "_")}.wbz.m3u"

        if os.path.exists(path):
            os.remove(os.path.expanduser(path))

    def append_track(self, track):
        self.tracks.append(track)

        if self.view:
            self.view.append_track(track)

    def h_last_track(self):
        # get last track in history (only gets used in player.history)

        if len(self.tracks) > 1:
            return self.tracks[-2]