75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
from modules import configloader
|
|
from modules import printer
|
|
from shutil import copyfile
|
|
from glob import glob
|
|
import os
|
|
|
|
errors_count = 0
|
|
|
|
|
|
def get_dir_size(directory, files):
|
|
total_size = 0
|
|
for folder, folders, files in os.walk(directory):
|
|
for file in files:
|
|
if not os.path.islink(f"{folder}/{file}"):
|
|
total_size += os.path.getsize(f"{folder}/{file}")
|
|
return total_size
|
|
|
|
|
|
def get_compression(orig, comp):
|
|
processed_files = []
|
|
for folder, folders, files in os.walk(comp):
|
|
for file in files:
|
|
processed_files.append(file)
|
|
try:
|
|
orig = get_dir_size(orig, processed_files)
|
|
comp = get_dir_size(comp, processed_files)
|
|
|
|
print(f"Result: {orig/1024/1024:.2f}MB -> {comp/1024/1024:.2f}MB Δ {(orig - comp)/1024/1024:.2f}MB")
|
|
except ZeroDivisionError:
|
|
printer.warning("Nothing compressed!")
|
|
|
|
|
|
def get_compression_status(orig_folder):
|
|
orig_folder_len = 0
|
|
comp_folder_len = 0
|
|
|
|
for folder, folders, files in os.walk(orig_folder):
|
|
orig_folder_len += len(files)
|
|
|
|
for folder, folders, files in os.walk(f'{orig_folder}_compressed'):
|
|
for file in files:
|
|
if not os.path.splitext(file)[1].count(" (copy)"):
|
|
comp_folder_len += 1
|
|
|
|
if errors_count != 0:
|
|
printer.warning("Some files failed to compress!")
|
|
|
|
if orig_folder_len == comp_folder_len:
|
|
printer.info("Success!")
|
|
get_compression(orig_folder, f"{orig_folder}_compressed")
|
|
else:
|
|
printer.warning("Original and compressed folders are not identical!")
|
|
get_compression(orig_folder, f"{orig_folder}_compressed")
|
|
|
|
|
|
def add_unprocessed_files(orig_folder):
|
|
for folder, folders, files in os.walk(orig_folder):
|
|
for file in files:
|
|
new_folder = f"{folder}".replace(orig_folder, f"{orig_folder}_compressed")
|
|
if len(glob(f"{folder}/{os.path.splitext(file)[0]}.*")) > 1:
|
|
if len(glob(f"{new_folder}/{file}")):
|
|
copyfile(f"{folder}/{file}", f"{new_folder}/{file} (copy)")
|
|
printer.warning(
|
|
f'Duplicate file has been found! Check manually this files - "{file}", "{file} (copy)"')
|
|
else:
|
|
copyfile(f"{folder}/{file}", f"{new_folder}/{file}")
|
|
printer.info(f"File {file} copied to compressed folder.")
|
|
else:
|
|
if not len(glob(f"{new_folder}/{os.path.splitext(file)[0]}.*")):
|
|
copyfile(f"{folder}/{file}", f"{new_folder}/{file}")
|
|
printer.info(f"File {file} copied to compressed folder.")
|
|
|
|
|
|
def help_message():
|
|
return "Usage: ffmpeg-comp {folder}"
|