76 lines
3 KiB
Python
Executable file
76 lines
3 KiB
Python
Executable file
#!env python3
|
|
|
|
from modules import configloader
|
|
from modules import compressor
|
|
from modules import printer
|
|
from modules import utils
|
|
import shutil
|
|
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"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
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()
|
|
|
|
orig_folder = os.path.abspath(arg_path)
|
|
printer.orig_folder = os.path.abspath(arg_path)
|
|
|
|
printer.bar_init(orig_folder)
|
|
|
|
if os.path.exists(f"{orig_folder}_compressed"):
|
|
shutil.rmtree(f"{orig_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"))
|
|
|
|
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']['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.sys_pause()
|