winter-telescope / winterdrp / 3769534174

import logging

from winterdrp.data import ImageBatch

from winterdrp.errors import ErrorStack

from winterdrp.paths import PROC_FAIL_KEY

from winterdrp.processors.base_processor import BaseImageProcessor

logger = logging.getLogger(__name__)

class ErrorStackAnnotator(BaseImageProcessor):

    base_key = "errorannotate"

    def __init__(self, errorstack: ErrorStack, processed_images: list[str]):

        super().__init__()

        self.errorstack = errorstack

        self.image_dict = self.unpack_errorstack()

        self.processed_images = processed_images

    def unpack_errorstack(self) -> dict:

        image_dict = {}

        all_reports = self.errorstack.get_all_reports()

        for error_report in all_reports:

        return image_dict

    def _apply_to_images(self, batch: ImageBatch) -> ImageBatch:

        for image in batch:

            base_name = image.get_name()

            if base_name in self.image_dict:

            elif self.processed_images is not None:

                if base_name not in self.processed_images:

                    image[PROC_FAIL_KEY] += "Not Processed"

        return batch