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

FFMpeg-Compressor: Refactor and bar fix

Browse source
This commit is contained in:
OleSTEEP 2024-05-05 03:14:27 +03:00
parent 9612d30d06
commit e71cad4d5f
4 changed files with 87 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

74
FFMpeg-Compressor/main.py
View file

@ -1,6 +1,5 @@
#!/usr/bin/env python3
from modules import configloader
from modules import compressor
from modules import printer
from modules import utils
@ -10,70 +9,39 @@ import sys
import os
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:
return "unknown"
def get_args():
try:
if sys.argv[1][len(sys.argv[1])-1] == "/":
path = sys.argv[1][:len(sys.argv[1])-1]
else:
path = sys.argv[1]
return path
except IndexError:
print(utils.help_message())
exit()
if __name__ == "__main__":
start_time = datetime.now()
printer.win_ascii_esc()
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()
req_folder = os.path.abspath(get_args())
orig_folder = os.path.abspath(arg_path)
printer.orig_folder = os.path.abspath(arg_path)
printer.bar_init(req_folder)
printer.bar_init(orig_folder)
if os.path.exists(f"{orig_folder}_compressed"):
shutil.rmtree(f"{orig_folder}_compressed")
if os.path.exists(f"{req_folder}_compressed"):
shutil.rmtree(f"{req_folder}_compressed")
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"))
for folder, folders, files in os.walk(req_folder):
if not os.path.exists(folder.replace(req_folder, f"{req_folder}_compressed")):
os.mkdir(folder.replace(req_folder, f"{req_folder}_compressed"))
printer.info(f"Compressing \"{folder.replace(orig_folder, orig_folder.split('/').pop())}\" folder...")
target_folder = folder.replace(orig_folder, f"{orig_folder}_compressed")
printer.info(f"Compressing \"{folder.replace(req_folder, req_folder.split('/').pop())}\" folder...")
target_folder = folder.replace(req_folder, f"{req_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)
compressor.compress_file(folder, file, target_folder, req_folder)
if configloader.config['FFMPEG']['MimicMode']:
try:
os.rename(comp_file, f'{folder}/{file}'.replace(orig_folder, f"{orig_folder}_compressed"))
except FileNotFoundError:
pass
utils.get_compression_status(orig_folder)
utils.get_compression_status(req_folder)
utils.sys_pause()
print(f"Time taken: {datetime.now() - start_time}")