Wobuzz/wobuzz/library/library.py

63 lines
1.8 KiB
Python

#!/usr/bin/python3
import os
from PyQt6.QtWidgets import QTabWidget
from ..player.playlist import Playlist
from ..ui.library_dock import LibraryDock
from ..ui.playlist import PlaylistView
class Library:
"""
Class that manages and syncs the library and library gui dock widgets.
"""
def __init__(self, app):
self.app = app
self.main_library_dock = LibraryDock()
self.library_docks = [self.main_library_dock]
self.temporary_playlist = Playlist(self.app, "Temporary Playlist")
self.playlists = [self.temporary_playlist]
def load(self):
path_playlists = f"{self.app.settings.library_path}/playlists"
if not os.path.exists(path_playlists):
os.makedirs(path_playlists)
self.load_playlist_views()
return
folder_content = os.listdir(path_playlists)
for file_name in folder_content:
if file_name.endswith(".m3u"):
path = f"{path_playlists}/{file_name}"
if file_name == "Temporary_Playlist.wbz.m3u":
playlist = self.temporary_playlist
else:
playlist = Playlist(self.app, file_name.replace("_", " ").split(".")[0])
self.playlists.append(playlist)
playlist.load_from_m3u(path)
self.load_playlist_views()
def load_playlist_views(self):
for library_dock in self.library_docks:
playlist_tabs: QTabWidget = library_dock.library.playlist_tabs
playlist_tabs.playlists = {}
for playlist in self.playlists:
playlist_view = PlaylistView(playlist)
playlist_tabs.addTab(playlist_view, playlist.title)
def on_exit(self, event):
for playlist in self.playlists:
playlist.save()