Skip to content

Commit 7a2dd89

Browse files
authored
Merge pull request #146 from ramnes/enhance-hooks-disabling
Enhance hooks disabling by supporting exceptions
2 parents eb24cea + f214721 commit 7a2dd89

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

datafiles/hooks.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,9 +159,10 @@ def enabled(mapper, args) -> bool:
159159
@contextmanager
160160
def disabled():
161161
"""Globally disable method hooks, temporarily."""
162-
if settings.HOOKS_ENABLED:
162+
enabled = settings.HOOKS_ENABLED
163+
if enabled:
163164
settings.HOOKS_ENABLED = False
165+
try:
164166
yield
165-
settings.HOOKS_ENABLED = True
166-
else:
167-
yield
167+
finally:
168+
settings.HOOKS_ENABLED = enabled

0 commit comments

Comments
 (0)