neosubhamoy/pytubepp
1
1
Fork 0
mirror of https://github.com/neosubhamoy/pytubepp.git synced 2025-12-20 06:59:35 +05:30
Code Issues Wiki Activity

(refactor): improved caption track language tagging

Browse Source
This commit is contained in:
Subhamoy Biswas
2025-01-21 23:31:35 +05:30
parent ea2ad9b258
commit ec11629b07
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pytubepp/download.py
View File

@@ -1,6 +1,6 @@
from tqdm import tqdm from tqdm import tqdm
from .config import get_temporary_directory, load_config from .config import get_temporary_directory, load_config
from .utils import get_unique_filename, postprocess_cleanup from .utils import get_unique_filename, postprocess_cleanup, unpack_caption
import os, re, requests, shutil, sys, random, ffmpy import os, re, requests, shutil, sys, random, ffmpy
userConfig = load_config() userConfig = load_config()
@@ -21,6 +21,7 @@ def download_progressive(stream, itag, title, resolution, file_extention, captio
if caption_code: if caption_code:
print(f'Downloading Caption ({caption_code})...') print(f'Downloading Caption ({caption_code})...')
caption = captions[caption_code] caption = captions[caption_code]
_, caption_lang = unpack_caption(caption)
caption_file = os.path.join(tempDIR, random_filename + '_cap.srt') caption_file = os.path.join(tempDIR, random_filename + '_cap.srt')
caption.save_captions(caption_file) caption.save_captions(caption_file)
print('Processing...') print('Processing...')
@@ -28,7 +29,7 @@ def download_progressive(stream, itag, title, resolution, file_extention, captio
output_temp_file_with_subs = os.path.join(tempDIR, random_filename + '_merged.' + file_extention) output_temp_file_with_subs = os.path.join(tempDIR, random_filename + '_merged.' + file_extention)
ff = ffmpy.FFmpeg( ff = ffmpy.FFmpeg(
inputs={output_temp_file: None}, inputs={output_temp_file: None},
outputs={output_temp_file_with_subs: ['-i', caption_file, '-c', 'copy', '-c:s', 'mov_text', '-metadata:s:s:0', f'language={caption_code}', '-metadata:s:s:0', f'title={caption_code}', '-metadata:s:s:0', f'handler_name={caption_code}']} outputs={output_temp_file_with_subs: ['-i', caption_file, '-c', 'copy', '-c:s', 'mov_text', '-metadata:s:s:0', f'language={caption_code}', '-metadata:s:s:0', f'title={caption_lang}', '-metadata:s:s:0', f'handler_name={caption_lang}']}
) )
ff.run(stdout=devnull, stderr=devnull) ff.run(stdout=devnull, stderr=devnull)
devnull.close() devnull.close()

5
pytubepp/postprocess.py
View File

@@ -1,6 +1,6 @@
from mutagen.id3 import ID3, APIC, TIT2, TPE1, TALB from mutagen.id3 import ID3, APIC, TIT2, TPE1, TALB
from .config import get_temporary_directory, load_config from .config import get_temporary_directory, load_config
from .utils import get_unique_filename, postprocess_cleanup from .utils import get_unique_filename, postprocess_cleanup, unpack_caption
from .download import download_thumbnail from .download import download_thumbnail
import os, shutil, ffmpy import os, shutil, ffmpy
@@ -17,6 +17,7 @@ def merge_audio_video(title, resolution, file_extention, random_filename, captio
if caption_code: if caption_code:
print(f'Downloading Caption ({caption_code})...') print(f'Downloading Caption ({caption_code})...')
caption = captions[caption_code] caption = captions[caption_code]
_, caption_lang = unpack_caption(caption)
srt_file = os.path.join(tempDIR, random_filename + '_cap.srt') srt_file = os.path.join(tempDIR, random_filename + '_cap.srt')
caption.save_captions(srt_file) caption.save_captions(srt_file)
vtt_file = os.path.join(tempDIR, random_filename + '_cap.vtt') vtt_file = os.path.join(tempDIR, random_filename + '_cap.vtt')
@@ -38,7 +39,7 @@ def merge_audio_video(title, resolution, file_extention, random_filename, captio
input_params = {video_file: None, audio_file: None} input_params = {video_file: None, audio_file: None}
output_params = {output_temp_file: ['-i', subtitle_file, '-c:v', 'copy', '-c:a', 'copy', output_params = {output_temp_file: ['-i', subtitle_file, '-c:v', 'copy', '-c:a', 'copy',
'-c:s', subtitle_codec, '-metadata:s:s:0', f'language={caption_code}', '-c:s', subtitle_codec, '-metadata:s:s:0', f'language={caption_code}',
'-metadata:s:s:0', f'title={caption_code}', '-metadata:s:s:0', f'handler_name={caption_code}']} '-metadata:s:s:0', f'title={caption_lang}', '-metadata:s:s:0', f'handler_name={caption_lang}']}
devnull = open(os.devnull, 'w') devnull = open(os.devnull, 'w')
ff = ffmpy.FFmpeg(inputs=input_params, outputs=output_params) ff = ffmpy.FFmpeg(inputs=input_params, outputs=output_params)