don't use http.client.HTTPResponse.getheader() to get the content-type header, because it can return a comma-delimited string

This commit is contained in:
Noah Levitt 2017-09-18 14:45:16 -07:00
parent a8adaaf527
commit 1bca9d0324
2 changed files with 2 additions and 2 deletions

View File

@ -49,7 +49,7 @@ except:
setuptools.setup(
name='warcprox',
version='2.2b1.dev98',
version='2.2b1.dev99',
description='WARC writing MITM HTTP/S proxy',
url='https://github.com/internetarchive/warcprox',
author='Noah Levitt',

View File

@ -192,7 +192,7 @@ class WarcProxyHandler(warcprox.mitmproxy.MitmProxyHandler):
warcprox_meta=warcprox_meta, status=prox_rec_res.status,
size=prox_rec_res.recorder.len,
client_ip=self.client_address[0],
content_type=prox_rec_res.getheader("Content-Type"),
content_type=prox_rec_res.headers.get("Content-Type"),
method=self.command, timestamp=timestamp, host=self.hostname,
duration=datetime.datetime.utcnow()-timestamp)
self.server.recorded_url_q.put(recorded_url)