Ticket #3328: trap_early_errors-r3491.diff

trac/web/main.py

@@ -160 +160 @@
 
         # Select the component that should handle the request
         chosen_handler = None
-        if not req.path_info or req.path_info == '/':
-            chosen_handler = self.default_handler
-        else:
-            for handler in self.handlers:
-                if handler.match_request(req):
-                    chosen_handler = handler
-                    break
+        early_error = None
+        try:
+            if not req.path_info or req.path_info == '/':
+                chosen_handler = self.default_handler
+            else:
+                for handler in self.handlers:
+                    if handler.match_request(req):
+                        chosen_handler = handler
+                        break
 
-        for filter_ in self.filters:
-            chosen_handler = filter_.pre_process_request(req, chosen_handler)
+            for filter_ in self.filters:
+                chosen_handler = filter_.pre_process_request(req,
+                                                             chosen_handler)
+        except Exception, e:
+            early_error = (e, sys.exc_traceback)
+            
+        if not chosen_handler and not early_error:
+            early_error = (HTTPNotFound('No handler matched request to %s',
+                                        req.path_info), None)
 
-        if not chosen_handler:
-            raise HTTPNotFound('No handler matched request to %s',
-                               req.path_info)
-
         # Attach user information to the request
-        anonymous_request = getattr(chosen_handler, 'anonymous_request', False)
-        if anonymous_request:
-            req.authname = 'anonymous'
-            req.perm = NoPermissionCache()
-        else:
-            req.authname = self.authenticate(req)
-            req.perm = PermissionCache(self.env, req.authname)
-            req.session = Session(self.env, req)
+        try:
+            anonymous_request = getattr(chosen_handler, 'anonymous_request',
+                                        False)
+            if anonymous_request:
+                req.authname = 'anonymous'
+                req.perm = NoPermissionCache()
+            else:
+                req.authname = self.authenticate(req)
+                req.perm = PermissionCache(self.env, req.authname)
+                req.session = Session(self.env, req)
+        except Exception, e:
+            early_error = (e, sys.exc_traceback)
 
         # Prepare HDF for the clearsilver template
         use_template = getattr(chosen_handler, 'use_template', True)
@@ -193 +202 @@
             populate_hdf(req.hdf, self.env, req)
             chrome.populate_hdf(req, chosen_handler)
 
+        if early_error:
+            raise early_error[0], None, early_error[1]
+
         # Process the request and render the template
         try:
             try: