| """Simple HTTP Server. |
| |
| This module builds on BaseHTTPServer by implementing the standard GET |
| and HEAD requests in a fairly straightforward manner. |
| |
| """ |
| |
| |
| __version__ = "0.4" |
| |
| |
| import os |
| import string |
| import posixpath |
| import BaseHTTPServer |
| import urllib |
| import cgi |
| import shutil |
| from StringIO import StringIO |
| |
| |
| class SimpleHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler): |
| |
| """Simple HTTP request handler with GET and HEAD commands. |
| |
| This serves files from the current directory and any of its |
| subdirectories. It assumes that all files are plain text files |
| unless they have the extension ".html" in which case it assumes |
| they are HTML files. |
| |
| The GET and HEAD requests are identical except that the HEAD |
| request omits the actual contents of the file. |
| |
| """ |
| |
| server_version = "SimpleHTTP/" + __version__ |
| |
| def do_GET(self): |
| """Serve a GET request.""" |
| f = self.send_head() |
| if f: |
| self.copyfile(f, self.wfile) |
| f.close() |
| |
| def do_HEAD(self): |
| """Serve a HEAD request.""" |
| f = self.send_head() |
| if f: |
| f.close() |
| |
| def send_head(self): |
| """Common code for GET and HEAD commands. |
| |
| This sends the response code and MIME headers. |
| |
| Return value is either a file object (which has to be copied |
| to the outputfile by the caller unless the command was HEAD, |
| and must be closed by the caller under all circumstances), or |
| None, in which case the caller has nothing further to do. |
| |
| """ |
| path = self.translate_path(self.path) |
| f = None |
| if os.path.isdir(path): |
| for index in "index.html", "index.htm": |
| index = os.path.join(path, index) |
| if os.path.exists(index): |
| path = index |
| break |
| else: |
| return self.list_directory(path) |
| ctype = self.guess_type(path) |
| if ctype.startswith('text/'): |
| mode = 'r' |
| else: |
| mode = 'rb' |
| try: |
| f = open(path, mode) |
| except IOError: |
| self.send_error(404, "File not found") |
| return None |
| self.send_response(200) |
| self.send_header("Content-type", ctype) |
| self.end_headers() |
| return f |
| |
| def list_directory(self, path): |
| """Helper to produce a directory listing (absent index.html). |
| |
| Return value is either a file object, or None (indicating an |
| error). In either case, the headers are sent, making the |
| interface the same as for send_head(). |
| |
| """ |
| try: |
| list = os.listdir(path) |
| except os.error: |
| self.send_error(404, "No permission to list directory"); |
| return None |
| list.sort(lambda a, b: cmp(a.lower(), b.lower())) |
| f = StringIO() |
| f.write("<h2>Directory listing for %s</h2>\n" % self.path) |
| f.write("<hr>\n<ul>\n") |
| for name in list: |
| fullname = os.path.join(path, name) |
| displayname = linkname = name = cgi.escape(name) |
| # Append / for directories or @ for symbolic links |
| if os.path.isdir(fullname): |
| displayname = name + "/" |
| linkname = name + os.sep |
| if os.path.islink(fullname): |
| displayname = name + "@" |
| # Note: a link to a directory displays with @ and links with / |
| f.write('<li><a href="%s">%s</a>\n' % (linkname, displayname)) |
| f.write("</ul>\n<hr>\n") |
| f.seek(0) |
| self.send_response(200) |
| self.send_header("Content-type", "text/html") |
| self.end_headers() |
| return f |
| |
| def translate_path(self, path): |
| """Translate a /-separated PATH to the local filename syntax. |
| |
| Components that mean special things to the local file system |
| (e.g. drive or directory names) are ignored. (XXX They should |
| probably be diagnosed.) |
| |
| """ |
| path = posixpath.normpath(urllib.unquote(path)) |
| words = string.splitfields(path, '/') |
| words = filter(None, words) |
| path = os.getcwd() |
| for word in words: |
| drive, word = os.path.splitdrive(word) |
| head, word = os.path.split(word) |
| if word in (os.curdir, os.pardir): continue |
| path = os.path.join(path, word) |
| return path |
| |
| def copyfile(self, source, outputfile): |
| """Copy all data between two file objects. |
| |
| The SOURCE argument is a file object open for reading |
| (or anything with a read() method) and the DESTINATION |
| argument is a file object open for writing (or |
| anything with a write() method). |
| |
| The only reason for overriding this would be to change |
| the block size or perhaps to replace newlines by CRLF |
| -- note however that this the default server uses this |
| to copy binary data as well. |
| |
| """ |
| shutil.copyfileobj(source, outputfile) |
| |
| def guess_type(self, path): |
| """Guess the type of a file. |
| |
| Argument is a PATH (a filename). |
| |
| Return value is a string of the form type/subtype, |
| usable for a MIME Content-type header. |
| |
| The default implementation looks the file's extension |
| up in the table self.extensions_map, using text/plain |
| as a default; however it would be permissible (if |
| slow) to look inside the data to make a better guess. |
| |
| """ |
| |
| base, ext = posixpath.splitext(path) |
| if self.extensions_map.has_key(ext): |
| return self.extensions_map[ext] |
| ext = string.lower(ext) |
| if self.extensions_map.has_key(ext): |
| return self.extensions_map[ext] |
| else: |
| return self.extensions_map[''] |
| |
| extensions_map = { |
| '': 'text/plain', # Default, *must* be present |
| '.html': 'text/html', |
| '.htm': 'text/html', |
| '.gif': 'image/gif', |
| '.jpg': 'image/jpeg', |
| '.jpeg': 'image/jpeg', |
| } |
| |
| |
| def test(HandlerClass = SimpleHTTPRequestHandler, |
| ServerClass = BaseHTTPServer.HTTPServer): |
| BaseHTTPServer.test(HandlerClass, ServerClass) |
| |
| |
| if __name__ == '__main__': |
| test() |