musicpom/components/ExportPlaylistWindow.py
tsi-billypom aea3a6cdc3 what
2024-08-16 11:16:55 -04:00

69 lines
1.9 KiB
Python

import logging
from PyQt5.QtWidgets import (
QDialog,
QHBoxLayout,
QLineEdit,
QLabel,
QPushButton,
QVBoxLayout,
QListWidget,
QListWidgetItem,
)
from PyQt5.QtGui import QFont
import DBA
class ExportPlaylistWindow(QDialog):
def __init__(self):
super(ExportPlaylistWindow, self).__init__()
self.setWindowTitle("Export playlist")
layout = QVBoxLayout()
# button_layout = QHBoxLayout()
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]
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("Export")
save_button.clicked.connect(self.save)
layout.addWidget(save_button)
self.setLayout(layout)
self.show()
def save(self) -> None:
"""Exports the chosen database playlist to a .m3u file"""
value = self.input.text()
if value == "" or value is None:
self.close()
return
else:
try:
with DBA.DBAccess() as db:
db.execute("INSERT INTO playlist (name) VALUES (?);", (value,))
except Exception as e:
logging.error(
f"CreatePlaylistWindow.py save() | Could not create playlist: {e}"
)
self.close()
def cancel(self) -> None:
self.close()
return