o Q`#@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z m Z dd lmZejrUdd lmZmZmZmZmZmZmZmZd d l mZGd ddeZeddZdS)zB`Registry` class mapping protocols and FS URLs to their `Opener`. )absolute_import)print_function)unicode_literalsN)Opener)UnsupportedProtocolEntryPointError) parse_fs_url)CallableDictIteratorListTextTypeTupleUnionFSc@sxeZdZdZdddZddZdd Zed d Zd d Z    dddZ    dddZ e j   dddZdS)Registryz'A registry for `Opener` instances. osfsFcCs||_||_i|_dS)aCreate a registry object. Arguments: default_opener (str, optional): The protocol to use, if one is not supplied. The default is to use 'osfs', so that the FS URL is treated as a system path if no protocol is given. load_extern (bool, optional): Set to `True` to load openers from PyFilesystem2 extensions. Defaults to `False`. N)default_opener load_extern _protocols)selfrrr4/usr/lib/python3/dist-packages/fs/opener/registry.py__init__%s  zRegistry.__init__cCs d|jS)Nz)format protocols)rrrr__repr__5s zRegistry.__repr__cCsNt|tr|n|}t|tsJd|jsJd|jD]}||j|<q|S)aInstall an opener. Arguments: opener (`Opener`): an `Opener` instance, or a callable that returns an opener instance. Note: May be used as a class decorator. For example:: registry = Registry() @registry.install class ArchiveOpener(Opener): protocols = ['zip', 'tar'] zOpener instance requiredzmust list one or more protocols) isinstancerrr)ropenerZ_openerprotocolrrrinstall9s   zRegistry.installcCs>t|j}|jr|ddtdDttj|}|S)z1`list`: the list of supported protocols. css|]}|jVqdS)N)name).0 entry_pointrrr Ws  z%Registry.protocols.. fs.opener) listrrextend pkg_resourcesiter_entry_points collections OrderedDictfromkeys)rrrrrrOs  zRegistry.protocolsc Cs|p|j}|jrttd|d}nd}|dur+||jvr$|j|}|Std|z|}Wnt yD}zt d|d}~wwt |t sNt dz|}W|St yg}zt d|d}~ww)aGet the opener class associated to a given protocol. Arguments: protocol (str): A filesystem protocol. Returns: Opener: an opener instance. Raises: ~fs.opener.errors.UnsupportedProtocol: If no opener could be found for the given protocol. EntryPointLoadingError: If the returned entry point is not an `Opener` subclass or could not be loaded successfully. r)Nzprotocol '{}' is not supportedzcould not load entry point; {}z$entry point did not return an openerz could not instantiate opener; {}) rrnextr,r-rrrload Exceptionr issubclassr)rr#r'Zopener_instancer"Z exceptionrrr get_opener^s@      zRegistry.get_openerT.c CsLd|vr d||}t|}|j}|j}||} | |||||} | |fS)aXOpen a filesystem from a FS URL. Returns a tuple of a filesystem object and a path. If there is no path in the FS URL, the path value will be `None`. Arguments: fs_url (str): A filesystem URL. writeable (bool, optional): `True` if the filesystem must be writeable. create (bool, optional): `True` if the filesystem should be created if it does not exist. cwd (str): The current working directory. Returns: (FS, str): a tuple of ``(, )`` z://z{}://{})rr r#pathr5open_fs) rfs_url writeablecreatecwddefault_protocolZ parse_resultr#Z open_pathr"r8rrropens  z Registry.openc Cs:ddlm}t||r|}|S|j|||||d\}}|S)aOpen a filesystem from a FS URL (ignoring the path component). Arguments: fs_url (str): A filesystem URL. writeable (bool, optional): `True` if the filesystem must be writeable. create (bool, optional): `True` if the filesystem should be created if it does not exist. cwd (str): The current working directory (generally only relevant for OS filesystems). default_protocol (str): The protocol to use if one is not supplied in the FS URL (defaults to ``"osfs"``). Returns: ~fs.base.FS: A filesystem instance. rr)r:r;r<r=)baserr!r>) rr9r:r;r<r=r_fs_pathrrrr8s    zRegistry.open_fsccsTddlm}t||r|VdS|j||||d}z |VW|dS|w)aGet a context manager to open and close a filesystem. Arguments: fs_url (FS or str): A filesystem instance or a FS URL. create (bool, optional): If `True`, then create the filesystem if it doesn't already exist. writeable (bool, optional): If `True`, then the filesystem must be writeable. cwd (str): The current working directory, if opening a `~fs.osfs.OSFS`. Sometimes it is convenient to be able to pass either a FS object *or* an FS URL to a function. This context manager handles the required logic for that. Example: >>> def print_ls(list_fs): ... '''List a directory.''' ... with manage_fs(list_fs) as fs: ... print(' '.join(fs.listdir())) This function may be used in two ways. You may either pass a ``str``, as follows:: >>> print_list('zip://projects.zip') Or, an filesystem instance:: >>> from fs.osfs import OSFS >>> projects_fs = OSFS('~/') >>> print_list(projects_fs) rr)r;r:r<N)r?rr!r8close)rr9r;r:r<rr@rrr manage_fss *  zRegistry.manage_fsN)rF)TFr6r)FFr6r)FFr6)__name__ __module__ __qualname____doc__rr r$propertyrr5r>r8 contextlibcontextmanagerrCrrrrr!s.  < * (rT)r)rGZ __future__rrrr.rItypingr,r?rerrorsrrparser Z TYPE_CHECKINGr r r r rrrrrobjectrregistryrrrrs"     ( |