aboutsummaryrefslogtreecommitdiff
path: root/netbox/django-driver.py
diff options
context:
space:
mode:
Diffstat (limited to 'netbox/django-driver.py')
-rw-r--r--netbox/django-driver.py8
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()