mirror of
https://github.com/tanrax/guetzli-recursively.git
synced 2024-09-19 17:27:21 +02:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# Libraries
|
|
import click
|
|
from os import path, walk, remove, rename
|
|
from imghdr import what
|
|
from subprocess import call
|
|
|
|
# Variables
|
|
TEMP_FILE = 'temp.jpg'
|
|
TYPES = ('jpeg',)
|
|
LIMIT_QUALITY = 84
|
|
|
|
|
|
@click.command()
|
|
@click.option(
|
|
'--quality',
|
|
default=100,
|
|
help='Quality >= {quality} [default 100].'.format(
|
|
quality=LIMIT_QUALITY
|
|
)
|
|
)
|
|
@click.option(
|
|
'--memlimit',
|
|
default=28000,
|
|
help='Memory limit in MB. Guetzli will fail if unable to stay under the limit. Default is 28000 MB'
|
|
)
|
|
@click.argument('folder', type=click.Path(exists=True))
|
|
def run(quality, memlimit, folder):
|
|
for dirpath, dirnames, files in walk(folder):
|
|
for name in files:
|
|
url = path.join(dirpath, name)
|
|
# Check type
|
|
if what(url) in TYPES or quality >= LIMIT_QUALITY:
|
|
# Get urls
|
|
click.echo(url)
|
|
url_out = path.join(folder, TEMP_FILE)
|
|
# Remove temp image
|
|
try:
|
|
remove(url_out)
|
|
except:
|
|
pass
|
|
# Execute guetzli
|
|
args = ['guetzli', '--quality',
|
|
str(quality), '--memlimit', str(memlimit), url, url_out]
|
|
print(' '.join(args))
|
|
call(args)
|
|
# Print your have saved
|
|
size_source = path.getsize(url)
|
|
try:
|
|
size_out = path.getsize(url_out)
|
|
except:
|
|
size_out = size_source
|
|
size_acurate = 100 * size_out / size_source
|
|
# Check if it is cost effective to replace it
|
|
if size_acurate < 100:
|
|
# Remove source
|
|
try:
|
|
remove(url)
|
|
except:
|
|
pass
|
|
# Move temp to source
|
|
rename(url_out, url)
|
|
click.echo(
|
|
'Save ' + str(round(100 - size_acurate, 2)) + '%')
|
|
else:
|
|
click.echo('It is not necessary to optimize')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|