import os
import base64
import re
from django.apps import apps
from django.core.files.base import ContentFile
from django.core.files.storage import default_storage


def extract_images(app_name, model_name, field_name, object_id=None):
    # Expressão regular para identificar a propriedade 'src' da imagem em base64
    image_data_re = re.compile(r'src=\"data:image/(?P<type>[a-z]+);base64,(?P<data>[^"]+)\"')

    try:
        Model = apps.get_model(app_name, model_name)
    except LookupError:
        print(f'O modelo "{model_name}" no aplicativo "{app_name}" não existe.')
        return

    # Obter todos os objetos ou um objeto específico
    if object_id is None:
        objects = Model.objects.all()
    else:
        objects = [Model.objects.get(id=object_id)]

    for obj in objects:
        new_text = getattr(obj, field_name)  # cópia do texto original
        for match in image_data_re.finditer(new_text):
            image_data = match.group('data')
            image_type = match.group('type')

            # Decodifica os dados da imagem em base64
            decoded_image = base64.b64decode(image_data)
            # Cria um objeto Django ContentFile para a imagem decodificada
            image_file = ContentFile(decoded_image)
            # Constroi um nome de arquivo
            filename = f'{obj.id}_{match.start()}.{image_type}'

            # Caminho personalizado para salvar o arquivo
            # Exemplo: sapl/public/dispositivo/123
            # todo: verificar se isso pode causar problemas com o armazenamento padrão
            folder_path = os.path.join('sapl', 'public', model_name.lower(), str(obj.id))

            # Salva o arquivo de imagem no armazenamento
            path = default_storage.save(os.path.join(folder_path, filename), image_file)

            # Substitui a propriedade 'src' da imagem pelos dados da imagem pela URL da imagem salva
            new_text = new_text.replace(match.group(), f'src="{default_storage.url(path)}"')
            print(f'Imagem salva em {path}')

        # Atualiza o campo com o novo texto
        print(f'Novo texto: {new_text}')
        setattr(obj, field_name, new_text)
        obj.save()

# Exemplo de uso:
# extract_images('compilacao', 'Dispositivo', 'texto')
# extract_images('compilacao', 'Dispositivo', 'texto', object_id=123)