setup_additional_images.py

import os
import django
from django.core.files import File
from django.conf import settings

# Initialize Django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'dyslexia.settings')
django.setup()

from api.models import GameImage, Letter

image_folder = os.path.join(settings.BASE_DIR, 'additional images')

for i in range(1, 27):
    letter_char = chr(96 + i)  # Convert number to corresponding lowercase letter (1 -> a, 2 -> b, etc.)
    image_path = os.path.join(image_folder, f'{i}.png')  # Assuming images are named 1.png, 2.png, etc.
    additional_image_path_1 = os.path.join(image_folder, f'{2 * i - 1}.png')  # Assuming additional images are named 1.png, 3.png, 5.png, etc.
    additional_image_path_2 = os.path.join(image_folder, f'{2 * i}.png')  # Assuming additional images are named 2.png, 4.png, 6.png, etc.
    
    if os.path.exists(image_path):
        with open(image_path, 'rb') as image_file:
            letter_obj, created = Letter.objects.get_or_create(letter=letter_char)
            game_image, created = GameImage.objects.get_or_create(letter=letter_obj)
            game_image.image.save(f'{letter_char}.png', File(image_file), save=True)
            
            if os.path.exists(additional_image_path_1):
                with open(additional_image_path_1, 'rb') as additional_image_file_1:
                    game_image.additional_image_1.save(f'{letter_char}_1.png', File(additional_image_file_1), save=True)
            
            if os.path.exists(additional_image_path_2):
                with open(additional_image_path_2, 'rb') as additional_image_file_2:
                    game_image.additional_image_2.save(f'{letter_char}_2.png', File(additional_image_file_2), save=True)
            
            print(f'Added images for letter {letter_char}')
    else:
        print(f'Image for letter {letter_char} not found')