VienDesuPorting/VNTools
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

FFMpeg-Compressor: Compress subfolders & Add progressbar

Browse source
This commit is contained in:
OleSTEEP 2023-08-01 17:20:57 +03:00
parent 250c4f789a
commit 1caebeb158
6 changed files with 63 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

26
FFMpeg-Compressor/modules/compressor.py
View file

@ -32,16 +32,16 @@ def has_transparency(img):
return False
def compress(folder):
def compress(root_folder, folder):
target_folder = folder.replace(root_folder, f"{root_folder}_compressed")
files = len(os.listdir(folder))
progress = 0
for file in os.listdir(folder):
if os.path.isfile(f'{folder}/{file}'):
if os.path.splitext(file)[1] in audio_exts:
bitrate = config['FFMPEG']['AudioBitRate']
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_audio_ext, f"{bitrate}bit/s")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_audio_ext}'")
printer.files(file, os.path.splitext(file)[0], req_audio_ext, f"{bitrate}bit/s")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{target_folder}/{os.path.splitext(file)[0]}.{req_audio_ext}'")
elif os.path.splitext(file)[1] in image_exts:
@ -49,25 +49,23 @@ def compress(folder):
if not has_transparency(Image.open(f'{folder}/{file}')):
jpg_comp = config['FFMPEG']['JpegComp']
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext, f"level {jpg_comp}")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -q {jpg_comp} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_image_ext}'")
printer.files(file, os.path.splitext(file)[0], req_image_ext, f"level {jpg_comp}")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -q {jpg_comp} '{target_folder}/{os.path.splitext(file)[0]}.{req_image_ext}'")
else:
printer.warning(f"{file} has transparency (.jpg not support it). Skipping...")
else:
comp_level = config['FFMPEG']['CompLevel']
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext, f"{comp_level}%")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -compression_level {comp_level} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_image_ext}'")
printer.files(file, os.path.splitext(file)[0], req_image_ext, f"{comp_level}%")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -compression_level {comp_level} '{target_folder}/{os.path.splitext(file)[0]}.{req_image_ext}'")
elif os.path.splitext(file)[1] in video_exts:
codec = config['FFMPEG']['VideoCodec']
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_video_ext, codec)
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -vcodec {codec} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_video_ext}'")
printer.files(file, os.path.splitext(file)[0], req_video_ext, codec)
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -vcodec {codec} '{target_folder}/{os.path.splitext(file)[0]}.{req_video_ext}'")
else:
printer.warning("File extension not recognized. This may affect the quality of the compression.")
print(f"\r[{int((progress/files) * 100)}%] \033[0;33m{file}\033[0m")
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{folder}_compressed/{file}'")
progress += 1
printer.unknown_file(file)
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{target_folder}/{file}'")