Apple-2-Tools/atrcopy
1
0
Fork 0
mirror of https://github.com/robmcmullen/atrcopy.git synced 2024-12-01 09:49:42 +00:00
Code Issues Projects Releases Wiki Activity

Updated omnivore loader for new API

Browse Source
This commit is contained in:
Rob McMullen 2019-02-10 09:02:12 -08:00
parent b8a858ee6d
commit 9f5addf645
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atrcopy/omnivore_framework_loader.py
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