71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
from PyQt5.QtWidgets import (
|
|
QDialog,
|
|
QWidget,
|
|
QListWidget,
|
|
QFrame,
|
|
QVBoxLayout,
|
|
QLabel,
|
|
QLineEdit,
|
|
QPushButton,
|
|
QListWidgetItem,
|
|
)
|
|
from PyQt5.QtGui import QFont
|
|
import DBA
|
|
import logging
|
|
|
|
|
|
class AddToPlaylistWindow(QDialog):
|
|
def __init__(self, song_db_ids: list):
|
|
super(AddToPlaylistWindow, self).__init__()
|
|
self.song_db_ids = song_db_ids
|
|
self.setWindowTitle("Add songs to playlist:")
|
|
# self.setMinimumSize(400, 400)
|
|
layout = QVBoxLayout()
|
|
|
|
playlist_dict: dict = {}
|
|
with DBA.DBAccess() as db:
|
|
data = db.query("SELECT id, name from playlist;", ())
|
|
for row in data:
|
|
playlist_dict[row[0]] = row[1]
|
|
|
|
self.item_dict = {}
|
|
self.listWidget = QListWidget(self)
|
|
for i, (k, v) in enumerate(playlist_dict.items()):
|
|
item_text = f"{i} | {v}"
|
|
item = QListWidgetItem(item_text)
|
|
self.listWidget.addItem(item)
|
|
self.item_dict[item_text] = k
|
|
|
|
# add ui elements to window
|
|
label = QLabel("Playlists")
|
|
label.setFont(QFont("Sans", weight=QFont.Bold))
|
|
layout.addWidget(label)
|
|
layout.addWidget(self.listWidget)
|
|
|
|
# Save button
|
|
save_button = QPushButton("Add")
|
|
save_button.clicked.connect(self.save)
|
|
layout.addWidget(save_button)
|
|
self.setLayout(layout)
|
|
self.show()
|
|
|
|
def save(self):
|
|
selected_playlists = [item.text() for item in self.listWidget.selectedItems()]
|
|
selected_playlists_db_ids = [
|
|
self.item_dict[item] for item in selected_playlists
|
|
]
|
|
for playlist in selected_playlists_db_ids:
|
|
for song in self.song_db_ids:
|
|
try:
|
|
with DBA.DBAccess() as db:
|
|
db.execute(
|
|
"INSERT INTO song_playlist (playlist_id, song_id) VALUES (?, ?);",
|
|
(playlist, song),
|
|
)
|
|
except Exception as e:
|
|
logging.error(
|
|
f"AddToPlaylistWindow.py save() | could not insert song into playlist: {e}"
|
|
)
|
|
|
|
self.close()
|