Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 1 | """Simple HTTP Server. |
| 2 | |
| 3 | This module builds on BaseHTTPServer by implementing the standard GET |
| 4 | and HEAD requests in a fairly straightforward manner. |
| 5 | |
| 6 | """ |
| 7 | |
| 8 | |
Guido van Rossum | 077153e | 2001-01-14 23:21:25 +0000 | [diff] [blame] | 9 | __version__ = "0.6" |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 10 | |
Skip Montanaro | e99d5ea | 2001-01-20 19:54:20 +0000 | [diff] [blame] | 11 | __all__ = ["SimpleHTTPRequestHandler"] |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 12 | |
| 13 | import os |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 14 | import posixpath |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 15 | import BaseHTTPServer |
Guido van Rossum | d7b147b | 1999-11-16 19:04:32 +0000 | [diff] [blame] | 16 | import urllib |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 17 | import cgi |
Moshe Zadka | 37c03ff | 2000-07-29 05:15:56 +0000 | [diff] [blame] | 18 | import shutil |
Guido van Rossum | 077153e | 2001-01-14 23:21:25 +0000 | [diff] [blame] | 19 | import mimetypes |
Raymond Hettinger | a617271 | 2004-12-31 19:15:26 +0000 | [diff] [blame] | 20 | try: |
| 21 | from cStringIO import StringIO |
| 22 | except ImportError: |
| 23 | from StringIO import StringIO |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 24 | |
| 25 | |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 26 | class SimpleHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler): |
| 27 | |
| 28 | """Simple HTTP request handler with GET and HEAD commands. |
| 29 | |
| 30 | This serves files from the current directory and any of its |
Andrew M. Kuchling | b839c1f | 2004-08-07 19:02:19 +0000 | [diff] [blame] | 31 | subdirectories. The MIME type for files is determined by |
| 32 | calling the .guess_type() method. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 33 | |
| 34 | The GET and HEAD requests are identical except that the HEAD |
| 35 | request omits the actual contents of the file. |
| 36 | |
| 37 | """ |
| 38 | |
| 39 | server_version = "SimpleHTTP/" + __version__ |
| 40 | |
| 41 | def do_GET(self): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 42 | """Serve a GET request.""" |
| 43 | f = self.send_head() |
| 44 | if f: |
| 45 | self.copyfile(f, self.wfile) |
| 46 | f.close() |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 47 | |
| 48 | def do_HEAD(self): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 49 | """Serve a HEAD request.""" |
| 50 | f = self.send_head() |
| 51 | if f: |
| 52 | f.close() |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 53 | |
| 54 | def send_head(self): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 55 | """Common code for GET and HEAD commands. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 56 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 57 | This sends the response code and MIME headers. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 58 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 59 | Return value is either a file object (which has to be copied |
| 60 | to the outputfile by the caller unless the command was HEAD, |
| 61 | and must be closed by the caller under all circumstances), or |
| 62 | None, in which case the caller has nothing further to do. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 63 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 64 | """ |
| 65 | path = self.translate_path(self.path) |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 66 | f = None |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 67 | if os.path.isdir(path): |
Andrew M. Kuchling | 4281902 | 2006-12-22 19:06:16 +0000 | [diff] [blame] | 68 | if not self.path.endswith('/'): |
| 69 | # redirect browser - doing basically what apache does |
| 70 | self.send_response(301) |
| 71 | self.send_header("Location", self.path + "/") |
| 72 | self.end_headers() |
| 73 | return None |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 74 | for index in "index.html", "index.htm": |
| 75 | index = os.path.join(path, index) |
| 76 | if os.path.exists(index): |
| 77 | path = index |
| 78 | break |
| 79 | else: |
| 80 | return self.list_directory(path) |
| 81 | ctype = self.guess_type(path) |
| 82 | if ctype.startswith('text/'): |
| 83 | mode = 'r' |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 84 | else: |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 85 | mode = 'rb' |
| 86 | try: |
| 87 | f = open(path, mode) |
| 88 | except IOError: |
| 89 | self.send_error(404, "File not found") |
| 90 | return None |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 91 | self.send_response(200) |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 92 | self.send_header("Content-type", ctype) |
Georg Brandl | 5d07696 | 2006-02-17 13:34:16 +0000 | [diff] [blame] | 93 | fs = os.fstat(f.fileno()) |
| 94 | self.send_header("Content-Length", str(fs[6])) |
| 95 | self.send_header("Last-Modified", self.date_time_string(fs.st_mtime)) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 96 | self.end_headers() |
| 97 | return f |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 98 | |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 99 | def list_directory(self, path): |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 100 | """Helper to produce a directory listing (absent index.html). |
| 101 | |
| 102 | Return value is either a file object, or None (indicating an |
| 103 | error). In either case, the headers are sent, making the |
| 104 | interface the same as for send_head(). |
| 105 | |
| 106 | """ |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 107 | try: |
| 108 | list = os.listdir(path) |
| 109 | except os.error: |
Jeremy Hylton | 5b48c45 | 2001-01-26 17:08:32 +0000 | [diff] [blame] | 110 | self.send_error(404, "No permission to list directory") |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 111 | return None |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 112 | list.sort(key=lambda a: a.lower()) |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 113 | f = StringIO() |
Georg Brandl | 6ee6952 | 2005-12-16 19:36:08 +0000 | [diff] [blame] | 114 | displaypath = cgi.escape(urllib.unquote(self.path)) |
Andrew M. Kuchling | a629852 | 2008-02-28 14:03:03 +0000 | [diff] [blame] | 115 | f.write('<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">') |
| 116 | f.write("<html>\n<title>Directory listing for %s</title>\n" % displaypath) |
| 117 | f.write("<body>\n<h2>Directory listing for %s</h2>\n" % displaypath) |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 118 | f.write("<hr>\n<ul>\n") |
| 119 | for name in list: |
| 120 | fullname = os.path.join(path, name) |
Johannes Gijsbers | 6d63a8d | 2004-08-21 10:43:29 +0000 | [diff] [blame] | 121 | displayname = linkname = name |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 122 | # Append / for directories or @ for symbolic links |
| 123 | if os.path.isdir(fullname): |
| 124 | displayname = name + "/" |
Guido van Rossum | 1d105d1 | 2000-09-04 15:55:31 +0000 | [diff] [blame] | 125 | linkname = name + "/" |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 126 | if os.path.islink(fullname): |
| 127 | displayname = name + "@" |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 128 | # Note: a link to a directory displays with @ and links with / |
Johannes Gijsbers | 6d63a8d | 2004-08-21 10:43:29 +0000 | [diff] [blame] | 129 | f.write('<li><a href="%s">%s</a>\n' |
| 130 | % (urllib.quote(linkname), cgi.escape(displayname))) |
Andrew M. Kuchling | a629852 | 2008-02-28 14:03:03 +0000 | [diff] [blame] | 131 | f.write("</ul>\n<hr>\n</body>\n</html>\n") |
Martin v. Löwis | 587c98c | 2002-03-17 18:37:22 +0000 | [diff] [blame] | 132 | length = f.tell() |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 133 | f.seek(0) |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 134 | self.send_response(200) |
| 135 | self.send_header("Content-type", "text/html") |
Martin v. Löwis | 587c98c | 2002-03-17 18:37:22 +0000 | [diff] [blame] | 136 | self.send_header("Content-Length", str(length)) |
Guido van Rossum | 1d10f3e | 2000-05-21 16:25:29 +0000 | [diff] [blame] | 137 | self.end_headers() |
Guido van Rossum | 57af072 | 2000-05-09 14:57:09 +0000 | [diff] [blame] | 138 | return f |
| 139 | |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 140 | def translate_path(self, path): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 141 | """Translate a /-separated PATH to the local filename syntax. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 142 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 143 | Components that mean special things to the local file system |
| 144 | (e.g. drive or directory names) are ignored. (XXX They should |
| 145 | probably be diagnosed.) |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 146 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 147 | """ |
Georg Brandl | 45ab233 | 2006-01-13 17:05:56 +0000 | [diff] [blame] | 148 | # abandon query parameters |
Facundo Batista | d14600e | 2008-02-18 12:48:43 +0000 | [diff] [blame] | 149 | path = path.split('?',1)[0] |
| 150 | path = path.split('#',1)[0] |
Guido van Rossum | d7b147b | 1999-11-16 19:04:32 +0000 | [diff] [blame] | 151 | path = posixpath.normpath(urllib.unquote(path)) |
Eric S. Raymond | 304b6a3 | 2001-02-09 10:26:06 +0000 | [diff] [blame] | 152 | words = path.split('/') |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 153 | words = filter(None, words) |
| 154 | path = os.getcwd() |
| 155 | for word in words: |
| 156 | drive, word = os.path.splitdrive(word) |
| 157 | head, word = os.path.split(word) |
| 158 | if word in (os.curdir, os.pardir): continue |
| 159 | path = os.path.join(path, word) |
| 160 | return path |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 161 | |
| 162 | def copyfile(self, source, outputfile): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 163 | """Copy all data between two file objects. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 164 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 165 | The SOURCE argument is a file object open for reading |
| 166 | (or anything with a read() method) and the DESTINATION |
| 167 | argument is a file object open for writing (or |
| 168 | anything with a write() method). |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 169 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 170 | The only reason for overriding this would be to change |
| 171 | the block size or perhaps to replace newlines by CRLF |
| 172 | -- note however that this the default server uses this |
| 173 | to copy binary data as well. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 174 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 175 | """ |
Moshe Zadka | 37c03ff | 2000-07-29 05:15:56 +0000 | [diff] [blame] | 176 | shutil.copyfileobj(source, outputfile) |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 177 | |
| 178 | def guess_type(self, path): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 179 | """Guess the type of a file. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 180 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 181 | Argument is a PATH (a filename). |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 182 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 183 | Return value is a string of the form type/subtype, |
| 184 | usable for a MIME Content-type header. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 185 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 186 | The default implementation looks the file's extension |
Andrew M. Kuchling | b839c1f | 2004-08-07 19:02:19 +0000 | [diff] [blame] | 187 | up in the table self.extensions_map, using application/octet-stream |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 188 | as a default; however it would be permissible (if |
| 189 | slow) to look inside the data to make a better guess. |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 190 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 191 | """ |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 192 | |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 193 | base, ext = posixpath.splitext(path) |
Raymond Hettinger | 54f0222 | 2002-06-01 14:18:47 +0000 | [diff] [blame] | 194 | if ext in self.extensions_map: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 195 | return self.extensions_map[ext] |
Eric S. Raymond | bf97c9d | 2001-02-09 10:18:37 +0000 | [diff] [blame] | 196 | ext = ext.lower() |
Raymond Hettinger | 54f0222 | 2002-06-01 14:18:47 +0000 | [diff] [blame] | 197 | if ext in self.extensions_map: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 198 | return self.extensions_map[ext] |
| 199 | else: |
| 200 | return self.extensions_map[''] |
Tim Peters | 5d538b6 | 2006-06-14 22:38:13 +0000 | [diff] [blame] | 201 | |
Georg Brandl | 83cc0d0 | 2006-06-14 08:50:03 +0000 | [diff] [blame] | 202 | if not mimetypes.inited: |
| 203 | mimetypes.init() # try to read system mime.types |
Guido van Rossum | 077153e | 2001-01-14 23:21:25 +0000 | [diff] [blame] | 204 | extensions_map = mimetypes.types_map.copy() |
| 205 | extensions_map.update({ |
| 206 | '': 'application/octet-stream', # Default |
| 207 | '.py': 'text/plain', |
| 208 | '.c': 'text/plain', |
| 209 | '.h': 'text/plain', |
| 210 | }) |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 211 | |
| 212 | |
| 213 | def test(HandlerClass = SimpleHTTPRequestHandler, |
Guido van Rossum | 5c3b384 | 1998-12-07 04:08:30 +0000 | [diff] [blame] | 214 | ServerClass = BaseHTTPServer.HTTPServer): |
Guido van Rossum | e7e578f | 1995-08-04 04:00:20 +0000 | [diff] [blame] | 215 | BaseHTTPServer.test(HandlerClass, ServerClass) |
| 216 | |
| 217 | |
| 218 | if __name__ == '__main__': |
| 219 | test() |