o ADi,@sdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZddlZgd ZdZGdddeedfZeZddd'ddZddd(d!d"Ze d#dd)d%d&Z dS)*aupath.registry -- registry for file system specific implementations Retrieve UPath implementations via `get_upath_class`. Register custom UPath subclasses in one of two ways: ### directly from Python >>> from upath import UPath >>> from upath.registry import register_implementation >>> my_protocol = "myproto" >>> class MyPath(UPath): ... pass >>> register_implementation(my_protocol, MyPath) ### via entry points ```toml # pyproject.toml [project.entry-points."universal_pathlib.implementations"] myproto = "my_module.submodule:MyPath" ``` ```ini # setup.cfg [options.entry_points] universal_pathlib.implementations = myproto = my_module.submodule:MyPath ``` ) annotationsN)ChainMap) lru_cache) import_module) entry_points) TYPE_CHECKING)Iterator)MutableMapping)get_filesystem_class)known_implementations)get_upath_classavailable_implementationsregister_implementationz!universal_pathlib.implementationsc@seZdZUdZiddddddddddd d d d d d dd ddddddddddddddddddddddZd ed!<erNd"ed#<d?d&d'Zd@d+d,ZdAd/d0Z dBd3d4Z dCd6d7Z dDd9d:Z dEdS)F _Registryz&internal registry for UPath subclassesabfsz%upath.implementations.cloud.AzurePathabfssadlazdataz#upath.implementations.data.DataPathfilez$upath.implementations.local.FilePathlocalgcsz#upath.implementations.cloud.GCSPathgshdfsz#upath.implementations.hdfs.HDFSPathhttpz#upath.implementations.http.HTTPPathhttpsmemoryz'upath.implementations.memory.MemoryPaths3z"upath.implementations.cloud.S3Paths3asftpz#upath.implementations.sftp.SFTPPathsshz'upath.implementations.webdav.WebdavPathz'upath.implementations.github.GitHubPathz!upath.implementations.smb.SMBPath)webdavz webdav+httpz webdav+httpsgithubsmbzdict[str, str]r z,MutableMapping[str, str | type[upath.UPath]]_mreturnNonecCsFtjdkr ttd}nttg}dd|D|_ti|j|_dS)N) )groupcSsi|]}|j|qS)name).0epr*r*G/var/www/Datamplify/venv/lib/python3.10/site-packages/upath/registry.py `sz&_Registry.__init__..) sys version_infor_ENTRY_POINT_GROUPget_entriesrr r$)selfepsr*r*r.__init__[s  z_Registry.__init__itemobjectboolcCs|t|j|jvSN)setunionr$r4)r5r8r*r*r. __contains__cz_Registry.__contains__strtype[upath.UPath]cCs|j|}|dur||jvr|j|}|j|<|dur&t|dt|tr>|dd\}}t|}t ||}|S|}|S)Nz not in registry.) r$r3r4loadKeyError isinstancer@rsplitrgetattr)r5r8fqn module_namer+modclsr*r*r. __getitem__fs    z_Registry.__getitem__valuetype[upath.UPath] | strcCsXt|tr t|tjst|tstdt|j|r!||jvr%t ||j|<dS)Nz,expected UPath subclass or FQN-string, got: ) rFtype issubclassupathUPathr@ ValueError__name__r$r cache_clear)r5r8rNr*r*r. __setitem__us z_Registry.__setitem__ _Registry__vcCstd)Nzremoval is unsupported)NotImplementedError)r5rXr*r*r. __delitem__sz_Registry.__delitem__intcCtt|j|jSr;)lenr<r=r$r4r5r*r*r.__len__r?z_Registry.__len__ Iterator[str]cCr\r;)iterr<r=r$r4r^r*r*r.__iter__r?z_Registry.__iter__N)r%r&)r8r9r%r:)r8r@r%rA)r8r@rNrOr%r&)rXr@r%r&)r%r[)r%r`)rU __module__ __qualname____doc__r __annotations__rr7r>rMrWrZr_rbr*r*r*r.r<sh           rrAF)fallbackrgr:r% list[str]cCs$tt}|s|Sth|ttS)zreturn a list of protocols for available implementations Parameters ---------- fallback: If True, also return protocols for fsspec filesystems without an implementation in universal_pathlib. )list _registry_fsspec_known_implementations)rgimplr*r*r.r s r )clobberprotocolr@rLrOrmr&cCs@td|s t|d|s|tvrt|d|t|<dS)aregister a UPath implementation with a protocol Parameters ---------- protocol: Protocol name to associate with the class cls: The UPath subclass for the protocol or a str representing the full path to an implementation class like package.module.class. clobber: Whether to overwrite a protocol with the same name; if False, will raise instead. z^[a-z][a-z0-9+_.]+$z is not a valid URI schemez- is already in registry and clobber is False!N)rematchrTrj)rnrLrmr*r*r.rs   rTtype[upath.UPath] | Nonec Cszt|WStyO|s'tjdkrddlm}|YSddlm}|YS|s,YdSzt|}Wn ty=YYdSwt j d|dt dd t j YSw) aJReturn the upath cls for the given protocol. Returns `None` if no matching protocol can be found. Parameters ---------- protocol: The protocol string fallback: If fallback is False, don't return UPath instances for fsspec filesystems that don't have an implementation registered. ntr) WindowsUPath) PosixUPathNzUPath zr filesystem not explicitly implemented. Falling back to default implementation. This filesystem may not be tested.) stacklevel)rjrEosr+upath.implementations.localrsrtr rTwarningswarn UserWarningrRrS)rnrgrsrt_r*r*r.r s.         r )rgr:r%rh)rnr@rLrOrmr:r%r&)rnr@rgr:r%rq)!re __future__rrwror0ry collectionsr functoolsr importlibrimportlib.metadatartypingrrr fsspec.corer fsspec.registryr rkrR__all__r2r@rrjr rr r*r*r*r.s4          O