151 lines
5.1 KiB
Python
151 lines
5.1 KiB
Python
import logging
|
|
from PyQt5.QtWidgets import (
|
|
QCheckBox,
|
|
QDialog,
|
|
QHBoxLayout,
|
|
QLineEdit,
|
|
QLabel,
|
|
QPushButton,
|
|
QVBoxLayout,
|
|
QListWidget,
|
|
QListWidgetItem,
|
|
)
|
|
from PyQt5.QtGui import QFont
|
|
from configparser import ConfigParser
|
|
from utils import get_id3_tags, get_reorganize_vars
|
|
from components import ErrorDialog
|
|
import DBA
|
|
import os
|
|
|
|
|
|
class ExportPlaylistWindow(QDialog):
|
|
def __init__(self):
|
|
super(ExportPlaylistWindow, self).__init__()
|
|
self.setWindowTitle("Export playlist")
|
|
config = ConfigParser()
|
|
config.read("config.ini")
|
|
self.relative_path = config.get("directories", "playlist_relative_path")
|
|
self.export_path = config.get("directories", "playlist_export_path")
|
|
self.selected_playlist_name = "my-playlist.m3u"
|
|
self.current_m3u_path = self.export_path
|
|
self.current_relative_path = self.relative_path
|
|
layout = QVBoxLayout()
|
|
|
|
# Header label
|
|
label = QLabel("Playlists")
|
|
label.setFont(QFont("Sans", weight=QFont.Bold))
|
|
layout.addWidget(label)
|
|
|
|
# Get playlists from db
|
|
playlist_dict = {}
|
|
with DBA.DBAccess() as db:
|
|
data = db.query("SELECT id, name from playlist;", ())
|
|
for row in data:
|
|
playlist_dict[row[0]] = row[1]
|
|
|
|
# Playlist list widget
|
|
self.item_dict = {}
|
|
self.playlist_listWidget = QListWidget(self)
|
|
for i, (k, v) in enumerate(playlist_dict.items()):
|
|
# item_text = f"{i} | {v}"
|
|
item_text = v
|
|
item = QListWidgetItem(item_text)
|
|
self.playlist_listWidget.addItem(item)
|
|
self.item_dict[item_text] = k
|
|
|
|
layout.addWidget(self.playlist_listWidget)
|
|
|
|
# Relative path checkbox widget
|
|
self.checkbox = QCheckBox(text="Use relative paths?")
|
|
self.checkbox.setChecked(True)
|
|
layout.addWidget(self.checkbox)
|
|
|
|
# Relative export path label
|
|
label = QLabel("Relative path")
|
|
label.setFont(QFont("Sans", weight=QFont.Bold)) # bold category
|
|
label.setStyleSheet("text-transform:lowercase;") # uppercase category
|
|
layout.addWidget(label)
|
|
|
|
# Relative export path line edit widget
|
|
self.relative_path_input = QLineEdit(self.relative_path)
|
|
layout.addWidget(self.relative_path_input)
|
|
|
|
# Playlist file save path label
|
|
label = QLabel("Playlist file path")
|
|
label.setFont(QFont("Sans", weight=QFont.Bold))
|
|
label.setStyleSheet("text-transform:lowercase;")
|
|
layout.addWidget(label)
|
|
|
|
# Playlist file save path line edit widget
|
|
self.m3u_path_input = QLineEdit(self.export_path)
|
|
layout.addWidget(self.m3u_path_input)
|
|
|
|
# Save button
|
|
self.save_button = QPushButton("Export")
|
|
layout.addWidget(self.save_button)
|
|
|
|
# Signals
|
|
self.save_button.clicked.connect(self.save)
|
|
self.checkbox.toggled.connect(self.relative_path_input.setEnabled)
|
|
self.playlist_listWidget.currentItemChanged.connect(
|
|
self.set_current_selected_playlist
|
|
)
|
|
self.setLayout(layout)
|
|
self.show()
|
|
|
|
def set_current_selected_playlist(self) -> None:
|
|
self.selected_playlist_name = (
|
|
self.playlist_listWidget.selectedItems()[0].text() + ".m3u"
|
|
)
|
|
print(self.selected_playlist_name)
|
|
|
|
def save(self) -> None:
|
|
"""Exports the chosen database playlist to a .m3u file"""
|
|
selected_items = [
|
|
item.text() for item in self.playlist_listWidget.selectedItems()
|
|
]
|
|
selected_db_ids = [self.item_dict[item] for item in selected_items]
|
|
relative_path = self.relative_path_input.text()
|
|
m3u_path = self.m3u_path_input.text()
|
|
if m3u_path == "" or m3u_path is None:
|
|
self.close()
|
|
return
|
|
for playlist in selected_db_ids:
|
|
try:
|
|
with DBA.DBAccess() as db:
|
|
db_paths = db.query(
|
|
"SELECT s.filepath FROM song_playlist as sp JOIN song as s ON s.id = sp.song_id WHERE sp.playlist_id = ?;",
|
|
(playlist,),
|
|
)[0]
|
|
except Exception as e:
|
|
logging.error(
|
|
f"ExportPlaylistWindow.py save() | could not retrieve playlist songs: {e}"
|
|
)
|
|
error_dialog = ErrorDialog(
|
|
"Could not get songs from this playlist. noooooo"
|
|
)
|
|
error_dialog.exec()
|
|
self.close()
|
|
# Gather playlist song paths
|
|
write_paths = []
|
|
if self.checkbox.isChecked:
|
|
# Relative paths
|
|
for song in db_paths:
|
|
artist, album = get_reorganize_vars(song)
|
|
write_path = os.path.join(
|
|
relative_path, artist, album, song.split("/")[-1]
|
|
)
|
|
write_paths.append(write_path)
|
|
else:
|
|
# Normal paths
|
|
for song in db_paths:
|
|
write_paths.append(song)
|
|
# Write playlist file
|
|
for line in write_paths:
|
|
pass
|
|
self.close()
|
|
|
|
def cancel(self) -> None:
|
|
self.close()
|
|
return
|