Source code for spectral_indices.sources.masks

from dataclasses import dataclass, field
from typing import Dict

from box import Box




@dataclass
class Mask:
    asset: str = field(default_factory=str)
    description: str = field(default_factory=str)



class Masks:
    # TODO docstrings & maybe refacto masks
    def __new__(self, mask_dict: Dict[str, str]) -> Box:
        masks_box = Box()
        for mask_name, maske_detail in mask_dict.items():
            masks_box[mask_name] = Mask(
                asset=maske_detail["asset"], description=maske_detail["description"]
            )

        return masks_box