Skip to content

Commit 03caded

Browse files
authored
Merge pull request #10 from florgon/gatey-fix-starlette-hooks
2 parents 4c94574 + 4c53a44 commit 03caded

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

gatey_sdk/integrations/starlette.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -66,15 +66,13 @@ def __init__(
6666
)
6767

6868
# Hooks.
69-
self.pre_capture_hook = (
70-
pre_capture_hook if pre_capture_hook else self._default_void_hook
71-
)
72-
self.post_capture_hook = (
73-
post_capture_hook if post_capture_hook else self._default_void_hook
74-
)
75-
self.on_request_hook = (
76-
on_request_hook if on_request_hook else self._default_void_hook
77-
)
69+
hooks = {
70+
"pre_capture_hook": pre_capture_hook,
71+
"post_capture_hook": post_capture_hook,
72+
"on_request_hook": on_request_hook,
73+
}
74+
for name, hook in hooks.items():
75+
setattr(self, name, hook or self._default_void_hook)
7876

7977
self.gatey_client = self.client_getter()
8078
if not isinstance(self.gatey_client, Client):

0 commit comments

Comments
 (0)