from pycs.database.File import File
from pycs.interfaces.MediaFile import MediaFile


class AnnotatedMediaFile(MediaFile):
    # pylint: disable=too-few-public-methods
    """
    contains various attributes of a saved media file including annotations
    """

    def __init__(self, file: File):
        super().__init__(file)

        self.results = []
        for result in file.results():
            if result.origin == 'user':
                self.results.append({**{
                    'type': result.type,
                    'label': result.label
                }, **result.data})