# pip3 install --upgrade tinify

import tinify
import os

tinify.key = "bMzDVQImTYCQa05xXmEO5JlYB_nVLGJd"


def optimize_scale(file, width):
    if ".png" in file:
        source = tinify.from_file(file)
        resized = source.resize(
            method="scale",
            width=width
        )
        resized.to_file(file)


def optimize(file):
    if ".png" in file:
        source = tinify.from_file(file)
        source.to_file(file)


print("################### CATEGORY ICONS")
for file in os.listdir("category_icons"):
    print(file)
    if 'estacao-' in file:
        optimize_scale("category_icons/{}".format(file), 300)
    else:
        optimize_scale("category_icons/{}".format(file), 128)

print("################### ART")
for file in os.listdir("assets"):
    print(file)
    optimize("assets/{}".format(file))