mew/ui/qt/dlg_meta_editor.py

import os

from PyQt5 import QtWidgets
from PyQt5.QtGui import QIntValidator
from path import Path

from core.repositories import TrackRepository
from ui.qt.dlg_meta_editor_ui import Ui_dlgMetaEditor


class DlgMetaEditor(QtWidgets.QDialog):
    def __init__(self, parent=None, track=None):
        super().__init__(parent)
        self.track = track
        self.createWidgets()

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

        self.ui.lineTitle.setText(self.track.title or "")
        self.ui.lineArtist.setText(self.track.artist or "")
        self.ui.lineAlbum.setText(self.track.album or "")
        self.ui.lineTrackNum.setText(str(self.track.track_num) or "")

        self.ui.lineTrackNum.setValidator(QIntValidator())

        self.ui.lblPath.setText(self.track.path)

        self.ui.btnOpenTrackFolder.clicked.connect(self.openTrackFolder)

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

    def openTrackFolder(self):
        folder = Path(self.track.path)
        os.startfile(folder)

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

    def ok(self):
        track_repo = TrackRepository()
        self.track.title = self.ui.lineTitle.text() or None
        self.track.artist = self.ui.lineArtist.text() or None
        self.track.album = self.ui.lineAlbum.text() or None
        self.track.track_num = int(self.ui.lineTrackNum.text()) or None
        track_repo.commit()
        self.done(1)

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