o ais@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZddlmZddlmZmZddlmZmZmZmZdd lmZmZmZmZmZmZm Z gd Z!e"e#Z$d Z%d Z&d eddde%dfZ'dZ(Gddde)Z*Gddde)Z+Gddde)Z,Gddde,Z-Gddde-Z.Gddde-Z/e.Z0e/Z1Gdd d e)Z2d*d"d#Z3d$d%Z4d&d'Z5d(d)Z6dS)+zPEP 376 implementation.)unicode_literalsN)DistlibException resources)StringIO) get_schemeUnsupportedVersionError)MetadataMETADATA_FILENAMEWHEEL_METADATA_FILENAMELEGACY_METADATA_FILENAME)parse_requirementcached_propertyparse_name_and_version read_exports write_exports CSVReader CSVWriter) DistributionBaseInstalledDistributionInstalledDistributionEggInfoDistributionDistributionPathzpydist-exports.jsonzpydist-commands.json INSTALLERRECORD REQUESTED RESOURCESSHAREDz .dist-infoc@s(eZdZdZddZddZddZdS) _CachezL A simple cache mapping names and .dist-info paths to distributions cCsi|_i|_d|_dS)zZ Initialise an instance. There is normally one for each DistributionPath. FN)namepath generatedselfr$/builddir/build/BUILDROOT/alt-python310-pip-21.3.1-5.el8.x86_64/opt/alt/python310/lib/python3.10/site-packages/pip/_vendor/distlib/database.py__init__1s z_Cache.__init__cCs|j|jd|_dS)zC Clear the cache, setting it to its initial state. FN)rclearr r!r"r$r$r%r'9s   z _Cache.clearcCs6|j|jvr||j|j<|j|jg|dSdS)z` Add a distribution to the cache. :param dist: The distribution to add. N)r r setdefaultkeyappendr#distr$r$r%addAs  z _Cache.addN)__name__ __module__ __qualname____doc__r&r'r-r$r$r$r%r-s  rc@seZdZdZdddZddZdd ZeeeZd d Z d d Z ddZ e ddZ ddZddZdddZddZdddZdS)rzU Represents a set of distributions installed on a path (typically sys.path). NFcCsD|durtj}||_d|_||_t|_t|_d|_td|_ dS)a Create an instance from a path, optionally including legacy (distutils/ setuptools/distribute) distributions. :param path: The path to use, as a list of directories. If not specified, sys.path is used. :param include_egg: If True, this instance will look for and return legacy distributions as well as those based on PEP 376. NTdefault) sysr _include_dist _include_eggr_cache _cache_egg_cache_enabledr_scheme)r#r Z include_eggr$r$r%r&Os zDistributionPath.__init__cCs|jSNr8r"r$r$r%_get_cache_enabledcsz#DistributionPath._get_cache_enabledcCs ||_dSr:r;)r#valuer$r$r%_set_cache_enabledfs z#DistributionPath._set_cache_enabledcCs|j|jdS)z, Clears the internal cache. N)r6r'r7r"r$r$r% clear_cacheks zDistributionPath.clear_cachec csTt}|jD]}t|}|durq|d}|r|jsqt|j}|D]}||}|r3|j|vr4q%|jr|t rt t t g}|D]}t ||} || } | rTnqCq%t| } t| dd} Wdn1snwYtd|j||jt|j| |dVq%|jr|drtd|j||jt|j|Vq%qdS)zD Yield .dist-info and/or .egg(-info) distributions. NlegacyfileobjschemezFound %s)metadataenv) .egg-info.egg)setr rfinder_for_pathfind is_containersortedr4endswith DISTINFO_EXTr r r posixpathjoin contextlibclosing as_streamr loggerdebugr-new_dist_classr5old_dist_class) r#seenr finderrZrsetentryZpossible_filenamesZmetadata_filenameZ metadata_pathZpydiststreamrEr$r$r%_yield_distributionsssR           z%DistributionPath._yield_distributionscCsx|jj }|jo |jj }|s|r8|D]}t|tr#|j|q|j|q|r0d|j_|r:d|j_dSdSdS)zk Scan the path for distributions and populate the cache with those that are found. TN)r6r!r5r7r^ isinstancerr-)r#Zgen_distZgen_eggr,r$r$r%_generate_caches     z DistributionPath._generate_cachecCs|dd}d||gtS)ao The *name* and *version* parameters are converted into their filename-escaped form, i.e. any ``'-'`` characters are replaced with ``'_'`` other than the one in ``'dist-info'`` and the one separating the name from the version number. :parameter name: is converted to a standard distribution name by replacing any runs of non- alphanumeric characters with a single ``'-'``. :type name: string :parameter version: is converted to a standard version string. Spaces become dots, and all other non-alphanumeric characters (except dots) become dashes, with runs of multiple dashes condensed to a single dash. :type version: string :returns: directory name :rtype: string-_)replacerQrO)clsrversionr$r$r%distinfo_dirnames z!DistributionPath.distinfo_dirnameccsf|js|D]}|VqdS||jjD]}|Vq|jr/|jjD]}|Vq)dSdS)a5 Provides an iterator that looks for distributions and returns :class:`InstalledDistribution` or :class:`EggInfoDistribution` instances for each one of them. :rtype: iterator of :class:`InstalledDistribution` and :class:`EggInfoDistribution` instances N)r8r^r`r6r valuesr5r7r+r$r$r%get_distributionss z"DistributionPath.get_distributionscCsd}|}|js|D] }|j|kr|}|Sq |S|||jjvr0|jj|d}|S|jrA||jjvrA|jj|d}|S)a= Looks for a named distribution on the path. This function only returns the first result found, as no more than one value is expected. If nothing is found, ``None`` is returned. :rtype: :class:`InstalledDistribution`, :class:`EggInfoDistribution` or ``None`` Nr) lowerr8r^r)r`r6rr5r7)r#rresultr,r$r$r%get_distributions"     z!DistributionPath.get_distributionc csd}|dur#z |jd||f}Wnty"td||fw|D]8}t|ds5td|q'|j}|D]$}t |\}}|durP||krO|Vnq:||kr^| |r^|Vnq:q'dS)a Iterates over all distributions to find which distributions provide *name*. If a *version* is provided, it will be used to filter the results. This function only returns the first result found, since no more than one values are expected. If the directory is not found, returns ``None``. :parameter version: a version specifier that indicates the version required, conforming to the format in ``PEP-345`` :type name: string :type version: string N%s (%s)zinvalid name or version: %r, %rprovideszNo "provides": %s) r9matcher ValueErrorrrhhasattrrUrVrmrmatch) r#rrernr,providedpp_namep_verr$r$r%provides_distributions4    z&DistributionPath.provides_distributioncCs(||}|durtd|||S)z5 Return the path to a resource file. Nzno distribution named %r found)rk LookupErrorget_resource_path)r#r relative_pathr,r$r$r% get_file_path!s   zDistributionPath.get_file_pathccsZ|D]%}|j}||vr*||}|dur ||vr||Vq|D]}|Vq$qdS)z Return all of the exported entries in a particular category. :param category: The category to search for entries. :param name: If specified, only entries with that name are returned. N)rhexportsrg)r#categoryrr,r[dvr$r$r%get_exported_entries*s   z%DistributionPath.get_exported_entries)NFr:)r.r/r0r1r&r<r>propertyZ cache_enabledr?r^r` classmethodrfrhrkrvrzrr$r$r$r%rKs   ,  ) rc@seZdZdZdZ dZ ddZeddZeZ eddZ ed d Z d d Z ed dZ eddZeddZeddZeddZddZddZddZddZdS) rz A base class for distributions, whether installed or from indexes. Either way, it must have some metadata, so that's all that's needed for construction. FcCsL||_|j|_|j|_|j|_d|_d|_d|_d|_t |_ i|_ dS)z Initialise an instance. :param metadata: The instance of :class:`Metadata` describing this distribution. N) rErrir)relocatordigestextrascontextrIZ download_urlsdigests)r#rEr$r$r%r&Os  zDistribution.__init__cCs|jjS)zH The source archive download URL for this distribution. )rE source_urlr"r$r$r%r`szDistribution.source_urlcCd|j|jfS)zX A utility property which displays the name and version in parentheses. rlrrer"r$r$r%name_and_versioniszDistribution.name_and_versioncCs.|jj}d|j|jf}||vr|||S)z A set of distribution names and versions provided by this distribution. :return: A set of "name (version)" strings. rl)rErmrrer*)r#plistsr$r$r%rmps  zDistribution.providescCs8|j}td|t||}t|j||j|jdS)Nz%Getting requirements from metadata %r)rrF) rErUrVZtodictgetattrrIget_requirementsrr)r#Zreq_attrmdZreqtsr$r$r%_get_requirements|s   zDistribution._get_requirementscC |dS)N run_requiresrr"r$r$r%r zDistribution.run_requirescCr)N meta_requiresrr"r$r$r%rrzDistribution.meta_requirescCr)Nbuild_requiresrr"r$r$r%rrzDistribution.build_requirescCr)N test_requiresrr"r$r$r%rrzDistribution.test_requirescCr)N dev_requiresrr"r$r$r%rrzDistribution.dev_requiresc Cst|}t|jj}z||j}Wnty,td|| d}||}Ynw|j }d}|j D]}t |\}} ||krBq5z | | }W|StyTYq5w|S)z Say if this instance matches (fulfills) a requirement. :param req: The requirement to match. :rtype req: str :return: True if it matches, else False. +could not read version %r - using name onlyrF)r rrErDrn requirementrrUwarningsplitr)rmrrq) r#reqr[rDrnrrjrsrtrur$r$r%matches_requirements2       z Distribution.matches_requirementcCs(|jr d|j}nd}d|j|j|fS)zC Return a textual representation of this instance, z [%s]r@z)rrre)r#suffixr$r$r%__repr__s zDistribution.__repr__cCs@t|t|ur d}|S|j|jko|j|jko|j|jk}|S)a< See if this distribution is the same as another. :param other: The distribution to compare with. To be equal to one another. distributions must have the same type, name, version and source_url. :return: True if it is the same, else False. F)typerrer)r#otherrjr$r$r%__eq__s  zDistribution.__eq__cCst|jt|jt|jS)zH Compute hash in a way which matches the equality test. )hashrrerr"r$r$r%__hash__szDistribution.__hash__N)r.r/r0r1Zbuild_time_dependency requestedr&rr download_urlrrmrrrrrrrrrrr$r$r$r%r=s:         " rcs0eZdZdZdZdfdd ZdddZZS)rz] This is the base class for installed distributions (whether PEP 376 or legacy). Ncs tt||||_||_dS)a Initialise an instance. :param metadata: An instance of :class:`Metadata` which describes the distribution. This will normally have been initialised from a metadata file in the ``path``. :param path: The path of the ``.dist-info`` or ``.egg-info`` directory for the distribution. :param env: This is normally the :class:`DistributionPath` instance where this distribution was found. N)superrr&r dist_path)r#rEr rF __class__r$r%r&s  z"BaseInstalledDistribution.__init__cCsd|dur|j}|durtj}d}n tt|}d|j}||}t|dd}d||fS)a Get the hash of some data, using a particular hash algorithm, if specified. :param data: The data to be hashed. :type data: bytes :param hasher: The name of a hash implementation, supported by hashlib, or ``None``. Examples of valid values are ``'sha1'``, ``'sha224'``, ``'sha384'``, '``sha256'``, ``'md5'`` and ``'sha512'``. If no hasher is specified, the ``hasher`` attribute of the :class:`InstalledDistribution` instance is used. If the hasher is determined to be ``None``, MD5 is used as the hashing algorithm. :returns: The hash of the data. If a hasher was explicitly specified, the returned hash will be prefixed with the specified hasher followed by '='. :rtype: str Nr@z%s==ascii%s%s) hasherhashlibmd5rrbase64urlsafe_b64encoderstripdecode)r#datarprefixrr$r$r%get_hashs    z"BaseInstalledDistribution.get_hashr:)r.r/r0r1rr&r __classcell__r$r$rr%rs rcseZdZdZdZd'fdd ZddZdd Zd d Ze d d Z ddZ ddZ ddZ ddZd(ddZddZe ddZd(ddZdd Zd!d"Zd#d$Zd%d&ZejZZS))ra  Created with the *path* of the ``.dist-info`` directory provided to the constructor. It reads the metadata contained in ``pydist.json`` when it is instantiated., or uses a passed in Metadata instance (useful for when dry-run mode is being used). sha256Nc szg|_t||_}|durtd||r(|jr(||jjvr(|jj|j}nD|durl| t }|dur:| t }|durC| t }|durOtdt |ft |}t|dd}Wdn1sgwYtt|||||r|jr|j|| d}|du|_tj|d}tj|rt|d}|d} Wdn1swY| |_dSdS) Nzfinder unavailable for %szno %s found in %srArBr top_level.txtrbutf-8)modulesrrJrZror8r6r rErKr r r rRrSrTr rrr&r-rosrQexistsopenreadr splitlines) r#r rErFrZr[r]rsfrrr$r%r&s>          zInstalledDistribution.__init__cCd|j|j|jfS)Nz#rrer r"r$r$r%r= zInstalledDistribution.__repr__cCrNz%s %srr"r$r$r%__str__AzInstalledDistribution.__str__c Csg}|d}t|F}t|d(}|D]}ddtt|dD}||\}}} |||| fqWdn1s?wYWd|SWd|S1sWwY|S)a" Get the list of installed files for the distribution :return: A list of tuples of path, hash and size. Note that hash and size might be ``None`` for some entries. The path is exactly as stored in the file (which is as in PEP 376). rr]cSsg|]}dqSr:r$).0ir$r$r% Ssz6InstalledDistribution._get_records..N)get_distinfo_resourcerRrSrTrrangelenr*) r#resultsr[r]Z record_readerrowmissingr checksumsizer$r$r% _get_recordsDs"       z"InstalledDistribution._get_recordscCsi}|t}|r |}|S)a Return the information exported by this distribution. :return: A dictionary of exports, mapping an export category to a dict of :class:`ExportEntry` instances describing the individual export entries, and keyed by name. )rEXPORTS_FILENAMEr)r#rjr[r$r$r%r{[s  zInstalledDistribution.exportscCsNi}|t}|r%t| }t|}Wd|S1s wY|S)z Read exports data from a file in .ini format. :return: A dictionary of exports, mapping an export category to a list of :class:`ExportEntry` instances describing the individual export entries. N)rrrRrSrTr)r#rjr[r]r$r$r%ris   z"InstalledDistribution.read_exportscCsD|t}t|d}t||WddS1swYdS)a Write a dictionary of exports to a file in .ini format. :param exports: A dictionary of exports, mapping an export category to a list of :class:`ExportEntry` instances describing the individual export entries. wN)get_distinfo_filerrr)r#r{rfrr$r$r%rxs   "z#InstalledDistribution.write_exportsc Cs|d}t|:}t|d%}|D]\}}||kr/|WdWdSqWdn1s:wYWdn1sIwYtd|)aW NOTE: This API may change in the future. Return the absolute path to a resource file with the given relative path. :param relative_path: The path, relative to .dist-info, of the resource of interest. :return: The absolute path where the resource is to be found. rrNz3no resource file with relative path %r is installed)rrRrSrTrKeyError)r#ryr[r]Zresources_readerZrelative destinationr$r$r%rxs   z'InstalledDistribution.get_resource_pathccs|D]}|VqdS)z Iterates over the ``RECORD`` entries and returns a tuple ``(path, hash, size)`` for each line. :returns: iterator of (path, hash, size) N)r)r#rjr$r$r%list_installed_filess z*InstalledDistribution.list_installed_filesFc CsPtj|d}tj|j}||}tj|d}|d}td||r)dSt|q}|D]Q}tj |s=| drBd} } n$dtj |} t |d} | | } Wdn1sawY||sr|ry||rytj||}||| | fq0||rtj||}||ddfWd|S1swY|S)z Writes the ``RECORD`` file, using the ``paths`` iterable passed in. Any existing ``RECORD`` file is silently overwritten. prefix is used to determine when to write absolute paths. r@r creating %sNz.pycz.pyoz%dr)rr rQdirname startswithrrUinforisdirrNgetsizerrrrelpathwriterow) r#pathsrdry_runbaseZbase_under_prefix record_pathwriterr hash_valuerfpr$r$r%write_installed_filess8        z+InstalledDistribution.write_installed_filesc Cs*g}tj|j}|d}|D]\}}}tj|s$tj||}||kr)qtj|s9||dddfqtj |rt tj |}|rW||krW||d||fq|rd|vrf| ddd}nd }t |d } || |} | |kr||d || fWd n1swYq|S)  Checks that the hashes and sizes of the files in ``RECORD`` are matched by the files themselves. Returns a (possibly empty) list of mismatches. Each entry in the mismatch list will be a tuple consisting of the path, 'exists', 'size' or 'hash' according to what didn't match (existence is checked first, then size, then hash), the expected value and the actual value. rrTFr=rrNrr)rr rrrisabsrQrr*isfilestrrrrrr) r# mismatchesrrr rrZ actual_sizerrZ actual_hashr$r$r%check_installed_filess4       z+InstalledDistribution.check_installed_filescCsi}tj|jd}tj|rMtj|ddd}|}Wdn1s)wY|D]}|dd\}}|dkrH| |g |q0|||<q0|S) a A dictionary of shared locations whose keys are in the set 'prefix', 'purelib', 'platlib', 'scripts', 'headers', 'data' and 'namespace'. The corresponding value is the absolute path of that category for this distribution, and takes into account any paths selected by the user at installation time (e.g. via command-line arguments). In the case of the 'namespace' key, this would be a list of absolute paths for the roots of namespace packages in this distribution. The first time this property is accessed, the relevant information is read from the SHARED file in the .dist-info directory. rr[rencodingNrr namespace) rr rQrcodecsrrrrr(r*)r#rj shared_pathrlinesliner)r=r$r$r%shared_locationss  z&InstalledDistribution.shared_locationsc Cstj|jd}td||rdSg}dD]}||}tj||r-|d||fq|ddD] }|d|q4tj |d d d }| d |Wd|S1sZwY|S) aa Write shared location information to the SHARED file in .dist-info. :param paths: A dictionary as described in the documentation for :meth:`shared_locations`. :param dry_run: If True, the action is logged but no file is actually written. :return: The path of the file written to. rrN)rlibheadersscriptsrz%s=%srr$z namespace=%srrr ) rr rQrUrrr*getrrwrite) r#rrrrr)r nsrr$r$r%write_shared_locationss$  z,InstalledDistribution.write_shared_locationscCsF|tvr td||jft|j}|durtd|j||S)N+invalid path for a dist-info file: %r at %rzUnable to get a finder for %s) DIST_FILESrr rrJrK)r#r rZr$r$r%rs  z+InstalledDistribution.get_distinfo_resourcecCs~|tjdkr*|tjdd\}}||jtjdkr*td||j|jf|tvr7td||jftj |j|S)a Returns a path located under the ``.dist-info`` directory. Returns a string representing the path. :parameter path: a ``'/'``-separated path relative to the ``.dist-info`` directory or an absolute path; If *path* is an absolute path and doesn't start with the ``.dist-info`` directory path, a :class:`DistlibException` is raised :type path: str :rtype: str rNz;dist-info file %r does not belong to the %r %s distributionr) rKrseprr rrrer rQ)r#r rfr$r$r%r$s z'InstalledDistribution.get_distinfo_fileccsTtj|j}|D]\}}}tj|stj||}||jr'|Vq dS)z Iterates over the ``RECORD`` entries and returns paths for each line if the path is pointing to a file located in the ``.dist-info`` directory or one of its subdirectories. :returns: iterator of paths N)rr rrrrQr)r#rr rrr$r$r%list_distinfo_filesAs  z)InstalledDistribution.list_distinfo_filescCt|to |j|jkSr:)r_rr r#rr$r$r%rQ  zInstalledDistribution.__eq__)NNF)r.r/r0r1rr&rrrrr{rrrxrrrrrrrr robjectrrr$r$rr%rs."    ##   rcsjeZdZdZdZiZdfdd ZddZdd Zd d Z d d Z ddZ dddZ ddZ ejZZS)raCreated with the *path* of the ``.egg-info`` directory or file provided to the constructor. It reads the metadata contained in the file itself, or if the given path happens to be a directory, the metadata is read from the file ``PKG-INFO`` under that directory.TNcsdd}||_||_|r%|jr%||jjvr%|jj|j}|||j|jn||}|||j|j|r=|jr=|j|t t | |||dS)NcSs||_||_||_dSr:)rrir)re)rnr~r$r$r%set_name_and_versioncs  z:EggInfoDistribution.__init__..set_name_and_version) r rr8r7rErre _get_metadatar-rrr&)r#r rFrrErr$r%r&bs   zEggInfoDistribution.__init__csd}ddfdd}d}}|drytj|rBtj|d}tj|d}t|dd }tj|d } tj|d }|| }nht|} t| d  d } t| dd}z| d} | d d}| d}Wn<t yxd}Yn2w|drtj|rtj|d } || }tj|d}tj|d }t|dd }nt d||r| ||dur|durtj|rt|d} |  d}Wdn1swY|sg}n|}||_|S)NcSsg}|}|D]H}|}|drtd||St|}|s)td|q|jr1td|js;||j qd dd|jD}|d|j |fq|S) zCreate a list of dependencies from a requires.txt file. *data*: the contents of a setuptools-produced requires.txt file. [z.Unexpected line: quitting requirement scan: %rz#Not recognised as a requirement: %rz4extra requirements in requires.txt are not supportedz, css|]}d|VqdS)rNr$)rcr$r$r% szQEggInfoDistribution._get_metadata..parse_requires_data..rl) rstriprrUrr r constraintsr*rrQ)rreqsrrr[Zconsr$r$r%parse_requires_datazs*    z>EggInfoDistribution._get_metadata..parse_requires_datacs^g}z"t|dd}|}WdW|S1swYW|Sty.Y|Sw)zCreate a list of dependencies from a requires.txt file. *req_path*: the path to a setuptools-produced requires.txt file. r[rN)rrrIOError)req_pathrrrr$r%parse_requires_paths  z>EggInfoDistribution._get_metadata..parse_requires_pathrHzEGG-INFOzPKG-INFOrA)r rDz requires.txtrzEGG-INFO/PKG-INFOutf8rBzEGG-INFO/requires.txtzEGG-INFO/top_level.txtrrGz,path must end with .egg-info or .egg, got %rr)rNrr rrQr zipimport zipimporterrget_datarrrZadd_requirementsrrrrr)r#r requiresr Ztl_pathZtl_datars meta_pathrErZzipfrCrrr$rr%rws\             z!EggInfoDistribution._get_metadatacCr)Nz!rr"r$r$r%rrzEggInfoDistribution.__repr__cCrrrr"r$r$r%rrzEggInfoDistribution.__str__cCs`g}tj|jd}tj|r.|D]\}}}||krqtj|s-||dddfq|S)rinstalled-files.txtrTF)rr rQrrr*)r#rrr rbr$r$r%rs   z)EggInfoDistribution.check_installed_filescCsdd}dd}tj|jd}g}tj|rqtj|dddA}|D]6}|}tjtj|j|}tj|sGt d || d rGq#tj |sY| |||||fq#Wd n1sdwY| |d d f|S) z Iterates over the ``installed-files.txt`` entries and returns a tuple ``(path, hash, size)`` for each line. :returns: a list of (path, hash, size) cSs8t|d}z |}W|n|wt|S)Nr)rrcloserr hexdigest)r rcontentr$r$r%_md5s  z6EggInfoDistribution.list_installed_files.._md5cSs t|jSr:)rstatst_size)r r$r$r%_sizes z7EggInfoDistribution.list_installed_files.._sizer'r[rrzNon-existent file: %srN) rr rQrrrrnormpathrUrrNrr*)r#r+r.rrjrrrsr$r$r%rs(      z(EggInfoDistribution.list_installed_filesFccstj|jd}tj|rYd}tj|ddd6}|D]*}|}|dkr)d}q|sFtjtj|j|}||jrF|rC|Vq|VqWddS1sRwYdSdS) a  Iterates over the ``installed-files.txt`` entries and returns paths for each line if the path is pointing to a file located in the ``.egg-info`` directory or one of its subdirectories. :parameter absolute: If *absolute* is ``True``, each returned path is transformed into a local absolute path. Otherwise the raw value from ``installed-files.txt`` is returned. :type absolute: boolean :returns: iterator of paths r'Tr[rrz./FN) rr rQrrrrr/r)r#absoluterskiprrrsr$r$r%r s(  "z'EggInfoDistribution.list_distinfo_filescCrr:)r_rr rr$r$r%r.rzEggInfoDistribution.__eq__r:r)r.r/r0r1rrr&rrrrrr rrrrr$r$rr%rYsZ &rc@s^eZdZdZddZddZdddZd d Zd d ZdddZ dddZ ddZ ddZ dS)DependencyGrapha Represents a dependency graph between distributions. The dependency relationships are stored in an ``adjacency_list`` that maps distributions to a list of ``(other, label)`` tuples where ``other`` is a distribution and the edge is labeled with ``label`` (i.e. the version specifier, if such was provided). Also, for more efficient traversal, for every distribution ``x``, a list of predecessors is kept in ``reverse_list[x]``. An edge from distribution ``a`` to distribution ``b`` means that ``a`` depends on ``b``. If any missing dependencies are found, they are stored in ``missing``, which is a dictionary that maps distributions to a list of requirements that were not provided by any other distributions. cCsi|_i|_i|_dSr:)adjacency_list reverse_listrr"r$r$r%r&Is zDependencyGraph.__init__cCsg|j|<g|j|<dS)zAdd the *distribution* to the graph. :type distribution: :class:`distutils2.database.InstalledDistribution` or :class:`distutils2.database.EggInfoDistribution` N)r3r4)r# distributionr$r$r%add_distributionNs z DependencyGraph.add_distributionNcCs:|j|||f||j|vr|j||dSdS)aAdd an edge from distribution *x* to distribution *y* with the given *label*. :type x: :class:`distutils2.database.InstalledDistribution` or :class:`distutils2.database.EggInfoDistribution` :type y: :class:`distutils2.database.InstalledDistribution` or :class:`distutils2.database.EggInfoDistribution` :type label: ``str`` or ``None`` N)r3r*r4)r#xylabelr$r$r%add_edgeXs zDependencyGraph.add_edgecCs&td|||j|g|dS)a Add a missing *requirement* for the given *distribution*. :type distribution: :class:`distutils2.database.InstalledDistribution` or :class:`distutils2.database.EggInfoDistribution` :type requirement: ``str`` z %s missing %rN)rUrVrr(r*)r#r5rr$r$r% add_missinggszDependencyGraph.add_missingcCsd|j|jfSrrr+r$r$r% _repr_distrrzDependencyGraph._repr_distrcCs||g}|j|D]4\}}||}|durd||f}|d|t||||d}|d}||ddq d|S)zPrints only a subgraphNz%s [%s]z rr)r<r3r*r repr_noderextendrQ)r#r,leveloutputrr9Z suboutputsubsr$r$r%r=us     zDependencyGraph.repr_nodeTcCsg}|d|jD]4\}}t|dkr|s|||D] \}}|dur4|d|j|j|fq|d|j|jfqq |sot|dkro|d|d|d|D]}|d |j|d qZ|d |d dS) a9Writes a DOT output for the graph to the provided file *f*. If *skip_disconnected* is set to ``True``, then all distributions that are not dependent on any other distribution are skipped. :type f: has to support ``file``-like operations :type skip_disconnected: ``bool`` zdigraph dependencies { rNz"%s" -> "%s" [label="%s"] z "%s" -> "%s" zsubgraph disconnected { zlabel = "Disconnected" zbgcolor = red z"%s"rz} )rr3itemsrr*r)r#rZskip_disconnected disconnectedr,adjsrr9r$r$r%to_dots*         zDependencyGraph.to_dotcsg}i}|jD] \}}|dd||<q gt|ddD]\}}|s1|||=q#s5n%|D]\}}fdd|D||<q9tdddD|q|t|fS)aa Perform a topological sort of the graph. :return: A tuple, the first element of which is a topologically sorted list of distributions, and the second element of which is a list of distributions that cannot be sorted because they have circular dependencies and so form a cycle. NTcs g|] \}}|vr||fqSr$r$)rr}r[ to_remover$r%rs z4DependencyGraph.topological_sort..zMoving to result: %scSsg|] }d|j|jfqS)rlr)rr}r$r$r%rs)r3rBlistr*rUrVr>keys)r#rjalistkr~r$rFr%topological_sorts*   z DependencyGraph.topological_sortcCs2g}|jD] \}}|||qd|S)zRepresentation of the graphr)r3rBr*r=rQ)r#r@r,rDr$r$r%rs zDependencyGraph.__repr__r:)r)T) r.r/r0r1r&r6r:r;r<r=rErLrr$r$r$r%r29s    r2r2c CsLt|}t}i}|D]&}|||jD]}t|\}}td|||||g||fqq |D]o}|j |j B|j B|j B}|D]^} z| | } Wntygtd| | d}| |} Ynw| j}d} ||vr||D]$\}} z| |} Wn tyd} Ynw| r||| | d} nqu| s||| qDq4|S)a6Makes a dependency graph from the given distributions. :parameter dists: a list of distributions :type dists: list of :class:`distutils2.database.InstalledDistribution` and :class:`distutils2.database.EggInfoDistribution` instances :rtype: a :class:`DependencyGraph` instance zAdd to provided: %s, %s, %srrFT)rr2r6rmrrUrVr(r*rrrrrnrrrr)rqr:r;)distsrDgraphrrr,rsrrer%rrnmatchedproviderrqr$r$r% make_graphsZ        rQcCsx||vr td|jt|}|g}|j|}|r5|}|||j|D] }||vr2||q'|s|d|S)zRecursively generate a list of distributions from *dists* that are dependent on *dist*. :param dists: a list of distributions :param dist: a distribution, member of *dists* for which we are interested 1given distribution %r is not a member of the listr)rrrQr4popr*)rMr,rNdeptodor}succr$r$r%get_dependent_distss"    rWcCsp||vr td|jt|}g}|j|}|r6|d}|||j|D] }||vr3||q(|s|S)zRecursively generate a list of distributions from *dists* that are required by *dist*. :param dists: a list of distributions :param dist: a distribution, member of *dists* for which we are interested rRr)rrrQr3rSr*)rMr,rNrrUr}predr$r$r%get_required_distss     rYcKs8|dd}tdi|}||_||_|pd|_t|S)zO A convenience method for making a dist given just a name and version. summaryzPlaceholder for summaryNr$)rSr rrerZr)rrekwargsrZrr$r$r% make_dist2s  r\)r2)7r1 __future__rrrrRrloggingrrPr3r"r@rrcompatrrerrrEr r r r utilr rrrrrr__all__ getLoggerr.rUrZCOMMANDS_FILENAMEr rOrrrrrrrrWrXr2rQrWrYr\r$r$r$r%sV  $ s7J] 6