a  d~@sfddlmZmZGdddedZGdddeZGdddeZGd d d eZGd d d eZd S))ABCMetaabstractmethodc@s(eZdZdZeddZeddZdS)BaseStorageBackendzAbstract class of storage backends. All backends need to implement two apis: ``get()`` and ``get_text()``. ``get()`` reads the file as a byte stream and ``get_text()`` reads the file as texts. cCsdSNselffilepathrrb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/file_client.pyget szBaseStorageBackend.getcCsdSrrrrrr get_textszBaseStorageBackend.get_textN)__name__ __module__ __qualname____doc__rr r rrrr rs  r) metaclassc@s*eZdZdZd ddZddZddZdS) MemcachedBackendaMemcached storage backend. Attributes: server_list_cfg (str): Config file for memcached server list. client_cfg (str): Config file for memcached client. sys_path (str | None): Additional path to be appended to `sys.path`. Default: None. NcCsr|durddl}|j|z ddl}WntyBtdYn0||_||_|j|j|j|_ | |_ dS)Nrz4Please install memcached to enable MemcachedBackend.) syspathappendmc ImportErrorserver_list_cfg client_cfgZMemcachedClientZ GetInstance_clientZpyvector _mc_buffer)rrrsys_pathrrrrr __init__ s   zMemcachedBackend.__init__cCs0t|}ddl}|j||j||j}|S)Nr)strrrZGetrZ ConvertBuffer)rr r value_bufrrr r /s  zMemcachedBackend.getcCstdSrNotImplementedErrorrrrr r 6szMemcachedBackend.get_text)Nr rrrrr r rrrr rs rc@s eZdZdZddZddZdS)HardDiskBackendzRaw hard disks storage backend.cCs>t|}t|d}|}Wdn1s00Y|S)Nrbropenreadrr frrrr r =s &zHardDiskBackend.getcCs>t|}t|d}|}Wdn1s00Y|S)Nrr%r(rrr r Cs &zHardDiskBackend.get_textN)r rrrr r rrrr r#:sr#c@s*eZdZdZd ddZddZd d Zd S) LmdbBackenda5Lmdb storage backend. Args: db_paths (str | list[str]): Lmdb database paths. client_keys (str | list[str]): Lmdb client keys. Default: 'default'. readonly (bool, optional): Lmdb environment parameter. If True, disallow any write operations. Default: True. lock (bool, optional): Lmdb environment parameter. If False, when concurrent access occurs, do not lock the database. Default: False. readahead (bool, optional): Lmdb environment parameter. If False, disable the OS filesystem readahead mechanism, which may improve random read performance when a database is larger than RAM. Default: False. Attributes: db_paths (list): Lmdb database path. _client (list): A list of several lmdb envs. defaultTFc Ksz ddl}Wnty&tdYn0t|tr8|g}t|trTdd|D|_nt|trjt|g|_t|t|jksJdt|dt|jdi|_t||jD](\}} |j | f|||d||j|<qdS) Nrz*Please install lmdb to enable LmdbBackend.cSsg|] }t|qSr)r).0vrrr hz(LmdbBackend.__init__..zCclient_keys and db_paths should have the same length, but received z and .)readonlylock readahead) lmdbr isinstancerlistdb_pathslenrzipr&) rr8Z client_keysr2r3r4kwargsr5clientrrrr r^s$       zLmdbBackend.__init__cCslt|}||jvs"Jd|d|j|}|jdd }||d}Wdn1s^0Y|S)zGet values according to the filepath from one lmdb named client_key. Args: filepath (str | obj:`Path`): Here, filepath is the lmdb key. client_key (str): Used for distinguishing different lmdb envs. z client_key z is not in lmdb clients.F)writeasciiN)rrbeginr encode)rr client_keyr<Ztxnrrrr r rs  .zLmdbBackend.getcCstdSrr rrrr r szLmdbBackend.get_textN)r,TFFr"rrrr r+Js r+c@s8eZdZdZeeedZd ddZd ddZ d d Z d S) FileClientaA general file client to access files in different backend. The client loads a file or text in a specified backend from its path and return it as a binary file. it can also register other backend accessor with a given name and backend class. Attributes: backend (str): The storage backend type. Options are "disk", "memcached" and "lmdb". client (:obj:`BaseStorageBackend`): The backend object. )diskZ memcachedr5rCcKsH||jvr(td|dt|j||_|j|fi||_dS)NzBackend z0 is not supported. Currently supported ones are ) _backends ValueErrorr7keysbackendr<)rrGr;rrr rs    zFileClient.__init__r,cCs(|jdkr|j||S|j|SdS)Nr5)rGr<r )rr rArrr r s zFileClient.getcCs |j|Sr)r<r rrrr r szFileClient.get_textN)rC)r,) r rrrr#rr+rDrr r rrrr rBs   rBN) abcrrrrr#r+objectrBrrrr s $: