AnnotatedMediaFile.py 592 B

1234567891011121314151617181920
  1. from pycs.database.File import File
  2. from pycs.interfaces.MediaFile import MediaFile
  3. class AnnotatedMediaFile(MediaFile):
  4. # pylint: disable=too-few-public-methods
  5. """
  6. contains various attributes of a saved media file including annotations
  7. """
  8. def __init__(self, file: File):
  9. super().__init__(file)
  10. self.results = []
  11. for result in file.results():
  12. if result.origin == 'user':
  13. self.results.append({**{
  14. 'type': result.type,
  15. 'label': result.label
  16. }, **result.data})