56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
|
import multiprocessing
|
||
|
|
import logging
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
from .icon import *
|
||
|
|
|
||
|
|
|
||
|
|
def _pack_animation(pair: set):
|
||
|
|
source, destination = pair
|
||
|
|
image = file2image(source)
|
||
|
|
image.write(destination)
|
||
|
|
|
||
|
|
|
||
|
|
def pack_animations(source: str, destination: str):
|
||
|
|
assert os.path.exists(source)
|
||
|
|
# Create list for processing
|
||
|
|
to_pack = []
|
||
|
|
dirpath, dirnames = next(os.walk(source))[:2]
|
||
|
|
dirnames.sort()
|
||
|
|
for dirname in dirnames:
|
||
|
|
current_directory = os.path.join(dirpath, dirname)
|
||
|
|
# Ensure destination folder
|
||
|
|
destination_directory = os.path.join(destination, dirname)
|
||
|
|
os.makedirs(destination_directory, exist_ok=True)
|
||
|
|
# Find all files
|
||
|
|
filenames = next(os.walk(current_directory))[2]
|
||
|
|
filenames.sort()
|
||
|
|
for filename in filenames:
|
||
|
|
if is_file_an_icon(filename):
|
||
|
|
source_filename = os.path.join(current_directory, filename)
|
||
|
|
destination_filename = os.path.join(
|
||
|
|
destination_directory, os.path.splitext(filename)[0] + ".bm"
|
||
|
|
)
|
||
|
|
to_pack.append((source_filename, destination_filename))
|
||
|
|
elif filename == "meta.txt":
|
||
|
|
source_filename = os.path.join(current_directory, filename)
|
||
|
|
destination_filename = os.path.join(destination_directory, filename)
|
||
|
|
shutil.copyfile(source_filename, destination_filename)
|
||
|
|
|
||
|
|
# Process images in parallel
|
||
|
|
pool = multiprocessing.Pool()
|
||
|
|
pool.map(_pack_animation, to_pack)
|
||
|
|
|
||
|
|
shutil.copyfile(
|
||
|
|
os.path.join(source, "manifest.txt"), os.path.join(destination, "manifest.txt")
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def pack_dolphin(source_directory: str, destination_directory: str):
|
||
|
|
pack_animations(
|
||
|
|
os.path.join(source_directory, "animations"),
|
||
|
|
os.path.join(destination_directory, "animations"),
|
||
|
|
)
|