Test file lock drop-in

Author: wdbasson

acapy_agent/anoncreds/revocation.py

@@ -10,6 +10,7 @@
 from pathlib import Path
 from typing import List, Mapping, NamedTuple, Optional, Sequence, Tuple, Union
 from urllib.parse import urlparse
+from filelock import FileLock
 
 import base58
 from anoncreds import (
@@ -62,7 +63,19 @@
 REV_REG_DEF_STATE_ACTIVE = "active"
 
 # Module level lock
-_credential_creation_lock = asyncio.Lock()
+class AsyncFileLock:
+    def __init__(self, lock_file_path: str):
+        self.lock = FileLock(lock_file_path)
+
+    async def __aenter__(self):
+        await asyncio.get_event_loop().run_in_executor(None, self.lock.acquire)
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await asyncio.get_event_loop().run_in_executor(None, self.lock.release)
+
+_credential_creation_lock = AsyncFileLock("/home/aries/.indy_client/acapy_credential_creation.lock")
+
 class AnonCredsRevocationError(BaseError):
     """Generic revocation error."""
 
@@ -985,14 +998,14 @@ def _handle_missing_entries(rev_list: Entry, rev_reg_def: Entry, rev_key: Entry)
                     raise AnonCredsRevocationError(
                         "Revocation registry definition private data not found"
                     )
-    
+
             def _has_required_id_and_tails_path():
                 return rev_reg_def_id and tails_file_path
-    
+
             revoc = None
             credential_revocation_id = None
             rev_list = None
-    
+
             if _has_required_id_and_tails_path():
                 async with self.profile.session() as session:
                     rev_reg_def = await session.handle.fetch(
@@ -1002,20 +1015,20 @@ def _has_required_id_and_tails_path():
                     rev_key = await session.handle.fetch(
                         CATEGORY_REV_REG_DEF_PRIVATE, rev_reg_def_id
                     )
-    
+
                 _handle_missing_entries(rev_list, rev_reg_def, rev_key)
-    
+
                 rev_list_value_json = rev_list.value_json
                 rev_list_tags = rev_list.tags
-    
+
                 # If the rev_list state is failed then the tails file was never uploaded,
                 # try to upload it now and finish the revocation list
                 if rev_list_tags.get("state") == RevListState.STATE_FAILED:
                     await self.upload_tails_file(
                         RevRegDef.deserialize(rev_reg_def.value_json)
                     )
                     rev_list_tags["state"] = RevListState.STATE_FINISHED
-    
+
                 rev_reg_index = rev_list_value_json["next_index"]
                 try:
                     rev_reg_def = RevocationRegistryDefinition.load(rev_reg_def.raw_value)
@@ -1024,7 +1037,7 @@ def _has_required_id_and_tails_path():
                     raise AnonCredsRevocationError(
                         "Error loading revocation registry"
                     ) from err
-    
+
                 # NOTE: we increment the index ahead of time to keep the
                 # transaction short. The revocation registry itself will NOT
                 # be updated because we always use ISSUANCE_BY_DEFAULT.
@@ -1041,19 +1054,19 @@ def _has_required_id_and_tails_path():
                         tags=rev_list_tags,
                     )
                     await txn.commit()
-    
+
                 revoc = CredentialRevocationConfig(
                     rev_reg_def,
                     rev_key.raw_value,
                     rev_list,
                     rev_reg_index,
                 )
                 credential_revocation_id = str(rev_reg_index)
-    
+
             cred_def, cred_def_private = await self._get_cred_def_objects(
                 credential_definition_id
             )
-    
+
             try:
                 credential = await asyncio.get_event_loop().run_in_executor(
                     None,
@@ -1070,7 +1083,7 @@ def _has_required_id_and_tails_path():
                 )
             except AnoncredsError as err:
                 raise AnonCredsRevocationError("Error creating credential") from err
-    
+
             return credential.to_json(), credential_revocation_id
 
     async def create_credential(