LEFT | RIGHT |
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-2013 Eyeo GmbH | 4 # Copyright (C) 2006-2013 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 sys, os, flask |
19 from ...utils import setupStderr | 19 from ...utils import setupStderr |
20 from ..utils import process_page | 20 from ..utils import process_page |
21 from ..sources import FileSource | 21 from ..sources import FileSource |
22 from ..converters import converters | 22 from ..converters import converters |
23 | 23 |
24 app = flask.Flask("sitescripts.web.bin.test_server") | 24 app = flask.Flask("sitescripts.web.bin.test_server") |
25 source = None | 25 source = None |
26 | 26 |
27 def get_data(path): | 27 def get_data(path): |
28 if "/" in path: | |
29 locale, page = path.split("/", 1) | |
30 for format in converters.iterkeys(): | |
31 if source.has_page(page, format): | |
32 return process_page(source, locale, page, format).encode("utf-8") | |
33 if source.has_localizable_file(locale, page): | |
34 return source.read_localizable_file(locale, page) | |
35 | |
36 if source.has_static(path): | 28 if source.has_static(path): |
37 return source.read_static(path) | 29 return source.read_static(path) |
38 | 30 |
| 31 path = path.rstrip("/") |
| 32 if path == "": |
| 33 path = source.read_config().get("general", "defaultlocale") |
| 34 if "/" not in path: |
| 35 path = "%s/%s" % (path, source.read_config().get("general", "defaultpage")) |
| 36 |
| 37 locale, page = path.split("/", 1) |
| 38 for format in converters.iterkeys(): |
| 39 if source.has_page(page, format): |
| 40 return process_page(source, locale, page, format).encode("utf-8") |
| 41 if source.has_localizable_file(locale, page): |
| 42 return source.read_localizable_file(locale, page) |
| 43 |
39 return None | 44 return None |
40 | 45 |
| 46 @app.route("/", methods = ["GET"]) |
41 @app.route("/<path:path>", methods = ["GET"]) | 47 @app.route("/<path:path>", methods = ["GET"]) |
42 def show(path): | 48 def show(path=""): |
43 data = get_data(path) | 49 data = get_data(path) |
44 if data == None: | 50 if data == None: |
45 flask.abort(404) | 51 flask.abort(404) |
46 | 52 |
47 root, ext = os.path.splitext(path) | 53 root, ext = os.path.splitext(path) |
48 if ext == ".js": | 54 if ext == ".js": |
49 mime = "text/js; charset=utf-8" | 55 mime = "application/javascript; charset=utf-8" |
50 elif ext == ".css": | 56 elif ext == ".css": |
51 mime = "text/css; charset=utf-8" | 57 mime = "text/css; charset=utf-8" |
52 elif ext == ".png": | 58 elif ext == ".png": |
53 mime = "image/png" | 59 mime = "image/png" |
54 elif ext == "": | 60 elif ext == "": |
55 mime = "text/html; charset=utf-8" | 61 mime = "text/html; charset=utf-8" |
56 else: | 62 else: |
57 mime = "application/octet-stream" | 63 mime = "application/octet-stream" |
58 return data, 200, {"Content-Type": mime} | 64 return data, 200, {"Content-Type": mime} |
59 | 65 |
60 if __name__ == "__main__": | 66 if __name__ == "__main__": |
61 setupStderr() | 67 setupStderr() |
62 if len(sys.argv) < 2: | 68 if len(sys.argv) < 2: |
63 print >>sys.stderr, "Usage: %s source_dir" % sys.argv[0] | 69 print >>sys.stderr, "Usage: %s source_dir" % sys.argv[0] |
64 sys.exit(1) | 70 sys.exit(1) |
65 | 71 |
66 source = FileSource(sys.argv[1]) | 72 source = FileSource(sys.argv[1]) |
67 | 73 |
68 # Make sure to "fix" argv to ensure that restart can succeed | 74 # Make sure to "fix" argv to ensure that restart can succeed |
69 sys.argv[0:1] = ["-m", "sitescripts.web.bin.test_server"] | 75 sys.argv[0:1] = ["-m", "sitescripts.web.bin.test_server"] |
70 | 76 |
71 app.run(debug=True) | 77 app.run(debug=True) |
LEFT | RIGHT |