1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from optparse import OptionParser
import os
from BaseHTTPServer import HTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler
from nikola.plugin_categories import Command
class CommandBuild(Command):
"""Start test server."""
name = "serve"
def run(self, *args):
"""Start test server."""
parser = OptionParser(usage="nikola %s [options]" % self.name)
parser.add_option("-p", "--port", dest="port",
help="Port numer (default: 8000)", default=8000,
type="int")
parser.add_option("-a", "--address", dest="address",
help="Address to bind (default: 127.0.0.1)",
default='127.0.0.1')
(options, args) = parser.parse_args(list(args))
out_dir = self.site.config['OUTPUT_FOLDER']
if not os.path.isdir(out_dir):
print "Error: Missing '%s' folder?" % out_dir
else:
os.chdir(out_dir)
httpd = HTTPServer((options.address, options.port),
OurHTTPRequestHandler)
sa = httpd.socket.getsockname()
print "Serving HTTP on", sa[0], "port", sa[1], "..."
httpd.serve_forever()
class OurHTTPRequestHandler(SimpleHTTPRequestHandler):
extensions_map = dict(SimpleHTTPRequestHandler.extensions_map)
extensions_map[""] = "text/plain"
|
