generate_smallvariants.py

#!/usr/bin/python3

import os

from PIL import Image

path = os.path.dirname(os.path.realpath(__file__)) + "/"

resources = ["res/drawable-nodpi"]

def generate_smallvariants(resource):
    global path
    wallpapers_path = path + resource + "/"
    clean(wallpapers_path)
    wallpapers = os.listdir(wallpapers_path)

    for wallpaper in wallpapers:
        # Append _small.jpg to the wallpaper
        wallpaper_small = wallpaper.split(".")[0] + "_small.png"
        wallpaper_small_path = wallpapers_path + wallpaper_small

        # Save the wallpaper with 1/5 size to wallpaper_small_path
        with Image.open(wallpapers_path + wallpaper) as img:
            small_width = img.width / 5
            small_height = img.height / 5

            size = int(small_width), int(small_height)

            img_small = img.resize(size, Image.Resampling.LANCZOS)
            img_small.save(wallpaper_small_path)

def clean(wallpapers_path):
    wallpapers = os.listdir(wallpapers_path)

    for wallpaper in wallpapers:
        # Get rid of existing small variants
        if wallpaper.endswith("_small.png"):
            os.remove(wallpapers_path + wallpaper)

for resource in resources:
    generate_smallvariants(resource)