diff options
Diffstat (limited to 'netbox/django-driver.py')
-rw-r--r-- | netbox/django-driver.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/netbox/django-driver.py b/netbox/django-driver.py index 65a9136..80bfa13 100644 --- a/netbox/django-driver.py +++ b/netbox/django-driver.py | |||
@@ -28,12 +28,20 @@ class DatabaseWrapper(OrigWrapper): | |||
28 | self._vault_cache_lock = threading.Lock() | 28 | self._vault_cache_lock = threading.Lock() |
29 | self._vault_cred_cache = Credential.empty() | 29 | self._vault_cred_cache = Credential.empty() |
30 | 30 | ||
31 | def close(self): | ||
32 | self._vault_cred_cache = Credential.empty() | ||
33 | super().close() | ||
34 | |||
31 | def close_if_unusable_or_obsolete(self): | 35 | def close_if_unusable_or_obsolete(self): |
32 | super().close_if_unusable_or_obsolete() | 36 | super().close_if_unusable_or_obsolete() |
33 | 37 | ||
34 | if self.connection is None: | 38 | if self.connection is None: |
35 | return | 39 | return |
36 | 40 | ||
41 | if not self.is_usable(): | ||
42 | self.close() | ||
43 | return | ||
44 | |||
37 | with self._vault_cache_lock: | 45 | with self._vault_cache_lock: |
38 | if not self._vault_cred_cache.is_valid: | 46 | if not self._vault_cred_cache.is_valid: |
39 | self.close() | 47 | self.close() |