Deepseek实现本地电影文件批量重命名为infuse格式，可匹配IMDB

import os
from openai import OpenAI# 初始化DeepSeek客户端
client = OpenAI(api_key="<DeepSeek API Key>", base_url="https://api.deepseek.com")def parse_filename_with_deepseek(filename):"""使用DeepSeek API解析文件名并生成INFUSE命名"""# 构建提示词prompt = (f"请根据以下文件名结合IMDB生成符合INFUSE规范的命名，要求包含标题、年份、分辨率、版本（如果有）、音频和字幕信息。"f"文件名: {filename}\n"f"INFUSE命名格式示例: 《标题》(年份) [分辨率-编码类型][音频格式][字幕].扩展名\n"f"请直接返回INFUSE命名，无需额外解释。")# 调用DeepSeek APIresponse = client.chat.completions.create(model="deepseek-chat",messages=[{"role": "system", "content": "你是一个电影命名专家，擅长根据文件名生成符合INFUSE规范的命名。"},{"role": "user", "content": prompt},],stream=False)# 返回生成的命名return response.choices[0].message.content.strip()def batch_rename_files(directory):"""批量重命名文件夹中的文件"""# 支持的视频文件扩展名supported_extensions = (".mp4", ".mkv", ".avi", ".mov", ".flv", ".wmv")# 遍历文件夹中的文件for filename in os.listdir(directory):# 检查文件扩展名if filename.lower().endswith(supported_extensions):file_path = os.path.join(directory, filename)try:# 生成新的INFUSE命名new_filename = parse_filename_with_deepseek(filename)new_file_path = os.path.join(directory, new_filename)# 重命名文件os.rename(file_path, new_file_path)print(f"重命名成功: {filename} -> {new_filename}")except Exception as e:print(f"重命名失败: {filename} ({e})")# 示例文件夹路径
directory = "/path/to/your/folder"  # 替换为你的文件夹路径# 执行批量重命名
if os.path.isdir(directory):batch_rename_files(directory)
else:print(f"错误: 文件夹路径无效 {directory}")