Briq/camelot-py
8
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make python2 compat

Browse Source
This commit is contained in:
Vinayak Mehta
2018-12-24 13:10:48 +05:30
parent c78957ae5a
commit 62ed4753cd
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
camelot/utils.py
+5 -1
View File
@@ -82,7 +82,11 @@ def download_url(url):
filename = '{}.pdf'.format(random_string(6)) filename = '{}.pdf'.format(random_string(6))
with tempfile.NamedTemporaryFile('wb', delete=False) as f: with tempfile.NamedTemporaryFile('wb', delete=False) as f:
obj = urlopen(url) obj = urlopen(url)
if obj.info().get_content_type() != 'application/pdf': if PY3:
content_type = obj.info().get_content_type()
else:
content_type = obj.info().getheader('Content-Type')
if content_type != 'application/pdf':
raise NotImplementedError("File format not supported") raise NotImplementedError("File format not supported")
f.write(obj.read()) f.write(obj.read())
filepath = os.path.join(os.path.dirname(f.name), filename) filepath = os.path.join(os.path.dirname(f.name), filename)