Fixes: expected bytes, got <class 'str'> in download_samples()

pull/22/head
Nils 2016-07-21 13:43:04 +02:00 committed by GitHub
parent 7e06196ff3
commit ca5a4d0960
1 changed files with 2 additions and 2 deletions

View File

@ -917,10 +917,10 @@ class PyMISP(object):
archive = zipfile.ZipFile(zipped)
try:
# New format
unzipped = BytesIO(archive.open(f['md5'], pwd='infected').read())
unzipped = BytesIO(archive.open(f['md5'], pwd=b'infected').read())
except KeyError:
# Old format
unzipped = BytesIO(archive.open(f['filename'], pwd='infected').read())
unzipped = BytesIO(archive.open(f['filename'], pwd=b'infected').read())
details.append([f['event_id'], f['filename'], unzipped])
except zipfile.BadZipfile:
# In case the sample isn't zipped