FFMpeg-Compressor: Switch config file to toml format
This commit is contained in:
parent
9ea9495266
commit
6a27699ccb
3 changed files with 18 additions and 12 deletions
|
@ -1,9 +0,0 @@
|
||||||
[FFMPEG]
|
|
||||||
AudioBitRate = 320k
|
|
||||||
AudioExt = mp3
|
|
||||||
CompLevel = 20
|
|
||||||
ImageExt = png
|
|
||||||
JpegComp = 3
|
|
||||||
FFmpegParams = -hide_banner -loglevel error
|
|
||||||
VideoCodec = libvpx-vp9
|
|
||||||
VideoExt = webm
|
|
9
FFMpeg-Compressor/config.toml
Normal file
9
FFMpeg-Compressor/config.toml
Normal file
|
@ -0,0 +1,9 @@
|
||||||
|
[FFMPEG]
|
||||||
|
AudioBitRate = "320k"
|
||||||
|
AudioExt = "mp3"
|
||||||
|
CompLevel = 20
|
||||||
|
ImageExt = "png"
|
||||||
|
JpegComp = 3
|
||||||
|
FFmpegParams = "-hide_banner -loglevel error"
|
||||||
|
VideoCodec = "libvpx-vp9"
|
||||||
|
VideoExt = "webm"
|
|
@ -1,14 +1,14 @@
|
||||||
from modules import printer
|
from modules import printer
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
import configparser
|
import tomllib
|
||||||
import os
|
import os
|
||||||
|
|
||||||
audio_exts = ['.aac', '.flac', '.m4a', '.mp3', '.ogg', '.opus', '.raw', '.wav', '.wma']
|
audio_exts = ['.aac', '.flac', '.m4a', '.mp3', '.ogg', '.opus', '.raw', '.wav', '.wma']
|
||||||
image_exts = ['.apng', '.avif', '.jfif', '.pjpeg', '.pjp', '.svg', '.webp', '.jpg', '.jpeg', '.png', '.raw']
|
image_exts = ['.apng', '.avif', '.jfif', '.pjpeg', '.pjp', '.svg', '.webp', '.jpg', '.jpeg', '.png', '.raw']
|
||||||
video_exts = ['.3gp' '.amv', '.avi', '.gif', '.m4v', '.mkv', '.mov', '.mp4', '.m4v', '.mpeg', '.mpv', '.webm', '.ogv']
|
video_exts = ['.3gp' '.amv', '.avi', '.gif', '.m4v', '.mkv', '.mov', '.mp4', '.m4v', '.mpeg', '.mpv', '.webm', '.ogv']
|
||||||
|
|
||||||
config = configparser.ConfigParser()
|
with open("config.toml", "rb") as f:
|
||||||
config.read("config.ini")
|
config = tomllib.load(f)
|
||||||
|
|
||||||
ffmpeg_params = config['FFMPEG']['FFmpegParams']
|
ffmpeg_params = config['FFMPEG']['FFmpegParams']
|
||||||
req_audio_ext = config['FFMPEG']['AudioExt']
|
req_audio_ext = config['FFMPEG']['AudioExt']
|
||||||
|
@ -36,19 +36,25 @@ def compress(folder):
|
||||||
files = len(os.listdir(path=folder))
|
files = len(os.listdir(path=folder))
|
||||||
progress = 0
|
progress = 0
|
||||||
for file in os.listdir(path=folder):
|
for file in os.listdir(path=folder):
|
||||||
|
|
||||||
if os.path.splitext(file)[1] in audio_exts:
|
if os.path.splitext(file)[1] in audio_exts:
|
||||||
|
|
||||||
bitrate = config['FFMPEG']['AudioBitRate']
|
bitrate = config['FFMPEG']['AudioBitRate']
|
||||||
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_audio_ext, f"{bitrate}bit/s")
|
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_audio_ext, f"{bitrate}bit/s")
|
||||||
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_audio_ext}'")
|
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_audio_ext}'")
|
||||||
|
|
||||||
elif os.path.splitext(file)[1] in image_exts:
|
elif os.path.splitext(file)[1] in image_exts:
|
||||||
|
|
||||||
if req_image_ext == "jpg" or req_image_ext == "jpeg":
|
if req_image_ext == "jpg" or req_image_ext == "jpeg":
|
||||||
|
|
||||||
if not has_transparency(Image.open(f'{folder}/{file}')):
|
if not has_transparency(Image.open(f'{folder}/{file}')):
|
||||||
jpg_comp = config['FFMPEG']['JpegComp']
|
jpg_comp = config['FFMPEG']['JpegComp']
|
||||||
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext,f"{jpg_comp}%")
|
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext,f"{jpg_comp}%")
|
||||||
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -q {jpg_comp} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_image_ext}'")
|
os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -q {jpg_comp} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_image_ext}'")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
printer.warning(f"{file} has transparency (.jpg not support it). Skipping...")
|
printer.warning(f"{file} has transparency (.jpg not support it). Skipping...")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
comp_level = config['FFMPEG']['CompLevel']
|
comp_level = config['FFMPEG']['CompLevel']
|
||||||
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext, f"{comp_level}%")
|
printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext, f"{comp_level}%")
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue