Set content-type header based on mime-type

Fixes #7
This commit is contained in:
Sam Whited 2015-07-22 22:41:44 -05:00
parent 7f6be7e97a
commit 63124a3026

View File

@ -130,7 +130,10 @@ class HttpHandler(BaseHTTPRequestHandler):
try: try:
with open(filename,'rb') as f: with open(filename,'rb') as f:
self.send_response(200) self.send_response(200)
self.send_header("Content-Type", 'application/octet-stream') mime, _ = mimetypes.guess_type(filename)
if mime is None:
mime = 'application/octet-stream'
self.send_header("Content-Type", mime)
self.send_header("Content-Disposition", 'attachment; filename="{}"'.format(os.path.basename(filename))) self.send_header("Content-Disposition", 'attachment; filename="{}"'.format(os.path.basename(filename)))
fs = os.fstat(f.fileno()) fs = os.fstat(f.fileno())
self.send_header("Content-Length", str(fs.st_size)) self.send_header("Content-Length", str(fs.st_size))