Wobuzz/wobuzz/library/library.py

#!/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)
        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 = os.path.expanduser(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_widget.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()

    def new_playlist(self):
        playlist = Playlist(self.app, self.app.utils.unique_name("New Playlist"))
        self.playlists.append(playlist)

        for library_dock in self.library_docks:
            playlist_tabs: QTabWidget = library_dock.library_widget.playlist_tabs

            playlist_view = PlaylistView(playlist)
            playlist_tabs.addTab(playlist_view, playlist.title)