Update agency_config_lock_manager.py

bonk1t · Dec 10, 2023 · 2b3080d · 2b3080d
1 parent 1e33730
commit 2b3080d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/nalgonda/agency_config_lock_manager.py b/src/nalgonda/agency_config_lock_manager.py
@@ -1,13 +1,11 @@
 import threading
-
+from collections import defaultdict
 
 class AgencyConfigLockManager:
     """Lock manager for agency config files"""
 
-    _locks: dict[str, threading.Lock] = {}
+    _locks: dict[str, threading.Lock] = defaultdict(threading.Lock)
 
     @classmethod
     def get_lock(cls, agency_id):
-        if agency_id not in cls._locks:
-            cls._locks[agency_id] = threading.Lock()
         return cls._locks[agency_id]