freckle 发表于 2024-9-10 21:02

用Fitten Code Chat 自动写了个修改CUE文件的python代码

本帖最后由 wushaominkk 于 2024-9-12 08:53 编辑

这个Fitten Code Chat是pycharm里的一个插件,官网https://code.fittentech.com/?ref=gzl
https://s1.locimg.com/2024/09/10/55ddef0496d48.png

诉求,如上图,从网上下的歌曲,其CUE因抓轨的人没写歌名,但提供了歌名,需要自己修改。
程序成功了一半,使用“浏览”按键可以获取正确的文件。使用拖拽获取时,会有个{ },试了几次,弄不好,请高手帮忙修改下{:1_893:}
https://s1.locimg.com/2024/09/10/e2b829052b413.png
import tkinter as tk
from tkinter import filedialog, messagebox
from tkinterdnd2 import TkinterDnD, DND_FILES


def read_file(file_path):
    # 尝试使用不同的编码读取文件
    for encoding in ['utf-8', 'ansi']:
      try:
            with open(file_path, 'r', encoding=encoding) as file:
                return
      except (UnicodeDecodeError, LookupError):
            continue
    raise ValueError("无法读取文件,文件编码不受支持")


def replace_track_names(cue_content, song_names):
    for i, song_name in enumerate(song_names, start=1):
      old_track = f"音轨{i:02}"
      new_track = song_name
      cue_content = cue_content.replace(old_track, new_track)
    return cue_content


def process_files():
    song_file_path = song_file_entry.get()
    cue_file_path = cue_file_entry.get()

    if not song_file_path or not cue_file_path:
      messagebox.showerror("错误", "请选择文件")
      return

    try:
      song_names = read_file(song_file_path)

      # 尝试以不同编码读取CUE文件
      cue_content = None
      for encoding in ['utf-8', 'ansi']:
            try:
                with open(cue_file_path, 'r', encoding=encoding) as cue_file:
                  cue_content = cue_file.read()
                  break
            except (UnicodeDecodeError, LookupError):
                continue

      if cue_content is None:
            raise ValueError("无法读取CUE文件,文件编码不受支持")

      new_cue_content = replace_track_names(cue_content, song_names)

      with open(cue_file_path, 'w', encoding='utf-8') as cue_file:
            cue_file.write(new_cue_content)

      messagebox.showinfo("成功", "文件处理完成")
    except Exception as e:
      messagebox.showerror("错误", f"处理文件时出错: {e}")


def browse_song_file():
    file_path = filedialog.askopenfilename(filetypes=[("Text files", "*.txt")])
    song_file_entry.delete(0, tk.END)
    song_file_entry.insert(0, file_path)


def browse_cue_file():
    file_path = filedialog.askopenfilename(filetypes=[("CUE files", "*.cue")])
    cue_file_entry.delete(0, tk.END)
    cue_file_entry.insert(0, file_path)


def drop_song_file(event):
    song_file_entry.delete(0, tk.END)
    song_file_entry.insert(0, event.data)


def drop_cue_file(event):
    cue_file_entry.delete(0, tk.END)
    cue_file_entry.insert(0, event.data)


# 创建主窗口
root = TkinterDnD.Tk()
root.title("CUE文件音轨替换工具")

# 创建并放置标签和输入框
tk.Label(root, text="歌曲名列表文件:").grid(row=0, column=0, padx=10, pady=10)
song_file_entry = tk.Entry(root, width=50)
song_file_entry.grid(row=0, column=1, padx=10, pady=10)
song_file_entry.drop_target_register(DND_FILES)
song_file_entry.dnd_bind('<<Drop>>', drop_song_file)
tk.Button(root, text="浏览", command=browse_song_file).grid(row=0, column=2, padx=10, pady=10)

tk.Label(root, text="CUE文件:").grid(row=1, column=0, padx=10, pady=10)
cue_file_entry = tk.Entry(root, width=50)
cue_file_entry.grid(row=1, column=1, padx=10, pady=10)
cue_file_entry.drop_target_register(DND_FILES)
cue_file_entry.dnd_bind('<<Drop>>', drop_cue_file)
tk.Button(root, text="浏览", command=browse_cue_file).grid(row=1, column=2, padx=10, pady=10)

# 创建并放置处理按钮
tk.Button(root, text="处理文件", command=process_files).grid(row=2, column=1, padx=10, pady=20)

# 运行主循环
root.mainloop()

''''''
''''''
''''''
'''
def clean_path(dropped_data):
    # 移除路径中可能存在的花括号
    return dropped_data.strip("{}")

def drop_song_file(event):
    song_file_entry.delete(0, tk.END)
    file_path = clean_path(event.data)
    song_file_entry.insert(0, file_path)

def drop_cue_file(event):
    cue_file_entry.delete(0, tk.END)
    file_path = clean_path(event.data)
    cue_file_entry.insert(0, file_path)

# 创建主窗口
root = TkinterDnD.Tk()
root.title("CUE文件音轨替换工具")

# 创建并放置标签和输入框
tk.Label(root, text="歌曲名列表文件:").grid(row=0, column=0, padx=10, pady=10)
song_file_entry = tk.Entry(root, width=50)
song_file_entry.grid(row=0, column=1, padx=10, pady=10)
song_file_entry.drop_target_register(DND_FILES)
song_file_entry.dnd_bind('<<Drop>>', drop_song_file)
tk.Button(root, text="浏览", command=browse_song_file).grid(row=0, column=2, padx=10, pady=10)

tk.Label(root, text="CUE文件:").grid(row=1, column=0, padx=10, pady=10)
cue_file_entry = tk.Entry(root, width=50)
cue_file_entry.grid(row=1, column=1, padx=10, pady=10)
cue_file_entry.drop_target_register(DND_FILES)
cue_file_entry.dnd_bind('<<Drop>>', drop_cue_file)
tk.Button(root, text="浏览", command=browse_cue_file).grid(row=1, column=2, padx=10, pady=10)

# 创建并放置处理按钮
tk.Button(root, text="处理文件", command=process_files).grid(row=2, column=1, padx=10, pady=20)

# 运行主循环
root.mainloop()
'''

freckle 发表于 2024-9-11 11:01

:lol我把它改好了,没有{}号了
import tkinter as tk
from tkinter import filedialog, messagebox
from tkinterdnd2 import TkinterDnD, DND_FILES


def read_file(file_path):
    # 尝试使用不同的编码读取文件
    for encoding in ['utf-8', 'ansi']:
      try:
            with open(file_path, 'r', encoding=encoding) as file:
                song_names = []
                for line in file:
                  # 移除行首的2位数字和".",再去除两端的空白字符
                  clean_name = line.strip()
                  song_names.append(clean_name)
                return song_names
      except (UnicodeDecodeError, LookupError):
            continue
    raise ValueError("无法读取文件,文件编码不受支持")



def replace_track_names(cue_content, song_names):
    for i, song_name in enumerate(song_names, start=1):
      old_track = f"音轨{i:02}"
      new_track = song_name
      cue_content = cue_content.replace(old_track, new_track)
    return cue_content


def process_files():
    song_file_path = song_file_entry.get()
    cue_file_path = cue_file_entry.get()

    if not song_file_path or not cue_file_path:
      messagebox.showerror("错误", "请选择文件")
      return

    try:
      song_names = read_file(song_file_path)

      # 尝试以不同编码读取CUE文件
      cue_content = None
      for encoding in ['utf-8', 'ansi']:
            try:
                with open(cue_file_path, 'r', encoding=encoding) as cue_file:
                  cue_content = cue_file.read()
                  break
            except (UnicodeDecodeError, LookupError):
                continue

      if cue_content is None:
            raise ValueError("无法读取CUE文件,文件编码不受支持")

      new_cue_content = replace_track_names(cue_content, song_names)

      with open(cue_file_path, 'w', encoding='utf-8') as cue_file:
            cue_file.write(new_cue_content)

      messagebox.showinfo("成功", "文件处理完成")
    except Exception as e:
      messagebox.showerror("错误", f"处理文件时出错: {e}")


def browse_song_file():
    file_path = filedialog.askopenfilename(filetypes=[("Text files", "*.txt")])
    song_file_entry.delete(0, tk.END)
    song_file_entry.insert(0, file_path)


def browse_cue_file():
    file_path = filedialog.askopenfilename(filetypes=[("CUE files", "*.cue")])
    cue_file_entry.delete(0, tk.END)
    cue_file_entry.insert(0, file_path)


def drop_song_file(event):
    file_path = event.data.strip('{}')# 删除路径中的花括号{}
    song_file_entry.delete(0, tk.END)
    song_file_entry.insert(0, file_path)


def drop_cue_file(event):
    file_path = event.data.strip('{}')# 删除路径中的花括号{}
    cue_file_entry.delete(0, tk.END)
    cue_file_entry.insert(0, file_path)



# 创建主窗口
root = TkinterDnD.Tk()
root.title("CUE文件音轨替换工具")

# 创建并放置标签和输入框
tk.Label(root, text="歌曲名列表文件:").grid(row=0, column=0, padx=10, pady=10)
song_file_entry = tk.Entry(root, width=50)
song_file_entry.grid(row=0, column=1, padx=10, pady=10)
song_file_entry.drop_target_register(DND_FILES)
song_file_entry.dnd_bind('<<Drop>>', drop_song_file)
tk.Button(root, text="浏览", command=browse_song_file).grid(row=0, column=2, padx=10, pady=10)

tk.Label(root, text="CUE文件:").grid(row=1, column=0, padx=10, pady=10)
cue_file_entry = tk.Entry(root, width=50)
cue_file_entry.grid(row=1, column=1, padx=10, pady=10)
cue_file_entry.drop_target_register(DND_FILES)
cue_file_entry.dnd_bind('<<Drop>>', drop_cue_file)
tk.Button(root, text="浏览", command=browse_cue_file).grid(row=1, column=2, padx=10, pady=10)

# 创建并放置处理按钮
tk.Button(root, text="处理文件", command=process_files).grid(row=2, column=1, padx=10, pady=20)

# 运行主循环
root.mainloop()

Nanoha123 发表于 2024-9-11 11:06

本帖最后由 Nanoha123 于 2024-9-11 11:13 编辑

虽然,之前叫别人也写了一个类似脚本,不过使用比较复杂而且 文本编码支持差点意思,有些特殊符号资料utf-8显示才正常

不知道是否支持多编码文本替换?

kaijinaini 发表于 2024-9-10 23:33

666666666666666666666

xixicoco 发表于 2024-9-10 23:51

ai写代码吗??

freckle 发表于 2024-9-11 08:16

xixicoco 发表于 2024-9-10 23:51
ai写代码吗??

应该是AI,我自己可不会写,看都看不懂,也就知道一点肤浅的基础

不苦小和尚 发表于 2024-9-11 08:22

免费的吗

freckle 发表于 2024-9-11 10:20

不苦小和尚 发表于 2024-9-11 08:22
免费的吗

应该是免费的吧,没看到要收费

freckle 发表于 2024-9-11 14:16

Nanoha123 发表于 2024-9-11 11:06
虽然,之前叫别人也写了一个类似脚本,不过使用比较复杂而且 文本编码支持差点意思,有些特殊符号资料utf-8 ...

'utf-8'和'ansi'两种,我遇到这2种,就让它写了这2种:loveliness:

Nanoha123 发表于 2024-9-11 20:50

freckle 发表于 2024-9-11 14:16
'utf-8'和'ansi'两种,我遇到这2种,就让它写了这2种

不知道,怎么封装?
大佬可以发一份可以运行的文件吗,表示不太会弄
页: [1] 2
查看完整版本: 用Fitten Code Chat 自动写了个修改CUE文件的python代码