import functools import mimetypes import os from platypush.utils import get_mime_type from . import MediaHandler class FileHandler(MediaHandler): prefix_handlers = ['file://'] def __init__(self, source, *args, **kwargs): super().__init__(source, *args, **kwargs) self.path = os.path.abspath(os.path.expanduser( self.source[len(self._matched_handler):])) self.filename = self.path.split('/')[-1] if not os.path.isfile(self.path): raise FileNotFoundError('{} is not a valid file'. format(self.path)) self.mime_type = get_mime_type(source) if self.mime_type[:5] not in ['video', 'audio', 'image']: raise AttributeError('{} is not a valid media file'.format(source)) self.extension = mimetypes.guess_extension(self.mime_type) if self.url: self.url += self.extension self.content_length = os.path.getsize(self.path) def get_data(self, from_bytes=None, to_bytes=None, chunk_size=None): if from_bytes is None: from_bytes = 0 if to_bytes is None: to_bytes = os.path.getsize(self.path) if chunk_size is None: chunk_size = os.path.getsize(self.path) - from_bytes with open(self.path, 'rb') as f: f.seek(from_bytes) for chunk in iter(functools.partial( f.read, min(to_bytes-from_bytes, chunk_size)), b''): yield chunk # vim:sw=4:ts=4:et: