B Wc@sBdZddlZddlZddlZddlZddlmZddlmZddl Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZee jddGdddZeedddZeddddZeddeedddZeeedd d!Zeeeed"d#d$Z dS)%z>Support for providing temporary directories to test functions.N)Path)Optional) LOCK_TIMEOUT)make_numbered_dir)make_numbered_dir_with_cleanup)rm_rf)final)Config)check_ispytest)fixture)FixtureRequest) MonkeyPatchF)Zinitc@seZdZdZejeedZeZ ejeedZ dddeeeee ddddZ e ddee dd d d Zeed d dZdee edddZedddZdS)TempPathFactoryzFactory for temporary directories under the common base temp directory. The base directory can be configured using the ``--basetemp`` option. )typeNF) _ispytest)given_basetempbasetemprreturncCs>t||dkrd|_nttjt||_||_||_dS)N) r _given_basetemprospathabspathstr_trace _basetemp)selfrtracerrr-lib/python3.7/site-packages/_pytest/tmpdir.py__init__#s zTempPathFactory.__init__)configrrcCs"t|||jj|jdddS)zTCreate a factory according to pytest configuration. :meta private: ZtmpdirT)rrr)r Zoptionrrget)clsr!rrrr from_config6s  zTempPathFactory.from_config)basenamercCs8tj|}||j|kr4t|d|S)Nz& is not a normalized and relative path)rrnormpath getbasetempresolveparent ValueError)rr%rrr_ensure_relative_to_basetempHs z,TempPathFactory._ensure_relative_to_basetempT)r%numberedrcCsL||}|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. i)mode)rootprefixr-mktemp)r+r'joinpathmkdirrr)rr%r,prrrr0Ns  zTempPathFactory.mktemp)rcCs^|jdk r|jS|jdk rF|j}|r0t||jdd|}ntjd}t |p^t }t pnd}| d|}y|jdddWn,tk r| d }|jdddYnXtjd kr$t}|}|d kr$|j|krtd |d |jd@dkr$t||jd@td|dtdd}|dk sHt|||_|d||S)zReturn the base temporary directory, creating it if needed. :returns: The base temporary directory. Ni)r-ZPYTEST_DEBUG_TEMPROOTunknownz pytest-of-T)r-exist_okzpytest-of-unknownZwin32zThe temporary directory z: is not owned by the current user. Fix this and try again.?rizpytest-)r/r.ZkeepZ lock_timeoutr-z new basetemp)rrexistsrr2r(renvironr"rtempfileZ gettempdirget_userr1OSErrorsysplatformgetuidstatst_uidst_modechmodrrAssertionErrorr)rrZfrom_envZtemprootuserZrootdirZuidZ rootdir_statrrrr'fsH           zTempPathFactory.getbasetemp)N)T)__name__ __module__ __qualname____doc__attrZibrrrrrboolr classmethodr r$rr+r0r'rrrrrs   r)rc Cs.yddl}|Sttfk r(dSXdS)zlReturn the current user name, or None if getuser() does not work in the current environment (see #1010).rN)getpassZgetuser ImportErrorKeyError)rNrrrr<s r<)r!rcCs6t}||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. T)r_tmp_path_factoryF)ZraisingN)rZ add_cleanupZundorr$setattr)r!ZmprQrrrpytest_configures rSZsession)Zscope)requestrcCs|jjS)zGReturn a :class:`pytest.TempPathFactory` instance for the test session.)r!rQ)rTrrrtmp_path_factorysrU)rTfactoryrcCs4|jj}tdd|}d}|d|}|j|ddS)Nz[\W]_T)r,)Znodenameresubr0)rTrVrYZMAXVALrrr_mk_tmps  r\)rTrUrcCs t||S)aReturn a temporary directory path object which is unique to each test function invocation, created as a sub directory of the base temporary directory. By default, a new base temporary directory is created each test session, and old bases are removed after 3 sessions, to aid in debugging. If ``--basetemp`` is used then it is cleared each session. See :ref:`base temporary directory`. The returned object is a :class:`pathlib.Path` object. )r\)rTrUrrrtmp_pathsr])!rJrrZr>r;ZpathlibrtypingrrKrrrrZ_pytest.compatr Z_pytest.configr Z_pytest.deprecatedr Z_pytest.fixturesr r Z_pytest.monkeypatchrsrrr<rSrUr\r]rrrrs6