from pycs.database.Result import Result


class MediaImageLabel:
    def __init__(self, result: Result):
        self.label = result.label
        self.frame = result.data['frame'] if 'frame' in result.data else None

    def serialize(self) -> dict:
        return dict({'type': 'image-label'}, **self.__dict__)