Merge pull request #5 from har777/fix-gracefully-fail-callback-function-loading-error

har777 · web-flow · commit 370500025c48 · 2022-07-18T14:11:45.000+05:30
Gracefully fail callback function if module not found
diff --git a/pellet/middleware.py b/pellet/middleware.py
@@ -94,7 +94,7 @@ def __call__(self, request):
                     response=response,
                     pellet_metrics=pellet_metrics.__dict__,
                 )
-            except (AttributeError, IndexError):
+            except (AttributeError, IndexError, ModuleNotFoundError):
                 logger.exception(
                     "Pellet callback: {callback_name} failed".format(
                         callback_name=callback_path
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pellet"
-version = "0.1.2"
+version = "0.1.3"
 description = "Pellet helps improve your Django app performance by discovering N+1 queries"
 authors = ["Harikrishnan Shaji <hihari777@gmail.com>"]
 readme = "README.md"

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def __call__(self, request):`
`94`	`94`	`response=response,`
`95`	`95`	`pellet_metrics=pellet_metrics.__dict__,`
`96`	`96`	`)`
`97`		`- except (AttributeError, IndexError):`
	`97`	`+ except (AttributeError, IndexError, ModuleNotFoundError):`
`98`	`98`	`logger.exception(`
`99`	`99`	`"Pellet callback: {callback_name} failed".format(`
`100`	`100`	`callback_name=callback_path`