a ljQ@sddlZddlZddlZddlZddlZddlZzddlmZWney^ddl mZYn0ddl m Z ddl m Z mZmZmZmZmZddlmZmZeeZdZdZGd d d eZdS) N)Thread)DistlibException)HTTPBasicAuthHandlerRequestHTTPPasswordMgrurlparse build_opener string_types)zip_dir ServerProxyzhttps://pypi.org/pypipypic@seZdZdZdZd*ddZddZdd Zd d Zd d Z ddZ ddZ d+ddZ d,ddZ d-ddZd.ddZddZd/ddZd0d d!Zd1d"d#Zd$d%Zd&d'Zd2d(d)ZdS)3 PackageIndexzc This class represents a package index compatible with PyPI, the Python Package Index. s.----------ThIs_Is_tHe_distlib_index_bouNdaRY_$Nc Cs|pt|_|t|j\}}}}}}|s<|s<|s<|dvrJtd|jd|_d|_d|_d|_t t j dZ}dD]D} z,t j | dg||d} | dkr| |_WqWqttyYqt0qtWdn1s0YdS) z Initialise an instance. :param url: The URL of the index. If not specified, the URL for PyPI is used. )httphttpszinvalid repository: %sNw)gpgZgpg2z --versionstdoutstderrr) DEFAULT_INDEXurlread_configurationrrpassword_handler ssl_verifierrgpg_homeopenosdevnull subprocessZ check_callOSError) selfrschemenetlocpathparamsqueryfragZsinksrcr*=/usr/lib/python3.9/site-packages/pip/_vendor/distlib/index.py__init__$s(    zPackageIndex.__init__cCsddlm}|S)zs Get the distutils command for interacting with PyPI configurations. :return: the command. r)_get_pypirc_command)utilr-)r!cmdr*r*r+r-As z PackageIndex._get_pypirc_commandcCsNddlm}||}|d|_|d|_|dd|_|d|j|_dS) z Read the PyPI access configuration as supported by distutils. This populates ``username``, ``password``, ``realm`` and ``url`` attributes from the configuration. r) _load_pypircusernamepasswordrealmr Z repositoryN)r.r0getr1r2r3r)r!r0Zcfgr*r*r+rIs    zPackageIndex.read_configurationcCs |ddlm}||dS)z Save the PyPI access configuration. You must have set ``username`` and ``password`` attributes before calling this method. r) _store_pypircN)check_credentialsr.r5)r!r5r*r*r+save_configurationVs zPackageIndex.save_configurationcCs\|jdus|jdurtdt}t|j\}}}}}}||j||j|jt||_ dS)zp Check that ``username`` and ``password`` have been set, and raise an exception if not. Nz!username and password must be set) r1r2rrrrZ add_passwordr3rr)r!Zpm_r#r*r*r+r6_s zPackageIndex.check_credentialscCs\|||}d|d<||g}||}d|d<||g}||S)aq Register a distribution on PyPI, using the provided metadata. :param metadata: A :class:`Metadata` instance defining at least a name and version number for the distribution to be registered. :return: The HTTP response received from PyPI upon submission of the request. Zverify:actionZsubmit)r6validatetodictencode_requestitems send_request)r!metadatadrequestZresponser*r*r+registerks  zPackageIndex.registercCsF|}|sq:|d}||td||fq|dS)ar Thread runner for reading lines of from a subprocess into a buffer. :param name: The logical name of the stream (used for logging only). :param stream: The stream to read from. This will typically a pipe connected to the output stream of a subprocess. :param outbuf: The list to append the read lines to. utf-8z%s: %sN)readlinedecoderstripappendloggerdebugclose)r!namestreamZoutbufr(r*r*r+_readers  zPackageIndex._readerc Cs|jdddg}|dur|j}|r.|d|g|durD|gdt}tj|tj|d}|dd d |d ||gt d d |||fS)a Return a suitable command for signing a file. :param filename: The pathname to the file to be signed. :param signer: The identifier of the signer of the file. :param sign_password: The passphrase for the signer's private key used for signing. :param keystore: The path to a directory which contains the keys used in verification. If not specified, the instance's ``gpg_home`` attribute is used instead. :return: The signing command as a list suitable to be passed to :class:`subprocess.Popen`. --status-fd2--no-ttyN --homedir)z--batchz--passphrase-fd0z.ascz --detach-signz--armorz --local-userz--output invoking: %s ) rrextendtempfileZmkdtemprr$joinbasenamerHrI)r!filenamesigner sign_passwordkeystorer/ZtdZsfr*r*r+get_sign_commands zPackageIndex.get_sign_commandc Cstjtjd}|dur tj|d<g}g}tj|fi|}t|jd|j|fd}|t|jd|j|fd}||dur|j ||j | | | |j ||fS)a Run a command in a child process , passing it any input data specified. :param cmd: The command to run. :param input_data: If specified, this must be a byte string containing data to be sent to the child process. :return: A tuple consisting of the subprocess' exit code, a list of lines read from the subprocess' ``stdout``, and a list of lines read from the subprocess' ``stderr``. rNstdinr)targetargsr)rPIPEPopenrrMrstartrr^writerJwaitrW returncode) r!r/Z input_datakwargsrrpt1t2r*r*r+ run_commands&    zPackageIndex.run_commandc CsD|||||\}}|||d\}}} |dkr@td||S)aR Sign a file. :param filename: The pathname to the file to be signed. :param signer: The identifier of the signer of the file. :param sign_password: The passphrase for the signer's private key used for signing. :param keystore: The path to a directory which contains the keys used in signing. If not specified, the instance's ``gpg_home`` attribute is used instead. :return: The absolute pathname of the file where the signature is stored. rCrz&sign command failed with error code %s)r]rkencoder) r!rYrZr[r\r/sig_filer)rrr*r*r+ sign_files  zPackageIndex.sign_filesdistsourcec CsR|tj|s td|||}d} |rZ|jsJt dn| ||||} t |d} | } Wdn1s0Yt | } t | } |dd||| | ddtj|| fg}| r8t | d} | }Wdn1s0Y|d tj| |fttj| |||}||S) a Upload a release file to the index. :param metadata: A :class:`Metadata` instance defining at least a name and version number for the file to be uploaded. :param filename: The pathname of the file to be uploaded. :param signer: The identifier of the signer of the file. :param sign_password: The passphrase for the signer's private key used for signing. :param filetype: The type of the file being uploaded. This is the distutils command which produced that file, e.g. ``sdist`` or ``bdist_wheel``. :param pyversion: The version of Python which the release relates to. For code compatible with any Python, this would be ``source``, otherwise it would be e.g. ``3.2``. :param keystore: The path to a directory which contains the keys used in signing. If not specified, the instance's ``gpg_home`` attribute is used instead. :return: The HTTP response received from PyPI upon submission of the request. z not found: %sNz)no signing program available - not signedrbZ file_upload1)r9Zprotocol_versionfiletype pyversion md5_digest sha256_digestcontentZ gpg_signature)r6rr$existsrr:r;rrHZwarningrnrreadhashlibmd5 hexdigestZsha256updaterXrGshutilrmtreedirnamer<r=r>)r!r?rYrZr[rsrtr\r@rmfZ file_datarurvfilesZsig_datarAr*r*r+ upload_filesD     & (zPackageIndex.upload_filec Cs|tj|s td|tj|d}tj|sFtd|||j|j }}t | }dd|fd|fg}d||fg}| ||} | | S)a2 Upload documentation to the index. :param metadata: A :class:`Metadata` instance defining at least a name and version number for the documentation to be uploaded. :param doc_dir: The pathname of the directory which contains the documentation. This should be the directory that contains the ``index.html`` for the documentation. :return: The HTTP response received from PyPI upon submission of the request. znot a directory: %rz index.htmlz not found: %r)r9Z doc_uploadrKversionrw)r6rr$isdirrrWrxr:rKrr getvaluer<r>) r!r?Zdoc_dirfnrKrZzip_datafieldsrrAr*r*r+upload_documentation!s         z!PackageIndex.upload_documentationcCsT|jdddg}|dur|j}|r.|d|g|d||gtdd||S) a| Return a suitable command for verifying a file. :param signature_filename: The pathname to the file containing the signature. :param data_filename: The pathname to the file containing the signed data. :param keystore: The path to a directory which contains the keys used in verification. If not specified, the instance's ``gpg_home`` attribute is used instead. :return: The verifying command as a list suitable to be passed to :class:`subprocess.Popen`. rNrOrPNrQz--verifyrSrT)rrrUrHrIrW)r!signature_filename data_filenamer\r/r*r*r+get_verify_command=szPackageIndex.get_verify_commandcCsH|jstd||||}||\}}}|dvr@td||dkS)a6 Verify a signature for a file. :param signature_filename: The pathname to the file containing the signature. :param data_filename: The pathname to the file containing the signed data. :param keystore: The path to a directory which contains the keys used in verification. If not specified, the instance's ``gpg_home`` attribute is used instead. :return: True if the signature was verified, else False. z0verification unavailable because gpg unavailable)rrz(verify command failed with error code %sr)rrrrk)r!rrr\r/r)rrr*r*r+verify_signatureUszPackageIndex.verify_signaturec Cs|durd}tdn6t|ttfr0|\}}nd}tt|}td|t|d}|t |}z| } d} d} d} d} d | vrt | d } |r|| | | | | }|sq| t |7} |||r||| d 7} |r|| | | qW|n |0Wdn1s 0Y| dkrN| | krNtd | | f|r|}||krztd ||||ftd|dS)a This is a convenience method for downloading a file from an URL. Normally, this will be a file from the index, though currently no check is made for this (i.e. a file can be downloaded from anywhere). The method is just like the :func:`urlretrieve` function in the standard library, except that it allows digest computation to be done during download and checking that the downloaded data matched any expected value. :param url: The URL of the file to be downloaded (assumed to be available via an HTTP GET request). :param destfile: The pathname where the downloaded file is to be saved. :param digest: If specified, this must be a (hasher, value) tuple, where hasher is the algorithm used (e.g. ``'md5'``) and ``value`` is the expected value. :param reporthook: The same as for :func:`urlretrieve` in the standard library. NzNo digest specifiedr{zDigest specified: %swbi rzcontent-lengthzContent-Lengthrz1retrieval incomplete: got only %d out of %d bytesz.%s digest mismatch for %s: expected %s, got %szDigest verified: %s)rHrI isinstancelisttuplegetattrrzrr>rinfointrylenrdr}rJrr|)r!rZdestfileZdigestZ reporthookZdigesterZhasherZdfpZsfpheaders blocksizesizeryZblocknumblockactualr*r*r+ download_filens\          4 zPackageIndex.download_filecCs:g}|jr||j|jr(||jt|}||S)z Send a standard library :class:`Request` to PyPI and return its response. :param req: The request to send. :return: The HTTP response from PyPI (a standard library HTTPResponse). )rrGrr r)r!Zreqhandlersopenerr*r*r+r>s  zPackageIndex.send_requestc Csg}|j}|D]L\}}t|ttfs*|g}|D]*}|d|d|dd|dfq.q|D].\}} } |d|d|| fdd| fq`|d|ddfd|} d|} | tt| d} t |j | | S) a& Encode fields and files for posting to an HTTP server. :param fields: The fields to send as a list of (fieldname, value) tuples. :param files: The files to send as a list of (fieldname, filename, file_bytes) tuple. s--z)Content-Disposition: form-data; name="%s"rCz8Content-Disposition: form-data; name="%s"; filename="%s"s smultipart/form-data; boundary=)z Content-typezContent-length) boundaryrrrrUrlrWstrrrr)r!rrpartsrkvaluesvkeyrYvaluebodyZctrr*r*r+r<sD     zPackageIndex.encode_requestcCsNt|trd|i}t|jdd}z|||p.dW|dS|d0dS)NrKg@)timeoutandrJ)rr r rsearch)r!ZtermsoperatorZ rpc_proxyr*r*r+rs  zPackageIndex.search)N)N)N)N)NNrorpN)N)N)NN)N)__name__ __module__ __qualname____doc__rr,r-rr7r6rBrMr]rkrnrrrrrr>r<rr*r*r*r+rs2      #  9   M+r)rzZloggingrr~rrV threadingr ImportErrorZdummy_threadingrcompatrrrrr r r.r r Z getLoggerrrHrZ DEFAULT_REALMobjectrr*r*r*r+s