feat: a lot of a lot

decorators/locking.py

@@ -0,0 +1,32 @@
+from functools import wraps
+from typing import Any, Callable
+
+import redis
+
+import utils.hashing_utils
+from backend import config
+
+redis_client = redis.from_url(
+    config.CELERY_BROKER_URL,
+)
+
+
+def lock(lock_key: str, include_args_in_key=False) -> Callable:
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        def wrapper(*args, **kwargs) -> Any:
+            key = lock_key
+            if include_args_in_key:
+                key += '_' + utils.hashing_utils.hash_args_and_kwargs_with_crc32(*args, **kwargs)
+            redis_lock = redis_client.lock(key)
+            acquired = redis_lock.acquire(blocking=False)
+            if not acquired:
+                raise Exception(f'Lock {key} is already acquired')
+            try:
+                return func(*args, **kwargs)
+            finally:
+                redis_lock.release()
+
+        return wrapper
+
+    return decorator