U
    ½mœd‰  ã                   @   sN   d Z ddlZddlZddlZddlZdd„ Zdd„ Zdd„ Zed	koHeƒ  dS )
zÎ
Generate the zip test data files.

Run to build the tests/zipdataNN/ziptestdata.zip files from
files in tests/dataNN.

Replaces the file with the working copy, but does commit anything
to the source repo.
é    Nc                  C   s   d} t tt| ƒƒ dS )a;  
    >>> from unittest import mock
    >>> monkeypatch = getfixture('monkeypatch')
    >>> monkeypatch.setattr(zipfile, 'ZipFile', mock.MagicMock())
    >>> print(); main()  # print workaround for bpo-32509
    <BLANKLINE>
    ...data01... -> ziptestdata/...
    ...
    ...data02... -> ziptestdata/...
    ...
    )Z01Z02N)ÚtupleÚmapÚgenerate)Úsuffixes© r   ú^/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/importlib_resources/tests/update-zips.pyÚmain   s    r   c              	   C   sˆ   t  t¡j t ¡ ¡}|d| › d� }t |d¡�L}t	|d| › � ƒD ]2\}}dt  
| ¡ ¡ }t|d|ƒ | ||¡ qFW 5 Q R X d S )NZzipdataz/ziptestdata.zipÚwÚdataZziptestdataz->)ÚpathlibÚPathÚ__file__ÚparentÚrelative_toÚosÚgetcwdÚzipfileÚZipFileÚwalkÚPurePosixPathÚas_posixÚprintÚwrite)ÚsuffixÚrootZzfpathÚzfÚsrcÚrelÚdstr   r   r   r   !   s    r   c              
   c   sf   t  | ¡D ]V\}}}t t¡� | d¡ W 5 Q R X |D ]&}t |¡| }| | ¡}||fV  q8q
d S )NÚ__pycache__)	r   r   Ú
contextlibÚsuppressÚ
ValueErrorÚremover   r   r   )ÚdatapathÚdirpathÚdirnamesÚ	filenamesÚfilenameÚresr   r   r   r   r   +   s    
r   Ú__main__)	Ú__doc__r    r   r   r   r   r   r   Ú__name__r   r   r   r   Ú<module>   s   


