Updated omnivore loader for new API

This commit is contained in:
Rob McMullen 2019-02-10 09:02:12 -08:00
parent b8a858ee6d
commit 9f5addf645

View File

@ -4,10 +4,16 @@ import logging
log = logging.getLogger(__name__)
def identify_mime(header):
def identify_mime(header, fh):
mime_type = None
try:
parser, mime_type = find_diskimage_from_data(header, True)
fh.seek(0)
data = fh.read()
except IOError as e:
log.debug(f"atrcopy loader: error reading entire file: {e}")
else:
try:
parser, mime_type = find_diskimage_from_data(data, True)
except (errors.UnsupportedContainer, errors.UnsupportedDiskImage, IOError) as e:
print(f"error in atrcopy parser: {e}")
else:
@ -15,7 +21,7 @@ def identify_mime(header):
if mime_type:
log.debug(f"atrcopy loader: identified {mime_type}")
return dict(mime=mime_type, ext="")
return dict(mime=mime_type, ext="", atrcopy_parser=parser)
else:
log.debug(f"atrcopy loader: unidentified")
log.debug(f"atrcopy loader: not recognized")
return None