Another file lock

Author: sgonorov

tests/python_tests/requirements.txt

@@ -7,7 +7,6 @@ pytest==8.4.2
 transformers==4.55.4
 hf_transfer==0.1.9
 gguf==0.17.1
-filelock==3.20.0
 
 # rag requirements
 langchain_community==0.4

tests/python_tests/utils/constants.py

@@ -15,8 +15,8 @@
 from importlib import metadata
 from datetime import datetime, timedelta, timezone
 from pathlib import Path
-from filelock import FileLock
 from contextlib import contextmanager
+from utils.file_lock import FileLock
 
 ModelDownloaderCallable = Callable[[str], tuple[OVModel | None, AutoTokenizer | None, Path]]
 

tests/python_tests/utils/file_lock.py

@@ -0,0 +1,66 @@
+# Copyright (C) 2018-2025 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+import os
+import time
+import errno
+from pathlib import Path
+from contextlib import contextmanager
+
+class FileLock:
+    def __init__(self, lock_file: str, timeout: float = 300):
+        self.lock_file = Path(lock_file)
+        self.timeout = timeout
+        self._lock_file_fd = None
+    
+    def acquire(self, poll_interval: float = 0.1):
+        start_time = time.time()
+        
+        self.lock_file.parent.mkdir(parents=True, exist_ok=True)
+        
+        while True:
+            try:
+                self._lock_file_fd = os.open(
+                    str(self.lock_file),
+                    os.O_CREAT | os.O_EXCL | os.O_RDWR
+                )
+                break
+            except OSError as e:
+                if e.errno != errno.EEXIST:
+                    raise
+                
+                if self.timeout is not None and (time.time() - start_time) >= self.timeout:
+                    raise TimeoutError(f"Timeout acquiring lock on {self.lock_file}")
+                
+                time.sleep(poll_interval)
+    
+    def release(self):
+        if self._lock_file_fd is not None:
+            os.close(self._lock_file_fd)
+            self._lock_file_fd = None
+        
+        try:
+            self.lock_file.unlink()
+        except FileNotFoundError:
+            pass
+    
+    def __enter__(self):
+        self.acquire()
+        return self
+    
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.release()
+        return False
+    
+    def __del__(self):
+        self.release()
+
+@contextmanager
+def file_lock(lock_file: str, timeout: float = 300):
+    lock = FileLock(lock_file, timeout)
+    lock.acquire()
+    try:
+        yield lock
+    finally:
+        lock.release()
+