115 lines
4.5 KiB
Python
115 lines
4.5 KiB
Python
from modules import configloader
|
|
from modules import printer
|
|
from modules import utils
|
|
from PIL import Image
|
|
import pillow_avif
|
|
from ffmpeg import FFmpeg, FFmpegError
|
|
import os
|
|
|
|
|
|
def has_transparency(img):
|
|
if img.info.get("transparency", None) is not None:
|
|
return True
|
|
if img.mode == "P":
|
|
transparent = img.info.get("transparency", -1)
|
|
for _, index in img.getcolors():
|
|
if index == transparent:
|
|
return True
|
|
elif img.mode == "RGBA":
|
|
extrema = img.getextrema()
|
|
if extrema[3][0] < 255:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def compress_audio(folder, file, target_folder, extension):
|
|
bitrate = configloader.config['AUDIO']['BitRate']
|
|
|
|
printer.files(file, os.path.splitext(file)[0], extension, f"{bitrate}")
|
|
try:
|
|
(FFmpeg()
|
|
.input(f'{folder}/{file}')
|
|
.output(utils.check_duplicates(f'{target_folder}/{os.path.splitext(file)[0]}.{extension}'),
|
|
{"b:a": bitrate})
|
|
.execute()
|
|
)
|
|
except FFmpegError as e:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
utils.errors_count += 1
|
|
if not configloader.config['FFMPEG']['HideErrors']:
|
|
printer.error(f"File {file} can't be processed! Error: {e}")
|
|
return f'{target_folder}/{os.path.splitext(file)[0]}.{extension}'
|
|
|
|
|
|
def compress_video(folder, file, target_folder, extension):
|
|
if not configloader.config['VIDEO']['SkipVideo']:
|
|
codec = configloader.config['VIDEO']['Codec']
|
|
crf = configloader.config['VIDEO']['CRF']
|
|
|
|
printer.files(file, os.path.splitext(file)[0], extension, codec)
|
|
try:
|
|
(FFmpeg()
|
|
.input(f'{folder}/{file}')
|
|
.output(utils.check_duplicates(f'{target_folder}/{os.path.splitext(file)[0]}.{extension}'),
|
|
{"codec:v": codec, "v:b": 0}, crf=crf)
|
|
.execute()
|
|
)
|
|
except FFmpegError as e:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
utils.errors_count += 1
|
|
if not configloader.config['FFMPEG']['HideErrors']:
|
|
printer.error(f"File {file} can't be processed! Error: {e}")
|
|
else:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
return f'{target_folder}/{os.path.splitext(file)[0]}.{extension}'
|
|
|
|
|
|
def compress_image(folder, file, target_folder, extension):
|
|
quality = configloader.config['IMAGE']['Quality']
|
|
printer.files(file, os.path.splitext(file)[0], extension, f"{quality}%")
|
|
try:
|
|
image = Image.open(f'{folder}/{file}')
|
|
|
|
if (extension == "jpg" or extension == "jpeg" or
|
|
(extension == "webp" and not configloader.config['FFMPEG']['WebpRGBA'])):
|
|
if has_transparency(image):
|
|
printer.warning(f"{file} has transparency. Changing to fallback...")
|
|
extension = configloader.config['IMAGE']['FallBackExtension']
|
|
|
|
res_downscale = configloader.config['IMAGE']['ResDownScale']
|
|
if res_downscale != 1:
|
|
width, height = image.size
|
|
new_size = (int(width / res_downscale), int(height / res_downscale))
|
|
image = image.resize(new_size)
|
|
|
|
image.save(utils.check_duplicates(f"{target_folder}/{os.path.splitext(file)[0]}.{extension}"),
|
|
optimize=True,
|
|
lossless=configloader.config['IMAGE']['Lossless'],
|
|
quality=quality,
|
|
minimize_size=True)
|
|
except Exception as e:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
utils.errors_count += 1
|
|
if not configloader.config['FFMPEG']['HideErrors']:
|
|
printer.error(f"File {file} can't be processed! Error: {e}")
|
|
return f'{target_folder}/{os.path.splitext(file)[0]}.{extension}'
|
|
|
|
|
|
def compress(folder, file, target_folder):
|
|
if configloader.config["FFMPEG"]["ForceCompress"]:
|
|
printer.unknown_file(file)
|
|
try:
|
|
(FFmpeg()
|
|
.input(f'{folder}/{file}')
|
|
.output(f'{target_folder}/{file}')
|
|
.execute()
|
|
)
|
|
except FFmpegError as e:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
utils.errors_count += 1
|
|
if not configloader.config['FFMPEG']['HideErrors']:
|
|
printer.error(f"File {file} can't be processed! Error: {e}")
|
|
else:
|
|
utils.add_unprocessed_file(f'{folder}/{file}', f'{target_folder}/{file}')
|
|
return f'{target_folder}/{file}'
|