U e5dt@sddgZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m ZddlZdZGdddZdd Zd d ZGd ddZd dZddZdddZddZGdddZddZedkredS)TraceCoverageResultsN) monotonicz#pragma NO COVERc@seZdZdddZddZdS)_IgnoreNcCs:|s tnt||_|sgn dd|D|_ddi|_dS)NcSsg|]}tj|qS)ospathnormpath.0drr/usr/lib64/python3.8/trace.py Hsz$_Ignore.__init__..z)set_mods_dirs_ignore)selfmodulesdirsrrr __init__Fs z_Ignore.__init__cCs||jkr|j|S||jkr,d|j|<dS|jD]"}||dr2d|j|<dSq2|dkrld|j|<dS|jD]$}||tjrrd|j|<dSqrd|j|<dS)Nr.r)rr startswithrrsep)rfilename modulenamemodr rrr namesLs$          z _Ignore.names)NN)__name__ __module__ __qualname__rrrrrr rEs rcCs tj|}tj|\}}|SN)rrbasenamesplitext)rbaserextrrr _modnamews r'cCstj|}d}tjD]@}tj|}||r|t|tjkrt|t|kr|}q|rr|t|dd}n|}tj|\}}|tjd}tj r|tj d}tj |\}}| dS)Nrr) rrnormcasesysrlenr splitdrivereplacealtsepr$lstrip)rZ comparepathZlongestdirr%Zdriverr&rrr _fullmodname~s    r1c@s:eZdZdddZddZddZdd d Zdd d ZdS)rNc Cs||_|jdkri|_|j|_||_|jdkr8i|_|j|_||_|jdkrZi|_|j|_||_||_|jrz@t|jd}t |\}}}W5QRX| | |||Wn@t t tfk r}ztd|j|ftjdW5d}~XYnXdS)NrbzSkipping counts file %r: %sfile)countscopyZcounter calledfuncscallersinfileoutfileopenpickleloadupdate __class__OSErrorEOFError ValueErrorprintr*stderr)rr5r7r9r8r:ferrrrr rs2      zCoverageResults.__init__cCs|do|dS)N<>)rendswith)rrrrr is_ignored_filenamesz#CoverageResults.is_ignored_filenamec Csn|j}|j}|j}|j}|j}|j}|D]}||d||||<q(|D] }d||<qJ|D] }d||<q\dSNrr)r5r7r8get) rotherr5r7r8Z other_countsZother_calledfuncsZ other_callerskeyrrr r>s zCoverageResults.updateTFc" Cs|jr@ttd|j}t|D]\}}}td|||fq"|jrttdd}} t|jD]h\\} } } \} }}| |krttd| d| }d} | | kr| | krtd| | } td| | ||fqfi}|jD].\}}||i}||<|j||f||<qi}|D]\}}||r0q|drH|dd }|dkrpt j t j |}t |}n$|}t j |st |t|}|rt|}ni}t|}t j ||d }t|d }t|j\}}W5QRX||||||\}}|r|rtd ||}||||f||<q|rt|rttd t|D]&}||\}}}}td||qL|jrz6t|jd } t|j|j|jf| dW5QRXWn6tk r}!ztd|!tj dW5d}!~!XYnXdS)Nzfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r(z***z -->z %s.%s -> %s.%sz.pycz.coverr2dzlines cov% module (path)z%5d %3d%% %s (%s)wbrz"Can't save counts files because %sr3)!r7rCsortedr8r5rLitemsrJrIrrdirnameabspathr'existsmakedirsr1_find_executable_linenos linecachegetlinesjoinr;tokenizedetect_encodingreadlinewrite_results_fileintr:r<dumpr@r*rD)"rZ show_missingsummarycoverdirZcallsrrfuncnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr0lnotabsourceZ coverpathfpencoding_n_hitsn_linesZpercentmrErFrrr write_resultss               zCoverageResults.write_resultsc Cszt|d|d}Wn>tk rP}z td||ftjdWYdSd}~XYnXd}d} |t|dD]r\} } | |kr|d|| | d7} |d7}n.| |krt| kr|d |d7}n |d || d qjW5QRX| |fS) Nwrlz3trace: Could not open %r for writing: %s - skippingr3)rrrrz%5d: z>>>>>> z ) r;r@rCr*rD enumeratewritePRAGMA_NOCOVER expandtabs) rrlinesrirgrlr:rFrornrflinerrr r_)s.    z"CoverageResults.write_results_file)NNNNN)TFN)N)rr r!rrJr>rqr_rrrr rs  \cCs,i}t|D]\}}||krd||<q|S)Nr)disZfindlinestarts)codestrslinenosrmrfrrr _find_lines_from_codeIs  rcCs4t||}|jD]}t|r|t||q|Sr")r co_constsinspectZiscoder> _find_lines)r|r}r~crrr rSs    rc Csi}tj}t||dj}t|j}|D]R\}}}} } |tjkrv|tjkrv|\} } | \} }t| | dD] }d||<qh|}q(W5QRX|S)Nrsr)tokenINDENTr;r\generate_tokensr^STRINGrange)rrlr Z prev_ttyperEtokZttypeZtstrstartendrzZslineZscolZelineZecolirrr _find_strings_s    rc Csz(t|}|}|j}W5QRXWn@tk rh}z"td||ftjdiWYSd}~XYnXt||d}t ||}t ||S)Nz%Not printing coverage data for %r: %sr3exec) r\r;readrlr@rCr*rDcompilerr)rrEprogrlrFr|r}rrr rXvs   rXc @sveZdZdddZdd Zd d d Zd d Zde_ddZddZ ddZ ddZ ddZ ddZ ddZddZdS)!rrrrNFc Cs||_||_t|||_i|_i|_d|_||_i|_i|_ i|_ d|_ | rTt |_ |rb|j |_nZ|rp|j|_nL|r|r|j|_|j|_n2|r|j|_|j|_n|r|j|_|j|_nd|_dSrK)r9r:rignorer5Zpathtobasename donothingtrace _calledfuncs_callers _caller_cache start_time_timeglobaltrace_trackcallers globaltraceglobaltrace_countfuncsglobaltrace_ltlocaltrace_trace_and_count localtracelocaltrace_tracelocaltrace_count) rrhr countfuncs countcallers ignoremods ignoredirsr9r:timingrrr rs6      zTrace.__init__cCs ddl}|j}||||dS)Nr)__main____dict__runctx)rcmdrdictrrr runsz Trace.runc Csh|dkr i}|dkri}|js6t|jt|jzt|||W5|jsbtdtdXdSr")r threadingsettracerr*r)rrglobalslocalsrrr rs   z Trace.runctxc Ost|dkr|^}}}nV|s&tdnHd|krZ|d}|^}}ddl}|jdtddntdt|dd}|jst|j z|||}W5|jstdX|S) Nz8descriptor 'runfunc' of 'Trace' object needs an argumentfuncrz0Passing 'func' as keyword argument is deprecated) stacklevelz7runfunc expected at least 1 positional argument, got %dr) r+ TypeErrorpopwarningswarnDeprecationWarningrr*rr)argskwrrrresultrrr runfuncs.       z Trace.runfuncz($self, func, /, *args, **kw)c Cs|j}|j}|rt|}nd}|j}d}||jkrL|j|dk r|j|}nd|j|<ddt|D}t|dkrddt|dD}t|dkrddt|dD} t| dkr| dj}||j|<|dk rd||f}|||fS)NcSsg|]}t|r|qSr)rZ isfunction)r rErrr rs z1Trace.file_module_function_of..rcSsg|]}t|tr|qSr) isinstancerr rrr rs rcSsg|]}t|dr|qS) __bases__)hasattr)r rrrr rs z%s.%s) f_code co_filenamer'co_namergcZ get_referrersr+r) rframer|rrrdZclsnameZfuncsZdictsclassesrrr file_module_function_ofs,          zTrace.file_module_function_ofcCs0|dkr,||}||j}d|j||f<dSNcallr)rf_backr)rrwhyarg this_funcZ parent_funcrrr r s  zTrace.globaltrace_trackcallerscCs |dkr||}d|j|<dSr)rr)rrrrrrrr rs zTrace.globaltrace_countfuncscCsj|dkrf|j}|jdd}|rbt|}|dk rf|j||}|sf|jrZtd||jf|j SndSdS)Nr__file__z! --- modulename: %s, funcname: %s) r f_globalsrLr'rrrrCrr)rrrrr|rrZ ignore_itrrr r!szTrace.globaltrace_ltcCs|dkr~|jj}|j}||f}|j|dd|j|<|jrTtdt|jddtj |}td||t ||fdd|j S) Nrzrr%.2f r %s(%d): %sr()rrf_linenor5rLrrCrrrr#rYgetliner)rrrrrrfrNbnamerrr r8s  z Trace.localtrace_trace_and_countcCsd|dkr^|jj}|j}|jr4tdt|jddtj|}td||t ||fdd|j S)Nrzrrrrr() rrrrrCrrrr#rYrr)rrrrrrfrrrr rGs  zTrace.localtrace_tracecCs<|dkr6|jj}|j}||f}|j|dd|j|<|jS)Nrzrr)rrrr5rLr)rrrrrrfrNrrr rTs zTrace.localtrace_countcCst|j|j|j|j|jdS)N)r9r:r7r8)rr5r9r:rr)rrrr results\s  z Trace.results) rrrrrrNNF)NN)rr r!rrrr__text_signature__rrrrrrrrrrrr rs& 2 )   c sddl}|}|jdddd|dd}|jdd d d d |jd dd dd |jddd dd |jddd dd |d}|}|jddd dd |jddd dd |jdddd |jd!d"d#d |jd$d%d d&d |jd'd(d d)d |jd*d+d d,d |d-d.}|jd/d0gd1d2|jd3d0gd4d2|jd5d d6d7d2|jd8d9d:d;|jd<|jd=d;|}|jrt d>t d?fd@dAdBdC|j D|_ fdDdC|jD|_|j r|j s| dEt|j |j dF}||j|j|jSt|j|j|j|jgs | dG|jrB|js8|jrB| dH|jr\|js\| dI|jdkrr| dJt|j|j|j|j|j |j|j |j |jdK }z|jrddl}|j}||\} } } | jf|jt _!dL| j| j"| j#| ddM} n^|jf|jt _!t$j%&|jt j%d<t'(|j} t)| *|jdN} W5QRX|jdLdddO} |+| | | WnPt,k r}zt -dPt j!d|fW5d}~XYnt.k rYnX|/}|j0s||j|j|jdS)QNrz --versionversionz trace 2.0)actionrz Main optionsz(One of these (or --report) must be givenz-cz--count store_truezCount the number of times each line is executed and write the counts to .cover for each module executed, in the module's directory. See also --coverdir, --file, --no-report below.)rhelpz-tz--tracez3Print each line to sys.stdout before it is executedz-lz --listfuncszKeep track of which functions are executed at least once and write the results to sys.stdout after the program exits. Cannot be specified alongside --trace or --count.z-Tz --trackcallsz^Keep track of caller/called pairs and write the results to sys.stdout after the program exits.Z Modifiersz-rz--reportzGenerate a report from a counts file; does not execute any code. --file must specify the results file to read, which must have been created in a previous run with --count --file=FILEz-Rz --no-reportz^Do not generate the coverage report files. Useful if you want to accumulate over several runs.z-fz--filez+File to accumulate counts over several runs)rz-Cz --coverdirzDirectory where the report files go. The coverage report for . will be written to file //.coverz-mz --missingz?Annotate executable lines that were not executed with ">>>>>> "z-sz --summaryz\Write a brief summary for each file to sys.stdout. Can only be used with --count or --reportz-gz--timingzQPrefix each line with the time since the program started. Only used while tracingZFilterszCan be specified multiple timesz--ignore-moduleappendzqIgnore the given module(s) and its submodules (if it is a package). Accepts comma separated list of module names.)rdefaultrz --ignore-dirzWIgnore files in the given directory (multiple directories can be joined by os.pathsep).z--moduleFzTrace a module. progname?zfile to run as main program)nargsr argumentszarguments to the programZstdlibZ platstdlibcs4tjtj|}|dd}tj|S)Nz$prefixz $exec_prefix)rr expanduser expandvarsr-r )s) _exec_prefix_prefixrr parse_ignore_dirszmain..parse_ignore_dircSs$g|]}|dD] }|qqS),)splitstrip)r rrrrr rs zmain..cs&g|]}|tjD] }|qqSr)rrpathsep)r rr)rrr rsz-r/--report requires -f/--file)r9r:zLmust specify one of --trace, --count, --report, --listfuncs, or --trackcallsz8cannot specify both --listfuncs and (--trace or --count)z3--summary can only be used with --count or --reportz3progname is missing: required with the main options)rrrrr9r:rr)rr __package__ __loader____spec__ __cached__r)rrrrzCannot run file %r because: %s)1argparseArgumentParser add_argumentZadd_argument_groupZadd_mutually_exclusive_groupZ REMAINDER parse_argsZ ignore_dir sysconfigZget_pathZ ignore_moduleZreportr4errorrrqZmissingrbrcanyrrhZ listfuncsZ trackcallsrrrmodulerunpyZ_get_module_detailsrrr*argvparentloaderrrrTio open_coderrrr@exit SystemExitrZ no_report)rparserZgrpZ_grpZoptsrtrZ module_nameZmod_nameZmod_specr|ZglobsrkrFr)rrrr mainbs                        (rr)N)__all__rrYrr*rrr\rrr{r<timerrrrwrr'r1rrrrrXrrrrrrr 2s: 20  _