Youtube-caption-and-vedio/you_main.py at master · CodePythonFollow/Youtube-caption-and-vedio · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import os, re
from multiprocessing import Process, Pool
from en_zh_captiondownload import Youtube_language
from youtube_caption_deal import Deal_caption

def download_page(data):
    print(data)
    os.system(r"you-get " + r'https://www.youtube.com' + data)
    print(f'download page {data} success'.format(data))


def download_all_page():
    pool = Pool(processes=5)

    for data in datas:
        # p = Process(target=download_page, args=(data, ))
        pool.apply_async(download_page, args=(data,))
    pool.close()
    pool.join()


# 爬取字幕
def get_caption(datas):
    for data in datas:
        try:
            k_id= re.findall(r'watch.v=(.*?)&list', data)[0]
            Youtube_language(k_id).run()
        except Exception as e:
            print(e)
            continue

# 处理字幕文件
def deal_caption():
    dir_name = './'
    list_dir = os.listdir(dir_name)

    for file in list_dir:
        if '.srt' in file:
            deal_caption = Deal_caption()
            deal_caption.replace_srt(dir_name + file)
            print(f'{file}下载完成')


if __name__ == '__main__':
    # 传入数据，可以写成链接
    datas = [
        '/watch?v=xFciV6Ew5r4&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=1',
        '/watch?v=xqcTfplzr7c&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=2',
        '/watch?v=DjEuROpsvp4&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=3',
        '/watch?v=YJC6ldI3hWk&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=4',
        '/watch?v=NDFbXIiqT4o&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=5',
        '/watch?v=U2ZN104hIcc&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=6',
        '/watch?v=N5vscPTWKOk&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=7',
        '/watch?v=06I63_p-2A4&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=8',
        '/watch?v=-nh9rCzPJ20&list=PL-osiE80TeTt66h8cVpmbayBKlMTuS55y&index=9'
    ]

    # 下载所有视频
    download_all_page()

    # 删除默认下载的srt文件
    dir_files = os.listdir('./')
    for file in dir_files:
        if '.srt' in file and '.mp4' not in file:
            os.remove(file)

    # caption_download
    get_caption(datas)

    # deal_caption
    deal_caption()