PyCS/labels/flat_moth_label_provider/moth_labels.py

import json
import typing as T

from pycs.interfaces.LabelProvider import LabelProvider
from pathlib import Path

class FlatMothLabels(LabelProvider):

    def __init__(self, root_folder: str, configuration: dict):

        mapping_file = Path(root_folder, configuration["mapping"])
        with open(mapping_file) as f:
            self.mapping = json.load(f)

    def close(self):
        pass

    def get_labels(self) -> T.List[dict]:

        labels = []
        for key, entries in self.mapping.items():
            display_name = f"{entries['genus']} {entries['species']} ({entries['kr']})"
            print(key, display_name)
            label = self.create_label(key, display_name)
            labels.append(label)

        return labels