lock staticmethod extracted

chatsky/context_storages/database.py

@@ -79,6 +79,18 @@ def __eq__(self, other: Any) -> bool:
         return super().__eq__(other)
 
 
+def _lock(function: Callable[..., Awaitable[Any]]):
+        @wraps(function)
+        async def wrapped(self, *args, **kwargs):
+            if not self.is_concurrent:
+                async with self._sync_lock:
+                    return await function(self, *args, **kwargs)
+            else:
+                return await function(self, *args, **kwargs)
+
+        return wrapped
+
+
 class DBContextStorage(ABC):
     _default_subscript_value: int = 3
 
@@ -111,18 +123,6 @@ def __init__(
     def is_concurrent(self) -> bool:
         raise NotImplementedError
 
-    @staticmethod
-    def _lock(function: Callable[..., Awaitable[Any]]):
-        @wraps(function)
-        async def wrapped(self, *args, **kwargs):
-            if not self.is_concurrent:
-                async with self._sync_lock:
-                    return await function(self, *args, **kwargs)
-            else:
-                return await function(self, *args, **kwargs)
-
-        return wrapped
-
     @classmethod
     def _validate_field_name(cls, field_name: str) -> str:
         if field_name not in (NameConfig._labels_field, NameConfig._requests_field, NameConfig._responses_field):

tests/core/test_context.py

@@ -1,4 +1,3 @@
-from altair import Key
 import pytest
 
 from chatsky.core.context import Context, ContextError