FlightRisingUtils/format_images.py at master · TheFatRabbit/FlightRisingUtils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# used only when adding new images

import os
from PIL import Image

dirname = os.path.dirname(__file__)

# deletes .ini files and renames files with (1) suffix to the base name
for folder_path in os.listdir(os.path.join(dirname, "images")):
    if ".png" in folder_path:
        continue
    for subfolder_path in os.listdir(os.path.join(dirname, "images", folder_path)):
        for file_path in os.listdir(os.path.join(dirname, "images", folder_path, subfolder_path)):
            if ".ini" in file_path:
                os.remove(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))
                continue
            if " (1)" in file_path:
                if os.path.isfile(os.path.join(dirname, "images", folder_path, subfolder_path, file_path.replace(" (1)", ""))):
                    os.remove(os.path.join(dirname, "images", folder_path, subfolder_path, file_path.replace(" (1)", "")))
                os.rename(os.path.join(dirname, "images", folder_path, subfolder_path, file_path), os.path.join(dirname, "images", folder_path, subfolder_path, file_path.replace(" (1)", "")))

# replaces transparent parts of images with white
for folder_path in os.listdir(os.path.join(dirname, "images")):
    if ".png" in folder_path:
        continue
    for subfolder_path in os.listdir(os.path.join(dirname, "images", folder_path)):
        for file_path in os.listdir(os.path.join(dirname, "images", folder_path, subfolder_path)):
            image = Image.open(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))
            if image.mode == "RGBA":
                new_image = Image.new("RGB", image.size, (255, 255, 255))
                new_image.paste(image, (0, 0), image)
                new_image.save(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))
                image = Image.open(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))
            if image.size != (56, 56):
                new_image = image.resize((56, 56))
                new_image.save(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))
                image = Image.open(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))

# deletes any items that are already in the Universal folder
names = []
for subfolder_path in os.listdir(os.path.join(dirname, "images", "Universal")):
    for file_path in os.listdir(os.path.join(dirname, "images", "Universal", subfolder_path)):
        names.append(file_path)
for folder_path in os.listdir(os.path.join(dirname, "images")):
    if ".png" in folder_path or folder_path == "Universal":
        continue
    for subfolder_path in os.listdir(os.path.join(dirname, 'images', folder_path)):
        for file_path in os.listdir(os.path.join(dirname, "images", folder_path, subfolder_path)):
            if file_path in names:
                os.remove(os.path.join(dirname, "images", folder_path, subfolder_path, file_path))