Taiko_Stuff/TaikoPythonTools
1
0
Fork 0
mirror of https://github.com/cainan-c/TaikoPythonTools.git synced 2024-11-27 17:00:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update gui.py

Browse Source 
Fix bug caused by removing "unique id" column
This commit is contained in:
Cainan 2024-06-25 00:34:19 +01:00
parent 7fb2ba0cb9
commit b9175a1f29
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
TaikoSongConversionTool/gui.py
View File

@ -179,23 +179,22 @@ def toggle_checkbox(event):
selected_items = tree.selection() selected_items = tree.selection()
for item_id in selected_items: for item_id in selected_items:
values = list(tree.item(item_id, "values")) values = list(tree.item(item_id, "values"))
song_id = values[2] # Assume the song ID is in the third column song_id = values[1]
if values[0] == "☐": if values[0] == "☐":
values[0] = "☑" values[0] = "☑"
if song_id not in selected_song_ids: if song_id not in selected_song_ids:
selected_song_ids.append(song_id) selected_song_ids.append(song_id)
selection_count.set(selection_count.get() + 1) # Increment selection count selection_count.set(selection_count.get() + 1)
else: else:
values[0] = "☐" values[0] = "☐"
if song_id in selected_song_ids: if song_id in selected_song_ids:
selected_song_ids.remove(song_id) selected_song_ids.remove(song_id)
selection_count.set(selection_count.get() - 1) # Decrement selection count selection_count.set(selection_count.get() - 1)
tree.item(item_id, values=values) tree.item(item_id, values=values)
update_selection_count() # Call update_selection_count to update the count instantly update_selection_count()
return "break" return "break"
def filter_treeview(): def filter_treeview():
@ -386,30 +385,25 @@ def preview_audio(song_id):
custom_song_filename = os.path.join(custom_data_dir, "sound", f"song_{song_id}.mp3") custom_song_filename = os.path.join(custom_data_dir, "sound", f"song_{song_id}.mp3")
subprocess.run(["ffplay", "-autoexit", "-ss", f"{custom_preview_pos / 1000}", custom_song_filename]) subprocess.run(["ffplay", "-autoexit", "-ss", f"{custom_preview_pos / 1000}", custom_song_filename])
def get_preview_pos(song_id): def get_preview_pos(song_id):
# Load previewpos data from the default file
with open(previewpos_path, "r", encoding="utf-8") as previewpos_file: with open(previewpos_path, "r", encoding="utf-8") as previewpos_file:
previewpos_data = json.load(previewpos_file) previewpos_data = json.load(previewpos_file)
for item in previewpos_data: for item in previewpos_data:
if item["id"] == song_id: if item["id"] == song_id:
return item["previewPos"] return item["previewPos"]
# If use_custom is True, also try to load from the custom file
if custom_songs: if custom_songs:
with open(custom_previewpos_path, "r", encoding="utf-8") as custom_previewpos_file: with open(custom_previewpos_path, "r", encoding="utf-8") as custom_previewpos_file:
custom_previewpos_data = json.load(custom_previewpos_file) custom_previewpos_data = json.load(custom_previewpos_file)
for item in custom_previewpos_data: for item in custom_previewpos_data:
if item["id"] == song_id: if item["id"] == song_id:
return item["previewPos"] return item["previewPos"]
return None return None
def preview_selected(): def preview_selected():
selected_item = tree.selection() selected_item = tree.selection()
if selected_item: if selected_item:
song_id = tree.item(selected_item[0])["values"][2] song_id = tree.item(selected_item[0])["values"][1] # Ensure this points to the correct column for song ID
preview_audio(song_id) preview_audio(song_id)
def merge_ptb(): def merge_ptb():
@ -634,7 +628,7 @@ def export_data():
# Copy fumen folders for selected songs to output directory # Copy fumen folders for selected songs to output directory
for item_id in selected_items: for item_id in selected_items:
song_id = tree.item(item_id)["values"][2] song_id = tree.item(item_id)["values"][1]
fumen_folder_path = os.path.join(data_dir, "fumen", str(song_id)) fumen_folder_path = os.path.join(data_dir, "fumen", str(song_id))
if os.path.exists(fumen_folder_path): if os.path.exists(fumen_folder_path):
shutil.copytree(fumen_folder_path, os.path.join(fumen_output_dir, f"{song_id}")) shutil.copytree(fumen_folder_path, os.path.join(fumen_output_dir, f"{song_id}"))
@ -644,7 +638,7 @@ def export_data():
if custom_songs: if custom_songs:
for item_id in selected_items: for item_id in selected_items:
song_id = tree.item(item_id)["values"][2] song_id = tree.item(item_id)["values"][1]
custom_fumen_folder_path = os.path.join(custom_data_dir, "fumen", str(song_id)) custom_fumen_folder_path = os.path.join(custom_data_dir, "fumen", str(song_id))
if os.path.exists(custom_fumen_folder_path): if os.path.exists(custom_fumen_folder_path):
shutil.copytree(custom_fumen_folder_path, os.path.join(fumen_output_dir, f"{song_id}")) shutil.copytree(custom_fumen_folder_path, os.path.join(fumen_output_dir, f"{song_id}"))
@ -652,7 +646,7 @@ def export_data():
song_info = next((item for item in custom_music_info["items"] if item["id"] == song_id), None) song_info = next((item for item in custom_music_info["items"] if item["id"] == song_id), None)
for item_id in selected_items: for item_id in selected_items:
song_id = tree.item(item_id)["values"][2] song_id = tree.item(item_id)["values"][1]
if custom_songs: if custom_songs:
combined_items = custom_music_info["items"] + music_info["items"] combined_items = custom_music_info["items"] + music_info["items"]
else: else: