U f{j, @sddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z mZddlmZmZmZmZddl mZmZmZmZdd lmZdZdd d ZdS)N)rmtree) __version__)loggerwarn _safe_input)_bst_license_textRELEASESTESTING_VERSIONEDMISC_VERSIONED)_dataset_version_do_path_update _get_path _mne_path)_compare_versionFTc #Cs ddl} | dk r$t| dkr$td|dkr:| j|d}n|dkrN| j|d}t|tr^|g}dd |D} | d} |d}|d d}|d }t||| d }t ||}| d rt || }t |}t | d}| dkrtn|}t|| }|dk ot|d|}|r&td| d|d|t d}|s\|r\|s\td|rX||fS|S|s |s | ds t |r|dk rt|||| |r||fS|S|s|r||fS|S| d r |sdtjkrd}ntdtdd}|dkr tdttjtjkd}| dkr0d|d<| dk rB| |d<| dk r^dd | i|d!<| jf|}t}t}t| D]:\}}||}|d"}|d#}|d$} |||<| ||<q~| j|dkrt |n|dd||dd%}!| !"t#t$t| D]>}||d"}|!j%|||d&|dk rt&'t ||q| d'krlt(|d(d)t&)t |t*|n*| d*krt(|d(d)t&)t |t+||dk r| d rd+n| }"t||||"t|| }|rt|d,t,-d-rt.d.j/| |t,d/|r||fS|S)0axFetch an MNE-compatible dataset using pooch. Parameters ---------- dataset_params : list of dict | dict The dataset name(s) and corresponding parameters to download the dataset(s). The dataset parameters that contains the following keys: ``archive_name``, ``url``, ``folder_name``, ``hash``, ``config_key`` (optional). See Notes. processor : None | "unzip" | "untar" | instance of pooch.Unzip | instance of pooch.Untar What to do after downloading the file. ``"unzip"`` and ``"untar"`` will decompress the downloaded file in place; for custom extraction (e.g., only extracting certain files from the archive) pass an instance of :class:`pooch.Unzip` or :class:`pooch.Untar`. If ``None`` (the default), the files are left as-is. path : None | str Directory in which to put the dataset. If ``None``, the dataset location is determined by first checking whether ``dataset_params['config_key']`` is defined, and if so, whether that config key exists in the MNE-Python config file. If so, the configured path is used; if not, the location is set to the value of the ``MNE_DATA`` config key (if it exists), or ``~/mne_data`` otherwise. force_update : bool Force update of the dataset even if a local copy exists. Default is False. update_path : bool | None If True (default), set the mne-python config to the given path. If None, the user is prompted. download : bool If False and the dataset has not been downloaded yet, it will not be downloaded and the path will be returned as ``''`` (empty string). This is mostly used for testing purposes and can be safely ignored by most users. check_version : bool Whether to check the version of the dataset or not. Each version of the dataset is stored in the root with a ``version.txt`` file. return_version : bool Whether or not to return the version of the dataset or not. Defaults to False. accept : bool Some MNE-supplied datasets require acceptance of an additional license. Default is ``False``. auth : tuple | None Optional authentication tuple containing the username and password/token, passed to :class:`pooch.HTTPDownloader` (e.g., ``auth=('foo', 012345)``). token : str | None Optional authentication token passed to :class:`pooch.HTTPDownloader`. Returns ------- data_path : instance of Path The path to the fetched dataset. version : str Only returned if ``return_version`` is True. See Also -------- mne.get_config mne.set_config mne.datasets.has_dataset Notes ----- The ``dataset_params`` argument must contain the following keys: - ``archive_name``: The name of the (possibly compressed) file to download - ``url``: URL from which the file can be downloaded - ``folder_name``: the subfolder within the ``MNE_DATA`` folder in which to save and uncompress (if needed) the file(s) - ``hash``: the cryptographic hash type of the file followed by a colon and then the hash value (examples: "sha256:19uheid...", "md5:upodh2io...") - ``config_key`` (optional): key passed to :func:`mne.set_config` to store the on-disk location of the downloaded dataset (e.g., ``"MNE_DATASETS_EEGBCI_PATH"``). This will only work for the provided datasets listed :ref:`here `; do not use for user-defined datasets. An example would look like:: {'dataset_name': 'sample', 'archive_name': 'MNE-sample-data-processed.tar.gz', 'hash': 'md5:12b75d1cb7df9dfb4ad73ed82f61094f', 'url': 'https://osf.io/86qa2/download?version=5', 'folder_name': 'MNE-sample-data', 'config_key': 'MNE_DATASETS_SAMPLE_PATH'} For datasets where a single (possibly compressed) file must be downloaded, pass a single :class:`dict` as ``dataset_params``. For datasets where multiple files must be downloaded and (optionally) uncompressed separately, pass a list of dicts. rNrzEauth should be a 2-tuple consisting of a username and password/token.Zuntar) extract_dirZunzipcSsg|] }|dqS)Z dataset_name).0paramsrrR/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/datasets/_fetch.py sz!fetch_dataset.. config_key folder_name)pathkeynameZbst_Zfake>zDataset z version z out of date, latest version is zXDataset out of date but force_update=False and download=False, returning empty data_pathZhf_sef_z--accept-brainstorm-licenseyz%sAgree (y/[n])? )Zusez1You must agree to the license to use this dataset) progressbarFr auth Authorizationztoken headers archive_nameurlhash)rbase_urlversionurlsregistryZretry_if_failed)fname downloader processormiscT) ignore_errorsZtestingZ brainstormrKos.pathrr8shutilrrrrNutilsrrrconfigr r r r r rrrfixesrr;rWrrrrs*