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

FFMpeg-Compressor: Use ffmpeg-python and Pillow instead of ffmpeg cli

Browse source
This commit is contained in:
OleSTEEP 2023-11-16 01:26:23 +03:00
parent e48b7599d1
commit 37ff1f78b3
6 changed files with 126 additions and 125 deletions
Download patch file Download diff file Expand all files Collapse all files

99
FFMpeg-Compressor/main.py
View file

@ -8,53 +8,70 @@ import shutil
import sys
import os
try:
if sys.argv[1][len(sys.argv[1])-1] == "/":
arg_path = sys.argv[1][:len(sys.argv[1])-1]
def get_file_type(filename):
audio_ext = ['.aac', '.flac', '.m4a', '.mp3', '.ogg', '.opus', '.raw', '.wav', '.wma']
image_ext = ['.apng', '.avif', '.bmp', '.tga', '.tiff', '.dds', '.svg', '.webp', '.jpg', '.jpeg', '.png']
video_ext = ['.3gp' '.amv', '.avi', '.gif', '.m2t', '.m4v', '.mkv', '.mov', '.mp4', '.m4v', '.mpeg', '.mpv',
'.webm', '.ogv']
if os.path.splitext(filename)[1] in audio_ext:
return "audio"
elif os.path.splitext(filename)[1] in image_ext:
return "image"
elif os.path.splitext(filename)[1] in video_ext:
return "video"
else:
arg_path = sys.argv[1]
except IndexError:
print(utils.help_message())
exit()
return "unknown"
orig_folder = arg_path
printer.orig_folder = arg_path
printer.bar_init(orig_folder)
if __name__ == "__main__":
try:
if sys.argv[1][len(sys.argv[1])-1] == "/":
arg_path = sys.argv[1][:len(sys.argv[1])-1]
else:
arg_path = sys.argv[1]
except IndexError:
print(utils.help_message())
exit()
if os.path.exists(f"{orig_folder}_compressed"):
shutil.rmtree(f"{orig_folder}_compressed")
orig_folder = arg_path
printer.orig_folder = arg_path
printer.info("Creating folders...")
for folder, folders, files in os.walk(orig_folder):
if not os.path.exists(folder.replace(orig_folder, f"{orig_folder}_compressed")):
os.mkdir(folder.replace(orig_folder, f"{orig_folder}_compressed"))
printer.bar_init(orig_folder)
printer.info(f"Compressing \"{folder.replace(orig_folder, orig_folder.split('/').pop())}\" folder...")
target_folder = folder.replace(orig_folder, f"{orig_folder}_compressed")
for file in os.listdir(folder):
if os.path.isfile(f'{folder}/{file}'):
match compressor.get_file_type(file):
case "audio":
comp_file = compressor.compress_audio(folder, file, target_folder)
case "image":
comp_file = compressor.compress_image(folder, file, target_folder)
case "video":
comp_file = compressor.compress_video(folder, file, target_folder)
case "unknown":
comp_file = compressor.compress(folder, file, target_folder)
if os.path.exists(f"{orig_folder}_compressed"):
shutil.rmtree(f"{orig_folder}_compressed")
utils.check_file_existing(folder.replace(orig_folder, f"{orig_folder}_compressed"), file)
printer.info("Creating folders...")
for folder, folders, files in os.walk(orig_folder):
if not os.path.exists(folder.replace(orig_folder, f"{orig_folder}_compressed")):
os.mkdir(folder.replace(orig_folder, f"{orig_folder}_compressed"))
if configloader.config['FFMPEG']['MimicMode']:
try:
os.rename(comp_file, f'{folder}/{file}'.replace(orig_folder, f"{orig_folder}_compressed"))
except FileNotFoundError:
if not configloader.config['FFMPEG']['HideErrors']:
printer.error(f"File {file} can't be processed! Maybe it is ffmpeg error or unsupported file. "
f"You can change -loglevel in ffmpeg parameters to see full error.")
printer.info(f"Compressing \"{folder.replace(orig_folder, orig_folder.split('/').pop())}\" folder...")
target_folder = folder.replace(orig_folder, f"{orig_folder}_compressed")
for file in os.listdir(folder):
if os.path.isfile(f'{folder}/{file}'):
match get_file_type(file):
case "audio":
comp_file = compressor.compress_audio(folder, file, target_folder,
configloader.config['AUDIO']['Extension'])
case "image":
comp_file = compressor.compress_image(folder, file, target_folder,
configloader.config['IMAGE']['Extension'])
case "video":
comp_file = compressor.compress_video(folder, file, target_folder,
configloader.config['VIDEO']['Extension'])
case "unknown":
comp_file = compressor.compress(folder, file, target_folder)
if configloader.config['FFMPEG']['CopyUnprocessed']:
printer.info("Copying unprocessed files...")
utils.add_unprocessed_files(orig_folder)
utils.get_compression_status(orig_folder)
if configloader.config['FFMPEG']['MimicMode']:
try:
os.rename(comp_file, f'{folder}/{file}'.replace(orig_folder, f"{orig_folder}_compressed"))
except FileNotFoundError:
pass
if configloader.config['FFMPEG']['CopyUnprocessed']:
printer.info("Copying unprocessed files...")
utils.add_unprocessed_files(orig_folder)
utils.get_compression_status(orig_folder)