o (c<1@sddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZd d lmZmZzddlmZWn e ykdZYnwdeddfddZ!dededdeeddfddZ"dZ#dS)N)log)clean)ListSetTupleTypecast) build_ext)install) install_lib)install_scripts)sdist) Distribution)Literal)RustBin RustExtension) bdist_wheeldistreturncshttt|jdt}|j |j|j ddg ddd<g}|j D] }| | q0|rc|j sEd|i|_ n|j d}|r\|D] }||vrZ| |qOn|}||j d<G fddd|}||jd<ttt |jd t j d Gfd d d }||jd <ttt|jd tGfddd}||jd <ttt|jdtGfddd} | |jd<ttt|jdtGfddd} | |jd<ttt|jdtGfddd} | |jd<tdur2ttt|jdt} | j d Gfddd| } | |jd<dSdS)Nr ) vendor-cratesNzvendor Rust crates)no-vendor-cratesNz?don't vendor Rust crates.[default; enable with --vendor-crates]rrconsole_scriptscs8eZdZZZZdfdd Zdfdd ZZS)z0add_rust_extension..sdist_rust_extensionrNcstd|_dS)Nr)superinitialize_options vendor_cratesself __class__Busr/lib/python3.10/site-packages/setuptools_rust/setuptools_ext.pyrBs  zCadd_rust_extension..sdist_rust_extension.initialize_optionscs|jrg}t}|jjD]}||j|jr||jq |r|j}t j |d}| |t j |d}t j |d}ddg}|rL| t ||ddD] } |d|| qR| d|d|gtj|d d t} d} dt jd dt jdfD] } | |jjvr| } nq| rt j || }t| d } | | d 7} Wdn1swYt|d } | | Wdn1swYtdS)Nz.cargoz config.tomlvendorZcargorz--syncz--manifest-pathrT)checkconfigrb wb)rset distributionrust_extensionsappendpathcargo_manifest_argsupdate get_fullnameosjoinmkpathextendsorted subprocessrun_CARGO_VENDOR_CONFIGsepfilelistfilesopenreadwritermake_distribution)rZmanifest_pathsr-extbase_dirZdot_cargo_pathZcargo_config_pathZ vendor_pathcommand extra_pathZ cargo_configZexisting_cargo_configfilenamefrr r!r>FsT             zBadd_rust_extension..sdist_rust_extension.make_distributionrN) __name__ __module__ __qualname__ user_optionsboolean_options negative_optrr> __classcell__r )sdist_boolean_optionssdist_negative_opt sdist_optionsrr!sdist_rust_extension=s rPr )targetNzBuild for the target triplecs0eZdZZdfdd Zdfdd ZZS)z4add_rust_extension..build_ext_rust_extensionrNcttd|_dSNZCARGO_BUILD_TARGETrrr0getenvrQrrr r!r zGadd_rust_extension..build_ext_rust_extension.initialize_optionscsr|jjr2td|d}|j|_|j|_|j|_|j di}| dp*|j }||_ |  |dS)Nzrunning build_rust build_rustrz plat-name) r)r*rinfoget_finalized_commandZinplacerQverboseget_cmdline_optionsget plat_namer6)rrWoptionsr])build_ext_base_classr r!r6s  z8add_rust_extension..build_ext_rust_extension.runrE)rFrGrHrIrr6rLr )r_build_ext_optionsrr!build_ext_rust_extensionsrarceZdZdfdd ZdS)z0add_rust_extension..clean_rust_extensionrNcs"||js|ddSdS)NZ clean_rust)r6dry_run run_commandrclean_base_classr r!r6s z4add_rust_extension..clean_rust_extension.runrErFrGrHr6r rer r!clean_rust_extensionrhr crb)z2add_rust_extension..install_rust_extensionrNcsB|d}|jjrtdd|jjD}|r|ddSdS)NFcss|]}t|tVqdSN) isinstancer).0r?r r r! s  zIadd_rust_extension..install_rust_extension.run..r )r6r)r*anyrd)rZinstall_rustbininstall_base_classr r!r6s z6add_rust_extension..install_rust_extension.runrErgr ror r!install_rust_extensionrirqr cs"eZdZdeeffdd ZdS)z6add_rust_extension..install_lib_rust_extensionrcs|}|d}tj|j|jd}|jjr>t d}|jjD]}|j }|dur.||7}t |t r=|tj||q!|S)NrWscriptsEXE)get_exclusionsrYr0r,r1 install_dirdata_dirr)r* sysconfigget_config_varnamerkradd)rZ exclusionsrW scripts_pathexer?Zexecutable_nameinstall_lib_base_classr r!rts      zEadd_rust_extension..install_lib_rust_extension.get_exclusionsN)rFrGrHrstrrtr r}r r!install_lib_rust_extensionsrr crb)z:add_rust_extension..install_scripts_rust_extensionrNc s||d}|d}tj|j|jd}tj|rWt|D]4}tj||}tj |rVt tj|d}|j || ddWdn1sQwYq$dSdS)Nr rWrrr%b)mode) r6rYr0r,r1 build_librvisdirlistdirisfiler;Z write_scriptr<)rr rWr{fileZ script_pathZ script_readerinstall_scripts_base_classr r!r6s      z>add_rust_extension..install_scripts_rust_extension.runrErgr rr r!install_scripts_rust_extensionrirrcs>eZdZZdfdd Zdeeeefffdd ZZS)z6add_rust_extension..bdist_wheel_rust_extensionrNcrRrSrTrrr r!rrVzIadd_rust_extension..bdist_wheel_rust_extension.initialize_optionscst\}}}td}d}|jdr|rd|vod|v}|rJ|drJddlm}td }|durAd | d d d }||j d |}|||fS)NZ ARCHFLAGSFzmacosx-x86_64arm64Zmacosx_r)calculate_macosx_platform_tagMACOSX_DEPLOYMENT_TARGET._rzmacosx-{}-universal2) rget_tagr0rUr] startswithZwheel.macosx_libfilerr1splitZ bdist_dirformat)rpythonabiplatZ arch_flags universal2rZ macos_targetrr r!rs     z>add_rust_extension..bdist_wheel_rust_extension.get_tagrE) rFrGrHrIrrrrrLr )bdist_wheel_optionsrr!bdist_wheel_rust_extensions$r)rrr cmdclassr\rIcopyrJrKr3r+r* entry_pointsr rr r r r)rZsdist_base_classrr?Z ep_scriptsscriptrPrarhrqrrZbdist_wheel_base_classrr ) rr_r`rfrpr~rrMrNrOr!add_rust_extensionsx          F          rattrr*valuecsB|dksJt|dk|jfdd|_rt|dSdS)Nr*rcs pSrjr r Zhas_rust_extensionsZorig_has_ext_modulesr r!s z!rust_extensions..)lenhas_ext_modulesr)rrrr rr!r*s   sm [source.crates-io] replace-with = "vendored-sources" [source.vendored-sources] directory = ".cargo/vendor" )$r0r5rw distutilsrZdistutils.command.cleanrtypingrrrrrZsetuptools.command.build_extr Zsetuptools.command.installr Zsetuptools.command.install_libr Z"setuptools.command.install_scriptsr Zsetuptools.command.sdistr setuptools.distrtyping_extensionsr extensionrrZwheel.bdist_wheelr ImportErrorrr*r7r r r r!s>          x