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