Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: cms/bin/test_server.py

Issue 6451712864813056: Issue 2196 - [cms] Drop depencency on Flask for test server (Closed)
Patch Set: Addressed comments Created March 23, 2015, 12:26 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « README.md ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # coding: utf-8 1 # coding: utf-8
2 2
3 # This file is part of the Adblock Plus web scripts, 3 # This file is part of the Adblock Plus web scripts,
4 # Copyright (C) 2006-2015 Eyeo GmbH 4 # Copyright (C) 2006-2015 Eyeo GmbH
5 # 5 #
6 # Adblock Plus is free software: you can redistribute it and/or modify 6 # Adblock Plus is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License version 3 as 7 # it under the terms of the GNU General Public License version 3 as
8 # published by the Free Software Foundation. 8 # published by the Free Software Foundation.
9 # 9 #
10 # Adblock Plus is distributed in the hope that it will be useful, 10 # Adblock Plus is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details. 13 # GNU General Public License for more details.
14 # 14 #
15 # You should have received a copy of the GNU General Public License 15 # You should have received a copy of the GNU General Public License
16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
17 17
18 import sys, os, flask 18 import mimetypes
19 import os
20 import sys
21
22 import jinja2
23
19 from ..utils import process_page 24 from ..utils import process_page
20 from ..sources import FileSource 25 from ..sources import FileSource
21 from ..converters import converters 26 from ..converters import converters
22 27
23 app = flask.Flask("cms.bin.test_server")
24 source = None 28 source = None
25 29
26 mime_types = { 30 UNICODE_ENCODING = "utf-8"
27 "": "text/html; charset=utf-8", 31
28 ".htm": "text/html; charset=utf-8", 32 ERROR_TEMPLATE = """
29 ".html": "text/html; charset=utf-8", 33 <html>
30 ".js": "application/javascript; charset=utf-8", 34 <head>
31 ".css": "text/css; charset=utf-8", 35 <title>{{status}}</title>
32 ".xml": "text/xml; charset=utf-8", 36 </head>
33 ".png": "image/png", 37 <body>
34 ".jpg": "image/jpeg", 38 <h1>{{status}}</h1>
35 ".jpeg": "image/jpeg", 39 {% set code = status.split()|first|int %}
36 } 40 {% if code == 404 %}
41 <p>No page found for the address {{uri}}.</p>
42 {% elif code == 500 %}
43 <p>An error occurred while processing the request for {{uri}}:</p>
44 <pre>{{error}}</pre>
45 {% endif %}
46 </body>
47 </html>"""
48
49 # Create our own instance, the default one will introduce "random" host-specific
50 # behavior by parsing local config files.
51 mime_types = mimetypes.MimeTypes()
37 52
38 def get_data(path): 53 def get_data(path):
39 if source.has_static(path): 54 if source.has_static(path):
40 return source.read_static(path) 55 return source.read_static(path)
41 56
42 path = path.rstrip("/") 57 path = path.strip("/")
43 if path == "": 58 if path == "":
44 path = source.read_config().get("general", "defaultlocale") 59 path = source.read_config().get("general", "defaultlocale")
45 if "/" in path: 60 if "/" in path:
46 locale, page = path.split("/", 1) 61 locale, page = path.split("/", 1)
47 else: 62 else:
48 locale, page = path, "" 63 locale, page = path, ""
49 64
50 default_page = source.read_config().get("general", "defaultpage") 65 default_page = source.read_config().get("general", "defaultpage")
51 alternative_page = "/".join([page, default_page]).lstrip("/") 66 alternative_page = "/".join([page, default_page]).lstrip("/")
52 for format in converters.iterkeys(): 67 for format in converters.iterkeys():
53 for p in (page, alternative_page): 68 for p in (page, alternative_page):
54 if source.has_page(p, format): 69 if source.has_page(p, format):
55 return process_page(source, locale, p, format, "http://127.0.0.1:5000"). encode("utf-8") 70 return process_page(source, locale, p, format, "http://127.0.0.1:5000")
56 if source.has_localizable_file(locale, page): 71 if source.has_localizable_file(locale, page):
57 return source.read_localizable_file(locale, page) 72 return source.read_localizable_file(locale, page)
58 73
59 return None 74 return None
60 75
61 @app.route("/", methods = ["GET"]) 76 def show_error(start_response, status, **kwargs):
62 @app.route("/<path:path>", methods = ["GET"]) 77 env = jinja2.Environment(autoescape=True)
63 def show(path=""): 78 template = env.from_string(ERROR_TEMPLATE)
79 mime = "text/html; encoding=%s" % UNICODE_ENCODING
80 start_response(status, [("Content-Type", mime)])
81 for fragment in template.stream(status=status, **kwargs):
82 yield fragment.encode(UNICODE_ENCODING)
83
84 def handler(environ, start_response):
85 path = environ.get("PATH_INFO")
86
64 data = get_data(path) 87 data = get_data(path)
65 if data == None: 88 if data == None:
66 flask.abort(404) 89 return show_error(start_response, "404 Not Found", uri=path)
67 90
68 root, ext = os.path.splitext(path) 91 mime = mime_types.guess_type(path)[0] or "text/html"
69 mime = mime_types.get(ext.lower(), "application/octet-stream") 92
70 return data, 200, {"Content-Type": mime} 93 if isinstance(data, unicode):
94 data = data.encode(UNICODE_ENCODING)
95 mime = "%s; charset=%s" % (mime, UNICODE_ENCODING)
96
97 start_response("200 OK", [("Content-Type", mime)])
98 return [data]
71 99
72 if __name__ == "__main__": 100 if __name__ == "__main__":
73 if len(sys.argv) < 2: 101 if len(sys.argv) < 2:
74 print >>sys.stderr, "Usage: %s source_dir" % sys.argv[0] 102 print >>sys.stderr, "Usage: %s source_dir" % sys.argv[0]
75 sys.exit(1) 103 sys.exit(1)
76 104
77 source = FileSource(sys.argv[1]) 105 source = FileSource(sys.argv[1])
78 106
79 app.run(debug=True) 107 try:
108 from werkzeug.serving import run_simple
109 except ImportError:
110 from wsgiref.simple_server import make_server
111 def run_simple(host, port, app, **kwargs):
112 def wrapper(environ, start_response):
113 try:
114 return app(environ, start_response)
115 except Exception, e:
116 return show_error(start_response, "500 Internal Server Error",
117 uri=environ.get("PATH_INFO"), error=e)
118
119 server = make_server(host, port, wrapper)
120 print " * Running on http://%s:%i/" % server.server_address
121 server.serve_forever()
122
123 run_simple("localhost", 5000, handler, use_reloader=True, use_debugger=True)
OLDNEW
« no previous file with comments | « README.md ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld