ChatGPT写的字幕下载代码

为了自动给自己nas上的电影和美剧下字幕，测试了以下让chatgpt编写代码。经过不断提出要求，并针对运行的错误反馈过去，最后历经半天，终于写出来一个基本能用的字幕下载代码。总体感觉来说，他写的代码，并不是特别好读。在修改过程中，他也是经常不知道所以然在乱改。是否能改正确，取决于你交互指出问题的方式。而是否能发现并指出某些问题，需要能读懂代码。所以现阶段，至少chatgpt还没有到能让普通人变成程序员的可能。但是其确实能提高码农生产力，特别是技术过硬的，一眼能看出问题，指导低级码农的那种人。

特别说明，这个代码可能存在某些bug，我没有一行的审查，并不确保能真正用来下载字幕。
需要支持库的安装

pip install subliminal
pip install chardet

使用方法

sudo python3 getsub.py ./

代码如下：

import os
import sys
import glob
from subliminal import download_best_subtitles, region, Video
from subliminal import save_subtitles

def load_downloaded_files(downloaded_files_path):
    if os.path.exists(downloaded_files_path):
        with open(downloaded_files_path, 'r') as f:
            return set(line.strip() for line in f)
    else:
        return set()

def save_downloaded_file(downloaded_files_path, file_path):
    with open(downloaded_files_path, 'a') as f:
        f.write(f"{file_path}\n")

def get_subtitle_name_with_index(subtitle_filename, index):
    return f"{subtitle_filename[:-4]}_{index}.srt"

from subliminal import save_subtitles

def save_subtitle(subtitle, subtitle_filename):
    new_subtitle_content = subtitle.content
    detected_encoding = chardet.detect(new_subtitle_content)['encoding']
    new_subtitle_text = new_subtitle_content.decode(detected_encoding, errors='ignore')

    if os.path.exists(subtitle_filename):
        with open(subtitle_filename, 'r', encoding=detected_encoding) as f:
            existing_subtitle = f.read()
        
        if existing_subtitle == new_subtitle_text:
            print(f"Skipping {subtitle_filename} (already exists)")
            return
        else:
            index = 1
            while os.path.exists(get_subtitle_name_with_index(subtitle_filename, index)):
                index += 1
            subtitle_filename = get_subtitle_name_with_index(subtitle_filename, index)
    
    with open(subtitle_filename, "w", encoding=detected_encoding) as f:
        f.write(new_subtitle_text)
    print(f"Downloaded subtitle saved as {subtitle_filename}")

def download_subtitles(folder_path, languages, downloaded_files_path):
    region.configure("dogpile.cache.dbm", arguments={"filename": "cachefile.dbm"})
    video_exts = ["*.mkv", "*.mp4", "*.avi", "*.mov", "*.wmv"]

    video_files = []
    for video_ext in video_exts:
        all_files = glob.glob(os.path.join(folder_path, f"**/{video_ext}"), recursive=True)
        valid_files = [file for file in all_files if "@eaDir" not in file]
        video_files.extend(valid_files)

    downloaded_files = load_downloaded_files(downloaded_files_path)

    total_files = len(video_files)
    processed_files = 0

    for video_file in video_files:
     
        if video_file in downloaded_files:
            print(f"Skipping {video_file} (already downloaded)")
            continue

        try:
            video = Video.fromname(video_file)
            print(f"Downloading subtitles for {video_file}")
            best_subtitles = download_best_subtitles([video], languages)
            downloaded_subtitles = False
            for subtitle in best_subtitles[video]:
                file_name, file_extension = os.path.splitext(video_file)
                subtitle_filename = f"{file_name}.{subtitle.language.alpha3}.srt"
                save_subtitle(subtitle, subtitle_filename)
                downloaded_subtitles = True

            if not downloaded_subtitles:
                print(f"No subtitles found for {video_file}")
            else:
                save_downloaded_file(downloaded_files_path, video_file)
        except Exception as e:
            print(f"Error processing {video_file}: {e}")

        processed_files += 1
        progress = (processed_files / total_files) * 100
        print(f"Progress: {processed_files}/{total_files} ({progress:.2f}%)")

if __name__ == '__main__':
    if len(sys.argv) < 2:
        folder_path = "./"
    else:
        folder_path = sys.argv[1]       
    languages = {"zho", "eng"}
    downloaded_files_path = "downloaded_files.txt"
    download_subtitles(folder_path, languages, downloaded_files_path)