U 1_+@sddlZddlZddlZddlZddlZddlmZddlmZddlm Z ej dkrhddl Z ddZ nddlZd dZ ej d krejZnGd d d ejZGd ddeZdZddZdaddZddZdddZddZddZdS)N)__version_verifier_modules__) ffiplatform)VerificationError)rcCstjjddSN) importlib machineryEXTENSION_SUFFIXESr r 1/tmp/pip-unpacked-wheel-9t_xacz3/cffi/verifier.py_extension_suffixes sr cCsddtDS)NcSs g|]\}}}|tjkr|qSr )imp C_EXTENSION).0suffix_typer r r s z'_extension_suffixes..)rZ get_suffixesr r r r r srcseZdZfddZZS)NativeIOcs(t|tr|d}tt||dS)Nascii) isinstanceunicodeencodesuperrwrite)selfs __class__r r rs  zNativeIO.write)__name__ __module__ __qualname__r __classcell__r r rr rsrc@szeZdZdddZd ddZd d Zd d Zd dZddZddZ ddZ ddZ ddZ d!ddZ ddZddZdS)"VerifierNF.cc Ksv|jjrtd|jjf||_||_|s4t| } t||} | ||_|j | | |_ | | | |_ |rz|rxt dndtjddt|| g|j}tjdkr|d}tt|dddd @}|d d }tt|d ddd @}|d d }d||jj||f}td}|p0t|_tj|j|||_ tj|j|||_!||_"d|_#d|_$dS)NzKfeature not supported with ffi.verify(), but only with ffi.set_source(): %sz)can't specify both 'modulename' and 'tag'rrzutf-8rl0xLr0z_cffi_%s_%s%s%sF)%_parserZ_uses_new_featurerffipreamblerflatten_locate_engine_class_vengineZpatch_extension_kwdsflagsmake_relative_tokwds TypeErrorjoinsysversionrZ _cdefsources version_inforhexbinasciicrc32lstriprstripZ _class_key_get_so_suffixes_caller_dir_pycachetmpdirospathsourcefilenamemodulefilename ext_package _has_source _has_module)rr.r/rB modulenamerGtagforce_generic_engineZsource_extensionr3 relative_tor5Zflattened_kwdsZ vengine_classkeyZk1Zk2rr r r __init__ sR         zVerifier.__init__c Cs8|jj&|jr |dkr td||W5QRXdS)zkWrite the C source code. It is produced in 'self.sourcefilename', which can be tweaked beforehand.Nzsource code already written)r._lockrHr _write_sourcerfiler r r write_sourceHs  zVerifier.write_sourcec Cs<|jj*|jrtd|js&||W5QRXdS)zWrite the C source code (if not done already) and compile it. This produces a dynamic link library in 'self.modulefilename'.zmodule already compiledN)r.rPrIrrHrQ_compile_modulerr r r compile_moduleQs  zVerifier.compile_modulec CsR|jj@|js4||js4|js,|||W5QRSQRXdS)a"Get a C module from this Verifier instance. Returns an instance of a FFILibrary class that behaves like the objects returned by ffi.dlopen(), but that delegates all operations to the C module. If necessary, the C code is written and compiled first. N)r.rPrI_locate_modulerHrQrU _load_libraryrVr r r load_library[s zVerifier.load_librarycCsBtj|j}|ddd}|dr>ttdr>|dd}|S)N.rrZ_dgettotalrefcount)rCrDbasenamerFsplitendswithhasattrr8)rr^r r r get_module_nameks  zVerifier.get_module_namec CsVt|js0|jj|js&|W5QRXt|j}|}tj ||f|j Sr) rZ_hack_at_distutilsrHr.rPrQZmaybe_relative_pathrErb get_extensionr5)rZ sourcenamemodnamer r r rcvs  zVerifier.get_extensioncCs|jjSr)r2Z_gen_python_modulerVr r r generates_python_modulesz Verifier.generates_python_modulecsz|rvtj|rvtj||}tjD]J}||kr*||}t|ttfsZt d|ffdd|D}|||<q*|S)Nz&keyword '%s' should be a list or tuplecsg|]}tj|qSr )rCrDr7)rfndirnamer r rsz-Verifier.make_relative_to..) rCrDrhcopyrZLIST_OF_FILE_NAMESrlisttupler6)rr5rMrNlstr rgr r4s   zVerifier.make_relative_tocCstj|jsv|jrJzt|jdddg}Wntk r@YdSX|j}nd}|j | |t }|dkrpdS||_|j d|_ dS)N__doc__T)rCrDisfilerFrG __import__ ImportError__path__r2 find_modulerbr@Z collect_typesrI)rpkgrDfilenamer r r rXs  zVerifier._locate_modulecCs$||j_z|jW5|j`XdSr)r2_fZwrite_source_to_frRr r r _write_source_toszVerifier._write_source_toc Cs|dk r||nt}|||}tj|jrbt|jd}||k }W5QRXnd}|rt |jt|jd}| |W5QRXd|_ dS)NrTw) rvrgetvaluerCrDexistsrEopenread _ensure_dirrrH)rrSfZ source_datafpZ needs_writtenr r r rQs   zVerifier._write_sourcecCsptj|j}t||}zt||j}Wnt k rHd}YnX|sft |jt ||jd|_ dS)NFT)rCrDrhrErcompilercsamefilerFOSErrorr}shutilmoverI)rrBZoutputfilenameZsamer r r rUs  zVerifier._compile_modulecCs0|js t|jdk r"|j|jS|jSdSr)rIAssertionErrorr3r2rZrVr r r rYs  zVerifier._load_library)NNNr&Fr'NN)N)N)r!r"r#rOrTrWrZrbrcrer4rXrvrQrUrYr r r r r%s( (      r%FcCs|trd}|sPdtjkrd}n4z ddl}Wntk r@d}YnX|j|k rPd}|rfddlm}|jSddlm }|j SdS)NTZ__pypy__r?r) vengine_gen) vengine_cpy) _FORCE_GENERIC_ENGINEr8builtin_module_names _cffi_backendrpZ_backendr&rZVGenericEnginerZVCPythonEngine)r.rLrrrr r r r1s       r1cCsHtrtStjd}|r|Stdjj}tj tj tj |dS)NZ CFFI_TMPDIRr) __pycache__) _TMPDIRrCenvirongetr8 _getframef_code co_filenamerDabspathr7rh)resultrtr r r rAs rAcCs|adS)z:Set the temporary directory to use instead of __pycache__.N)rrgr r r set_tmpdirsrc Cs$|pt}zt|}Wntk r.YdSX|r:d}ntd}|D]\}|drL||sz|drLzttj ||WqLtk rYqLXqLtj |dg}|D]`}zDt|D]4}tj ||}tj |r| |qt|qWqtk rYqXqdS)z}Clean up the temporary directory by removing all files in it called `_cffi_*.{c,so}` as well as the `build` subdirectory.Nr'rZ_cffi_build) rArClistdirrr@lower startswithr`unlinkrDr7isdirappend)rBZkeep_sofilelistrrfZ clean_dirdirr r r cleanup_tmpdirs8     rcCs&t}|s"tjdkrdg}ndg}|S)Nwin32z.pydz.so)r r8platform)suffixesr r r r@$s  r@cCs*tj|}|r&tj|s&t|dSr)rCrDrhrmakedirs)rtrhr r r r}/s r})NF)r8rCr<rior&rrerrorrr:Zimportlib.machineryrr rStringIOrBytesIOobjectr%rr1rrArrr@r}r r r r s*(      ?