149 lines
5.6 KiB
Python
149 lines
5.6 KiB
Python
from PyQt5.QtWidgets import (
|
|
QDialog,
|
|
QFrame,
|
|
QHBoxLayout,
|
|
QMessageBox,
|
|
QVBoxLayout,
|
|
QLabel,
|
|
QLineEdit,
|
|
QPushButton,
|
|
)
|
|
from PyQt5.QtGui import QFont
|
|
from components.ErrorDialog import ErrorDialog
|
|
from utils import set_tag, get_tags, update_song_in_database
|
|
# import re
|
|
|
|
|
|
class ID3LineEdit(QLineEdit):
|
|
def __init__(self, text: str, tag: str):
|
|
""" """
|
|
super(ID3LineEdit, self).__init__()
|
|
self.setText(text)
|
|
self._original_text = text
|
|
self.tag = tag
|
|
|
|
def has_changed(self) -> bool:
|
|
return self.text() != self._original_text
|
|
|
|
|
|
class MetadataWindow(QDialog):
|
|
def __init__(self, refreshMusicTableSignal, headers, songs: list, ids: list):
|
|
"""
|
|
Window that allows batch editing of metadata for multiple files
|
|
|
|
Input: songs, ids
|
|
- list of strings, absolute paths to mp3 files
|
|
- list of database song ids
|
|
|
|
"""
|
|
super(MetadataWindow, self).__init__()
|
|
self.refreshMusicTableSignal = refreshMusicTableSignal
|
|
self.songs = list(zip(songs, ids))
|
|
self.headers = headers
|
|
# Keep a dictionary of the input fields for save function
|
|
self.input_fields = {}
|
|
self.setWindowTitle("Edit metadata")
|
|
# self.setMinimumSize(600, 800)
|
|
layout = QVBoxLayout()
|
|
h_separator = QFrame()
|
|
h_separator.setFrameShape(QFrame.HLine)
|
|
|
|
# Labels and categories and stuff
|
|
category_label = QLabel("Edit metadata")
|
|
category_label.setFont(QFont("Sans", weight=QFont.Bold)) # bold category
|
|
category_label.setStyleSheet("text-transform:uppercase;") # uppercase category
|
|
layout.addWidget(category_label)
|
|
layout.addWidget(h_separator)
|
|
|
|
tag_sets: dict[str, list] = {}
|
|
# Get a dict of all tags for all songs
|
|
# e.g., { "TIT2": ["song_title1", "song_title2"], ... }
|
|
# e.g., { "title": ["song_title1", "song_title2"], ... }
|
|
for song in self.songs:
|
|
song_data = get_tags(song[0])[0]
|
|
if not song_data:
|
|
QMessageBox.error(
|
|
self,
|
|
"Error",
|
|
f"Could not retrieve ID3 tags for {song[0]}",
|
|
QMessageBox.Ok,
|
|
QMessageBox.Ok,
|
|
)
|
|
return
|
|
for key in self.headers.db_list:
|
|
if key not in self.headers.get_editable_db_list():
|
|
continue
|
|
tag = self.headers.db[key].frame_id
|
|
if tag is not None:
|
|
try:
|
|
_ = tag_sets[tag]
|
|
except KeyError:
|
|
# If a tag doesn't exist in our dict, create an empty list
|
|
tag_sets[tag] = []
|
|
try:
|
|
tag_sets[tag].append(song_data[tag].text[0])
|
|
except KeyError:
|
|
pass
|
|
|
|
# UI Creation
|
|
current_layout = QHBoxLayout()
|
|
for idx, (tag, value) in enumerate(tag_sets.items()):
|
|
if idx % 2 == 0:
|
|
# Make a new horizontal layout for every 2 items
|
|
layout.addLayout(current_layout)
|
|
current_layout = QHBoxLayout()
|
|
|
|
# Field Creation
|
|
if len(set(value)) <= 1:
|
|
# If the ID3 tag is the same for every item we're editing
|
|
field_text = str(value[0]) if value else ""
|
|
# Normal field
|
|
label = QLabel(str(self.headers.frame_id[tag].db))
|
|
input_field = ID3LineEdit(field_text, tag)
|
|
input_field.setStyleSheet(None)
|
|
else:
|
|
# Danger field
|
|
# this means the metadata differs between the selected items for this tag
|
|
# so be careful...dangerous
|
|
field_text = ""
|
|
label = QLabel(str(self.headers.frame_id[tag].db))
|
|
input_field = ID3LineEdit(field_text, tag)
|
|
input_field.setStyleSheet("border: 1px solid red")
|
|
# Save each input field to our dict for saving
|
|
self.input_fields[tag] = input_field
|
|
current_layout.addWidget(label)
|
|
current_layout.addWidget(input_field)
|
|
layout.addLayout(current_layout)
|
|
|
|
# Save button
|
|
save_button = QPushButton("Save")
|
|
save_button.clicked.connect(self.save)
|
|
layout.addWidget(save_button)
|
|
self.setLayout(layout)
|
|
|
|
def save(self):
|
|
"""Save changes made to metadata for each song in dict"""
|
|
for song in self.songs:
|
|
for tag, field in self.input_fields.items():
|
|
if field.text() is not None and field.text() != "":
|
|
if field.has_changed():
|
|
# Update the ID3 tag if the tag is not blank,
|
|
# and has been edited
|
|
success = set_tag(
|
|
filepath=song[0], db_column=tag, value=field.text()
|
|
)
|
|
if success:
|
|
update_song_in_database(
|
|
song[1],
|
|
edited_column_name=self.headers.frame_id[tag].db,
|
|
user_input_data=field.text(),
|
|
)
|
|
else:
|
|
error_dialog = ErrorDialog(
|
|
f"Could not update metadata for {song}. Exiting early"
|
|
)
|
|
error_dialog.exec()
|
|
self.close()
|
|
self.refreshMusicTableSignal.emit()
|
|
self.close()
|