100 lines
3 KiB
Python
100 lines
3 KiB
Python
#!/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
|
|
|
|
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
|
|
|
|
if self.current_track is None: # set current track to the first track if there is no currently playing track
|
|
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
|
|
lines = lines[:-1]
|
|
print(lines)
|
|
|
|
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
|
|
|
|
if self.current_track is None: # set current track to the first track if there is no currently playing track
|
|
self.current_track = self.tracks[0]
|
|
|
|
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
|
|
|