| OLD | NEW |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import logging | 5 import logging |
| 6 | 6 |
| 7 from appengine_wrappers import webapp2 | 7 from appengine_wrappers import webapp2 |
| 8 from handler import Handler | 8 from handler import Handler |
| 9 from servlet import Request | 9 from servlet import Request |
| 10 | 10 |
| 11 | 11 |
| 12 class AppEngineHandler(webapp2.RequestHandler): | 12 class AppEngineHandler(webapp2.RequestHandler): |
| 13 '''Top-level handler for AppEngine requests. Just converts them into our | 13 '''Top-level handler for AppEngine requests. Just converts them into our |
| 14 internal Servlet architecture. | 14 internal Servlet architecture. |
| 15 ''' | 15 ''' |
| 16 | 16 |
| 17 def post(self): |
| 18 self._HandleRequest() |
| 19 |
| 17 def get(self): | 20 def get(self): |
| 21 self._HandleRequest() |
| 22 |
| 23 def _HandleRequest(self): |
| 18 profile_mode = self.request.get('profile') | 24 profile_mode = self.request.get('profile') |
| 19 if profile_mode: | 25 if profile_mode: |
| 20 import cProfile, pstats, StringIO | 26 import cProfile, pstats, StringIO |
| 21 pr = cProfile.Profile() | 27 pr = cProfile.Profile() |
| 22 pr.enable() | 28 pr.enable() |
| 23 | 29 |
| 24 try: | 30 try: |
| 25 response = None | 31 response = None |
| 32 arguments = {} |
| 33 for argument in self.request.arguments(): |
| 34 arguments[argument] = self.request.get(argument) |
| 26 request = Request(self.request.path, | 35 request = Request(self.request.path, |
| 27 self.request.url[:-len(self.request.path)], | 36 self.request.url[:-len(self.request.path)], |
| 28 self.request.headers) | 37 self.request.headers, |
| 38 arguments) |
| 29 response = Handler(request).Get() | 39 response = Handler(request).Get() |
| 30 except Exception as e: | 40 except Exception as e: |
| 31 logging.exception(e) | 41 logging.exception(e) |
| 32 finally: | 42 finally: |
| 33 if profile_mode: | 43 if profile_mode: |
| 34 pr.disable() | 44 pr.disable() |
| 35 s = StringIO.StringIO() | 45 s = StringIO.StringIO() |
| 36 pstats.Stats(pr, stream=s).sort_stats(profile_mode).print_stats() | 46 pstats.Stats(pr, stream=s).sort_stats(profile_mode).print_stats() |
| 37 self.response.out.write(s.getvalue()) | 47 self.response.out.write(s.getvalue()) |
| 38 self.response.headers['Content-Type'] = 'text/plain' | 48 self.response.headers['Content-Type'] = 'text/plain' |
| 39 self.response.status = 200 | 49 self.response.status = 200 |
| 40 elif response: | 50 elif response: |
| 41 self.response.out.write(response.content.ToString()) | 51 self.response.out.write(response.content.ToString()) |
| 42 self.response.headers.update(response.headers) | 52 self.response.headers.update(response.headers) |
| 43 self.response.status = response.status | 53 self.response.status = response.status |
| 44 else: | 54 else: |
| 45 self.response.out.write('Internal server error') | 55 self.response.out.write('Internal server error') |
| 46 self.response.status = 500 | 56 self.response.status = 500 |
| OLD | NEW |