j:vddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZGdde Zy) N)IteratorSequence)Path)OptionalUnion) ByteStore)InvalidKeyExceptionc eZdZdZdddddeeefdeedeede d df d Z d ed efd Z d ed dfdZ de ed eeefdZde eeefd dfdZde ed dfdZddeed eefdZy)LocalFileStorea BaseStore interface that works on the local file system. Examples: Create a LocalFileStore instance and perform operations on it: .. code-block:: python from langchain.storage import LocalFileStore # Instantiate the LocalFileStore with the root path file_store = LocalFileStore("/path/to/root") # Set values for keys file_store.mset([("key1", b"value1"), ("key2", b"value2")]) # Get values for keys values = file_store.mget(["key1", "key2"]) # Returns [b"value1", b"value2"] # Delete keys file_store.mdelete(["key1"]) # Iterate over keys for key in file_store.yield_keys(): print(key) # noqa: T201 NF) chmod_file chmod_dir update_atime root_pathr r rreturncjt|j|_||_||_||_y)anImplement the BaseStore interface for the local file system. Args: root_path (Union[str, Path]): The root path of the file store. All keys are interpreted as paths relative to this root. chmod_file: (optional, defaults to `None`) If specified, sets permissions for newly created files, overriding the current `umask` if needed. chmod_dir: (optional, defaults to `None`) If specified, sets permissions for newly created dirs, overriding the current `umask` if needed. update_atime: (optional, defaults to `False`) If `True`, updates the filesystem access time (but not the modified time) when a file is read. This allows MRU/LRU cache policies to be implemented for filesystems where access time updates are disabled. N)rabsoluterr r r)selfrr r rs g/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain/storage/file_system.py__init__zLocalFileStore.__init__)s.,i113$"(keycPtjd|sd|}t||j|z j }|jj }t j j||g}|t|k7rd|d|d|d|}t||S)zGet the full path for a given key relative to the root path. Args: key (str): The key relative to the root path. Returns: Path: The full path for the given key. z^[a-zA-Z0-9_.\-/]+$zInvalid characters in key: z Invalid key: z+. Key should be relative to the full path. z vs. z and full path of ) rematchr rresolveospath commonpathstr)rrmsg full_pathr common_paths r_get_full_pathzLocalFileStore._get_full_pathDsxx.4/u5C%c* *^^c)224 NN**, gg(()Y)?@ #i. (u$O+U;-/A)N &c* *rdir_pathc|jsV|j|j|jd|j|j |jyyy)aPMakes a store directory path (including parents) with specified permissions This is needed because `Path.mkdir()` is restricted by the current `umask`, whereas the explicit `os.chmod()` used here is not. Args: dir_path: (Path) The store directory to make Returns: None T)exist_okN)exists_mkdir_for_storeparentmkdirr chmod)rr$s rr(zLocalFileStore._mkdir_for_store\sR  ! !(// 2 NNDN )~~)t~~.*!rkeysc^g}|D]}|j|}|jrq|j}|j||jsRt j |tj|jjf|jd|S)a Get the values associated with the given keys. Args: keys: A sequence of keys. Returns: A sequence of optional values associated with the keys. If a key is not found, the corresponding value will be None. N) r#r' read_bytesappendrrutimetimestatst_mtime)rr,valuesrr!values rmgetzLocalFileStore.mgetns)+C++C0I!!,,. e$$$HHYinn6F6O6O(PQ d# rkey_value_pairsc|D]j\}}|j|}|j|j|j||jP|j |jly)zSet the values for the given keys. Args: key_value_pairs: A sequence of key-value pairs. Returns: None N)r#r(r) write_bytesr r+)rr7rr5r!s rmsetzLocalFileStore.msets^*JC++C0I  ! !)"2"2 3  ! !% (*0 *rcv|D]4}|j|}|js%|j6y)zDelete the given keys and their associated values. Args: keys (Sequence[str]): A sequence of keys to delete. Returns: None N)r#r'unlink)rr,rr!s rmdeletezLocalFileStore.mdeletes6C++C0I!  "rprefixc#K|r|j|n |j}|jdD];}|js|j |j}t |=yw)zGet an iterator over keys that match the given prefix. Args: prefix (Optional[str]): The prefix to match. Returns: Iterator[str]: An iterator over keys that match the given prefix. *N)r#rrglobis_file relative_tor)rr> prefix_pathfile relative_paths r yield_keyszLocalFileStore.yield_keyss]6%)#'" )d#)SM ) C= )  ) )6#$0//$/$#4+@.1HU3:->$?1D1 #HSM #d # )# )(3- )rr )rrr1collections.abcrrpathlibrtypingrrlangchain_core.storesrlangchain.storage.exceptionsr r rQrrrWs+ ."+<c)Yc)r