fixed status code provider to only throw on 500 and 404

MasoniteFramework · Jun 20, 2018 · 1bf03ff · vaibhavmule · Oct 10, 2018 · vaibhavmule
1 parent 44c3faf
commit 1bf03ff
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/masonite/providers/StatusCodeProvider.py b/masonite/providers/StatusCodeProvider.py
@@ -28,15 +28,16 @@ def boot(self, StatusCode, Request):
         if StatusCode == '200 OK':
             return
 
-        if self.app.make('ViewClass').exists('errors/{}'.format(StatusCode.split(' ')[0])):
-            rendered_view = self.app.make('View')('errors/{}'.format(StatusCode.split(' ')[0])).rendered_template
-        else:
-            rendered_view = self.app.make('View')('/masonite/snippets/statuscode', {
-                'code': StatusCode
-            }).rendered_template
-        Headers = [
-            ("Content-Length", str(len(rendered_view)))
-        ]
-        self.app.bind('Response', rendered_view)
-
-        self.app.bind('Headers', Headers)
+        if StatusCode in ('500 Internal Server Error', '404 Not Found'):
+            if self.app.make('ViewClass').exists('errors/{}'.format(StatusCode.split(' ')[0])):
+                rendered_view = self.app.make('View')('errors/{}'.format(StatusCode.split(' ')[0])).rendered_template
+            else:
+                rendered_view = self.app.make('View')('/masonite/snippets/statuscode', {
+                    'code': StatusCode
+                }).rendered_template
+            Headers = [
+                ("Content-Length", str(len(rendered_view)))
+            ]
+            self.app.bind('Response', rendered_view)
+
+            self.app.bind('Headers', Headers)