a *j+@sdZddlmZddlmZddlZddlZddlmZddl Z ddl m Z ddl Z ddl mZddl mZdd l mZd d lmZd d lmZd d lmZd dlmZd dlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*e*e+e,e-fZ.edZ/eej0GdddZ1dd d!d"Z2d#d$d%d&d'Z3d(d$d)d*d+Z4e!d,d-d.dd/d0d1Z5d.dd2d3d4d5Z6e!d.dd6d7d8d9Z7d:d;dd>d?d@dAdBdCdDZ9dS)Ez>Support for providing temporary directories to test functions.) annotations) GeneratorN)Path)rmtree)Any)final)Literal)cleanup_dead_symlinks) LOCK_TIMEOUT)make_numbered_dir)make_numbered_dir_with_cleanup)rm_rf) get_user_id)Config)ExitCode)hookimpl)Parser)check_ispytest)fixture)FixtureRequest) MonkeyPatch)Item) TestReport)StashKeyallfailednonec @seZdZUdZded<ded<ded<ded<d ed <d#d d ddd ddddddZed d ddddddZdddddZd$ddddddZ dd d!d"Z d S)%TempPathFactoryzFactory for temporary directories under the common base temp directory, as discussed at :ref:`temporary directory location and retention`. z Path | None_given_basetempr_trace _basetempint_retention_count RetentionType_retention_policyNF _ispytestboolNone)given_basetempretention_countretention_policybasetempr(returncCsJt||durd|_nttjt||_||_||_||_ ||_ dSN) rr rospathabspathstrr!r$r&r")selfr+r,r-tracer.r(r74E:\soccer\ml_env\Lib\site-packages\_pytest/tmpdir.py__init__6s zTempPathFactory.__init__r)configr(r/cCsnt|t|d}|dkr.td|d|d}|dvrPtd|d||jj|jd||d d S) zTCreate a factory according to pytest configuration. :meta private: tmp_path_retention_countrz6tmp_path_retention_count must be >= 0. Current input: .tmp_path_retention_policyrzKtmp_path_retention_policy must be either all, failed, none. Current input: ZtmpdirT)r+r6r,r-r()rr#Zgetini ValueErroroptionr.r6get)clsr:r(countpolicyr7r7r8 from_configMs$     zTempPathFactory.from_configr4)basenamer/cCs8tj|}||j|kr4t|d|S)Nz& is not a normalized and relative path)r1r2normpath getbasetempresolveparentr>)r5rEr7r7r8_ensure_relative_to_basetempms z,TempPathFactory._ensure_relative_to_basetempTr)rEnumberedr/cCsL||}|s*||}|jddnt||dd}|d||S)aCreate a new temporary directory managed by the factory. :param basename: Directory base name, must be a relative path. :param numbered: If ``True``, ensure the directory is unique by adding a numbered suffix greater than any existing one: ``basename="foo-"`` and ``numbered=True`` means that this function will create directories named ``"foo-0"``, ``"foo-1"``, ``"foo-2"`` and so on. :returns: The path to the new directory. mode)rootprefixrNmktemp)rJrGjoinpathmkdirr r!)r5rErKpr7r7r8rQss  zTempPathFactory.mktempr/c Csb|jdur|jS|jdurF|j}|r0t||jdd|}ntjd}t |p^t }t pnd}| d|}z|jdddWn*ty| d }|jdddYn0t}|dur|}|j|krtd |d |jd @d krt||jd@|j}|jdkr(d }td||tdd}|dusLJ|||_|d||S)zReturn the base temporary directory, creating it if needed. :returns: The base temporary directory. NrLrMZPYTEST_DEBUG_TEMPROOTunknownz pytest-of-T)rNexist_okzpytest-of-unknownzThe temporary directory z: is not owned by the current user. Fix this and try again.?rirzpytest-)rPrOkeepZ lock_timeoutrNz new basetemp)r"r existsrrSrHr1environr@rtempfile gettempdirget_userrROSErrorrstatst_uidst_modechmodr$r&r r r!) r5r.Zfrom_envZtemprootuserrootdiruidZ rootdir_statrYr7r7r8rGsP             zTempPathFactory.getbasetemp)N)T) __name__ __module__ __qualname____doc____annotations__r9 classmethodrDrJrQrGr7r7r7r8r(s rz str | NonerUc Cs2zddl}|WStttfy,YdS0dS)zlReturn the current user name, or None if getuser() does not work in the current environment (see #1010).rN)getpassgetuser ImportErrorr_KeyError)rmr7r7r8r^s  r^rr*)r:r/cCs6t}||jtj|dd}|j|d|dddS)aCreate a TempPathFactory and attach it to the config object. This is to comply with existing plugins which expect the handler to be available at pytest_configure time, but ideally should be moved entirely to the tmp_path_factory session fixture. Tr'_tmp_path_factoryF)ZraisingN)rZ add_cleanupZundorrDsetattr)r:mprqr7r7r8pytest_configures rtr)parserr/cCs$|jdddd|jdddddS)Nr;zfHow many sessions should we keep the `tmp_path` directories, according to `tmp_path_retention_policy`.)helpdefaultr=zvControls which directories created by the `tmp_path` fixture are kept around, based on test outcome. (all/failed/none)r)Zaddini)rur7r7r8pytest_addoptionsrysession)scoper)requestr/cCs|jjS)zGReturn a :class:`pytest.TempPathFactory` instance for the test session.)r:rq)r|r7r7r8tmp_path_factorysr}r)r|factoryr/cCs4|jj}tdd|}d}|d|}|j|ddS)Nz[\W]_T)rK)nodenameresubrQ)r|r~rZMAXVALr7r7r8_mk_tmps  rzGenerator[Path])r|r}r/ccsPt||}|V|j}|jjt}|dkrB|ddrBt|dd|jjt=dS)a;Return a temporary directory (as :class:`pathlib.Path` object) which is unique to each test function invocation. The temporary directory is created as a subdirectory of the base temporary directory, with configurable retention, as discussed in :ref:`temporary directory location and retention`. rcallT ignore_errorsN)rr&rstashtmppath_result_keyr@r)r|r}r2rCZ result_dictr7r7r8tmp_paths   rzint | ExitCode) exitstatuscCsb|jj}|j}|durdS|j}|dkrN|dkrN|jdurN|rNt|dd|r^t|dS)zAfter each session, remove base directory if all the tests passed, the policy is "failed", and the basetemp is not specified by a user. NrrTr)r:rqr"r&r is_dirrr )rzrr}r.rCr7r7r8pytest_sessionfinishs rT)wrapperZtryfirstrz'Generator[None, TestReport, TestReport])itemr/ccs2dV}|jdusJi}|j|jt||j<|Sr0)whenZpassedr setdefaultr)rrrepemptyr7r7r8pytest_runtest_makereport0s r):rj __future__rcollections.abcr dataclassesr1pathlibrrshutilrr\typingrrrr r r r rZ_pytest.compatrZ_pytest.configrrrZ_pytest.config.argparsingrZ_pytest.deprecatedrZ_pytest.fixturesrrZ_pytest.monkeypatchrZ _pytest.nodesrZ_pytest.reportsrZ _pytest.stashrdictr4r)rr% dataclassrr^rtryr}rrrrr7r7r7r8sZ