forked from Hay1tsme/artemis
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import os
|
|
import hashlib
|
|
|
|
|
|
def prepare_images(image_folder="./images"):
|
|
print(f"Preparing image delivery files in {image_folder}...")
|
|
|
|
for file in os.listdir(image_folder):
|
|
if file.endswith(".png") or file.endswith(".jpg"):
|
|
dpg_name = "adv-" + file[:-4].upper()
|
|
if file.endswith(".png"):
|
|
dpg_name += ".dpg"
|
|
else:
|
|
dpg_name += ".djg"
|
|
|
|
if os.path.exists(os.path.join(image_folder, dpg_name)):
|
|
continue
|
|
else:
|
|
with open(
|
|
os.path.join(image_folder, file), "rb"
|
|
) as original_image_file:
|
|
original_image = original_image_file.read()
|
|
image_hash = hashlib.md5(original_image).hexdigest()
|
|
print(
|
|
f"DPG for {file} not found, creating with hash {image_hash}..."
|
|
)
|
|
md5_buf = bytes.fromhex(image_hash)
|
|
dpg_buf = md5_buf + original_image
|
|
with open(os.path.join(image_folder, dpg_name), "wb") as dpg_file:
|
|
dpg_file.write(dpg_buf)
|
|
|
|
print(f"Created {dpg_name}.")
|
|
|
|
|
|
# Call the function to execute it
|
|
prepare_images()
|