Set content-type header based on mime-type

Fixes #7
2015-07-22 22:41:44 -05:00 · 2015-07-22 22:41:44 -05:00 · 63124a3026
parent 7f6be7e97a
commit 63124a3026
1 changed files with 4 additions and 1 deletions
--- a/server.py
+++ b/server.py
@ -130,7 +130,10 @@ class HttpHandler(BaseHTTPRequestHandler):
            try:
                with open(filename,'rb') as f:
                    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)))
                    fs = os.fstat(f.fileno())
                    self.send_header("Content-Length", str(fs.st_size))