PyCS/pycs/projects/MediaFile.py

from os import path, getcwd
from uuid import uuid1

from pycs.observable import ObservableDict


class MediaFile(ObservableDict):
    def __init__(self, obj, project_id):
        if 'predictionResults' not in obj.keys():
            obj['predictionResults'] = {}

        self.project_id = project_id
        super().__init__(obj)

    def _get_file(self, identifier):
        file_directory = path.join(getcwd(), 'projects', self.project_id, 'data')
        file_name = identifier + self['extension']

        return file_directory, file_name

    def get_file(self):
        return self._get_file(self['id'])

    def add_global_result(self, result, origin='user'):
        self.remove_global_result()
        self.add_result(result, origin)

    def remove_global_result(self):
        delete = []
        for result_id in self['predictionResults']:
            if 'x' not in self['predictionResults'][result_id]:
                delete.append(result_id)

        for result_id in delete:
            del self['predictionResults'][result_id]

    def add_result(self, result, origin='user'):
        result['id'] = str(uuid1())
        result['origin'] = origin

        self['predictionResults'][result['id']] = result

    def remove_result(self, identifier):
        del self['predictionResults'][identifier]

    def remove_pipeline_results(self):
        remove = list(filter(lambda k: self['predictionResults'][k]['origin'] == 'pipeline', self['predictionResults'].keys()))

        for key in remove:
            del self['predictionResults'][key]

    def remove_results(self):
        self['predictionResults'] = {}

    def update_result(self, identifier, result, origin='user'):
        result['id'] = identifier
        result['origin'] = origin

        self['predictionResults'][identifier] = result

    def resize(self, maximum_width):
        raise NotImplementedError