o v&i C@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlZddlmZddlmZddlmZdd lmZGd d d eZGd d d ZejddZddZGdddZGdddeZGdddeZGdddeZ dS)z8Coverage controllers for use by pytest-cov and nose-cov.N)Path)Union) CoverageData)filename_suffix)CentralCovContextWarning) DistCovErrorc@s eZdZdS)BrokenCovConfigErrorN)__name__ __module__ __qualname__r r W/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pytest_cov/engine.pyr sr c@seZdZeddZdS) _NullFilecCsdSNr )vr r rwritesz_NullFile.writeN)r r r staticmethodrr r r rrsrc csFt||}zt||t|dVWt|||dSt|||wr)getattrsetattrcopy)objattrbackupr r r_backup"s  rcstfdd}|S)Nc stzt}Wn tyd}Ynwt|jz|g|Ri|W|dur.t|SS|dur9t|wwr)rcwdOSErroroschdirtopdir)selfargskwargs original_cwdmethr rensure_topdir_wrapper-s     z-_ensure_topdir..ensure_topdir_wrapper) functoolswraps)r%r&r r$r_ensure_topdir,sr)c@seZdZdZdejdedefdedeffddZ e j dd Z e d d Ze d d ZddZddZeddZeddZddZe ddZdS) CovControllerz0Base class for different plugin implementations.optionsconfigNnodeidcCs||j|_|j|_|j|_|j|_|j|_|j|_||_||_d|_d|_ d|_ t |_ g|_ tt|_d|_d|_dS)z8Get some common config used by multiple derived classes.NF) cov_source cov_report cov_config cov_append cov_branch cov_precisionr,r-cov combining_cov data_fileset node_descsfailed_workersrfspathrrr is_collocatedstarted)r r+r,r-r r r__init__Bs  zCovController.__init__ccs*t}t|jdVt|dSr)rrrrr)r r#r r r ensure_topdirVs  zCovController.ensure_topdircCsd|_|jdSNF)r<r4stopr r r rpause]szCovController.pausecCs|jd|_dSNT)r4startr<rAr r rresumebs  zCovController.resumecC d|_dSrCr<rAr r rrDg zCovController.startcCrFr?rGrAr r rfinishjrHzCovController.finishcCsd|dj|ddS)z"Return a description of this node.zplatform {}, python {}z{}.{}.{}-{}-{}N)format)platform version_infor r r get_node_descmszCovController.get_node_desccCs tjdd\}}|dkrd}|S)N)P)fallback(rO)shutilget_terminal_size)width_r r r get_widthsszCovController.get_widthcCst|dr |||dS|}tjdkr|d8}t|t|ddt|d}||}|d|d|}t|t||krK||7}|d7}||dS)Nsepwin32r z ) hasattrrXrWsysrLmaxlenrstripr)r streamstxt fullwidthNfillliner r rrX|s  " zCovController.sepc sTd}js$tjdjjddtdWdS1swYtjdkr:|dddjn|dd t jD] }|d |qFj rp|dd | d j D] }| |j j d qctfdddDrdjvpdd|jd}tjtodjv}|d|pditjdjjd*i|}Wdn1swYdjvrjd}tjdjjd|dWdn1swYtjdjjdtd}Wdn1swY|r| d|d n| ddjvrSjd}tjdjjd|d}Wdn 1s;wY| d|durMjjjn|d djvrjd}tjdjjd|d}Wdn 1sxwY| d|durjjjn|d djvrˈjd}tjdjjd|d}Wdn 1swY| d |durLjjjjn|d!jvrjd!}tjd)t|d"} jjd| d!d#}Wdn 1swYWdn 1s wY| d$|d d%jvrcjd%}tjd)t|d&} jjd| d!d#}Wdn 1sEwYWdn 1sUwY| d'|d d(jvrjd(}tjdjjd|djjdtd}Wdn 1swY| d)|durjjjn|d |S)+zProduce coverage reports.Nr,T) show_missing ignore_errorsfilerrVz coverage: coverager[zcoverage: failed workerszlThe following workers failed to return coverage data, ensure that pytest-cov is installed on these workers.  c3s|]}|jvVqdSr)r/).0xrAr r sz(CovController.summary..)term term-missingrr)rhrirj precisionz skip-covered skip_coveredannotate)ri directory)rirjz)Coverage annotated source written to dir z1Coverage annotated source written next to source htmlzCoverage HTML written to dir xml)rioutfilezCoverage XML written to file jsonz!Coverage JSON written to file %s markdownw)rirj output_formatz.Coverage Markdown information written to file zmarkdown-appendaz/Coverage Markdown information appended to file lcovzCoverage LCOV written to file r ) r/rr4reportrr_r8rXjoinsortedr9rgatewayidanyr3 isinstancedictvaluesupdateru html_reportr,html_dir xml_report xml_output json_report json_outputropen lcov_report lcov_output) r ratotal node_descnoder+rt annotate_diroutput output_filer rArsummarys         &  &  "      &zCovController.summary)r r r __doc__argparse Namespacerobjectstrr= contextlibcontextmanagerr>r)rBrErDrIrrNrWrXrr r r rr*?s$(     r*cs4eZdZdZefddZefddZZS)Centralz)Implementation for centralised operation.cstj|j|jd|jd|_|jjjdkrd}tj t |ddtj|j|jt ddt j |jjj|jd|_|jsA|j|jtdS) NTsourcebranch data_suffix config_file test_functionzDetected dynamic_context=test_function in coverage configuration. This is unnecessary as this plugin provides the more complete --cov-context option.r) stacklevel.combinerrrr6r)rlCoverager.r2r0r4r,dynamic_contextwarningswarnrrrpathabspathr6r5r1eraserDsuper)r message __class__r rrDs*   z Central.startcsdt|j|j|j|_|j|j|j|t j t j }|j |dS)zSStop coverage, save data to file and set the list of coverage objects to report on.N)rrIr4r@saver5loadcombinerNr]rLrMr8add)r rrr rrI s      zCentral.finish)r r r rr)rDrI __classcell__r r rrrs rc@s8eZdZdZeddZddZddZedd Zd S) DistMasterz&Implementation for distributed master.cCstj|j|jd|jd|_|jjjdkrtdd|j_ d|j_ d|j_ tj|j|jt ddt j|jjj|jd|_|jsF|j|j|jg|jjjd<dS) NTrrzDetected dynamic_context=test_function in coverage configuration. This is known to cause issues when using xdist, see: https://github.com/pytest-dev/pytest-cov/issues/604 It is recommended to use --cov-context instead.Frrr)rlrr.r2r0r4r,rr _warn_no_data_warn_unimported_source_warn_preimported_sourcerrrrr6r5r1rrDrpathsrAr r rrD4s0   zDistMaster.startcCs,|jt|jdd|jjDddS)zFWorkers need to know if they are collocated and what files have moved.cSsg|]}t|qSr )r)rnrootr r r Xsz-DistMaster.configure_node..)cov_master_hostcov_master_topdircov_master_rsync_rootsN) workerinputrsocket gethostnamer nodemanagerroots)r rr r rconfigure_nodeQs zDistMaster.configure_nodec Cst|di}d|vr|j|dSd|vrBdtttdd|df}t |d}| |d|d }|j j j d ||j}||j|j}|j|dS) z#Collect data file name from worker. workeroutputcov_worker_node_idNcov_worker_dataz %s.%s.%06d.%sri?B)suffixcov_worker_pathr)rr9appendrrrgetpidrandomrandintrloadsr4r,rr_rinforNrLrMr8r) r rerrorrrcov_datarrinforr r r testnodedown\s(    zDistMaster.testnodedowncCs>|j|j|j|_|j|j|jdS)zJCombines coverage data and sets the list of coverage objects to report on.N)r4r@rr5rrrAr r rrI|s    zDistMaster.finishN) r r r rr)rDrrrIr r r rr1s   rcs<eZdZdZefddZefddZddZZS) DistWorkerz'Implementation for distributed workers.cst|jjdko|j|jjdk|_|js9|jjd|j|jdur1fdd|jD|_|j|_t j |j|j d|jd|_ d|j _ |j tdS)Nrrcsg|]}|qSr )replace)rnr master_topdir worker_topdirr rrsz$DistWorker.start..TrF)rrr,rrr;r.r0rrlrr2r4rrDrrArrrrDs$   zDistWorker.startcsvt|j|jr|j|j|jjd<dS|j |j|j }|jj |j |j|ddS)z8Stop coverage and send relevant info back to the master.r)rrrN)rrIr4r@r;rr-r,rrget_datadumpsrr)r datarr rrIs     zDistWorker.finishcCsdS)z&Only the master reports so do nothing.Nr )r rar r rrszDistWorker.summary) r r r rr)rDrIrrr r rrrs!r)!rrrrr'rrrSrr]rpathlibrtypingrrl coverage.datarcoverage.sqldatarrkrr Exceptionr rrrr)r*rrrr r r rs8        A2X