github-repos/wget
1
0
Fork 0
mirror of https://github.com/mirror/wget.git synced 2025-03-29 21:50:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Smartly guess content type header

Browse Source
This commit is contained in:
Darshit Shah 2013-09-13 14:19:37 +05:30
parent d4d0e6378c
commit 43bb61b4e3
2 changed files with 32 additions and 2 deletions

10
testenv/ChangeLog
View File

@ -1,3 +1,13 @@
2013-09-13 Darshit Shah <darnir@gmail.com>
* HTTPServer.py (_Handler.do_HEAD): If requested path is /, respond
with /index.html
(_Handler.do_HEAD): Smartly guess value of Content-Type Header from
file extension
(_Handler.guess_type): Use a preset list of extensions and
Content-Type strings. If the extension matches one in the list, use
that string, else default to "text/plain"
2013-09-13 Darshit Shah <darnir@gmail.com> 2013-09-13 Darshit Shah <darnir@gmail.com>
* WgetTest.py (CommonMethods._replace_substring): New method that will * WgetTest.py (CommonMethods._replace_substring): New method that will

24
testenv/HTTPServer.py
View File

@ -1,4 +1,5 @@
from http.server import HTTPServer, BaseHTTPRequestHandler from http.server import HTTPServer, BaseHTTPRequestHandler
from posixpath import basename, splitext
from base64 import b64encode from base64 import b64encode
from random import random from random import random
from hashlib import md5 from hashlib import md5
@ -328,7 +329,11 @@ class _Handler (WgetHTTPRequestHandler):
""" Common code for GET and HEAD Commands. """ Common code for GET and HEAD Commands.
This method is overriden to use the fileSys dict. This method is overriden to use the fileSys dict.
""" """
path = self.path[1:]
if self.path == "/":
path = "index.html"
else:
path = self.path[1:]
if path in self.server.fileSys: if path in self.server.fileSys:
self.rules = self.server.server_configs.get (path) self.rules = self.server.server_configs.get (path)
@ -368,7 +373,8 @@ class _Handler (WgetHTTPRequestHandler):
content_length - 1, content_length - 1,
content_length)) content_length))
content_length -= self.range_begin content_length -= self.range_begin
self.send_header ("Content-type", "text/plain") cont_type = self.guess_type (path)
self.send_header ("Content-type", cont_type)
self.send_header ("Content-Length", content_length) self.send_header ("Content-Length", content_length)
self.finish_headers () self.finish_headers ()
return (content, self.range_begin) return (content, self.range_begin)
@ -376,6 +382,20 @@ class _Handler (WgetHTTPRequestHandler):
self.send_error (404, "Not Found") self.send_error (404, "Not Found")
return (None, None) return (None, None)
def guess_type (self, path):
base_name = basename ("/" + path)
name, ext = splitext (base_name)
extension_map = {
".txt" : "text/plain",
".css" : "text/css",
".html" : "text/html"
}
if ext in extension_map:
return extension_map[ext]
else:
return "text/plain"
class HTTPd (threading.Thread): class HTTPd (threading.Thread):
server_class = StoppableHTTPServer server_class = StoppableHTTPServer
handler = _Handler handler = _Handler