mew/ui/qt/dlg_playlist.py

from datetime import datetime

from PyQt5 import QtWidgets

from core.models import Session
from core.repositories import SessionRepository
from ui.qt.dlg_playlist_ui import Ui_dlgPlaylist


class DlgPlaylist(QtWidgets.QDialog):
    def __init__(self, playlist=None, parent=None):
        super().__init__(parent)

        self.playlist = playlist if playlist is not None else Session()
        self.createWidgets()

    def createWidgets(self):
        self.ui = Ui_dlgPlaylist()
        self.ui.setupUi(self)

        self.ui.lineName.setText(self.playlist.name)
        if self.playlist.date:
            self.ui.lineDate.setDateTime(self.playlist.date)
        self.ui.textNotes.setText(self.playlist.notes or "")

        self.ui.btnSave.clicked.connect(self.save)
        self.ui.btnCancel.clicked.connect(self.cancel)

    @classmethod
    def edit(cls, parent, playlist=None):
        dlg = cls(playlist, parent)
        r = dlg.exec_()
        return r

    def cancel(self):
        self.done(0)

    def save(self):
        self.playlist.name = self.ui.lineName.text()
        if self.ui.lineDate.dateTime():
            self.playlist.date = self.ui.lineDate.dateTime().toPyDateTime()
        self.playlist.notes = self.ui.textNotes.toPlainText()

        repo = SessionRepository()
        if not self.playlist.id:
            repo.create(self.playlist)
        repo.commit()