U
    k/e	                     @   sV   d dl Z d dlmZ e jdddd Ze jdd Ze jd	d
 Ze jdd ZdS )    N)guidmodule)Zscopec                 C   s   | d S )NZparquet )Zbase_datadirr   r   B/tmp/pip-unpacked-wheel-seu8352k/pyarrow/tests/parquet/conftest.pydatadir   s    r   c           	      C   s|   t d}t d}| d \}}}}|jdd|||||jjdddd	}|d
}z|  W n tk
rv   Y nX d
S )Nboto3botocore
connections3http://{}:{}Zs3v4)Zsignature_versionz	us-east-1)endpoint_urlZaws_access_key_idZaws_secret_access_keyconfigZregion_namez	test-s3fs)	pytestimportorskipresourceformatclientZConfigZBucketcreate	Exception)		s3_serverr   r   hostport
access_key
secret_keyr
   Zbucketr   r   r   	s3_bucket   s"    



r   c           	      c   s   t d}| d \}}}}|j||dd||id}d|t }|| ||fV  z|j|dd W n tk
r~   Y nX d S )	Ns3fsr	   r   r   )keysecretZclient_kwargsz{}/{}T)	recursive)r   r   ZS3FileSystemr   r   mkdirZrmFileNotFoundError)	r   r   r   r   r   r   r   fsZ	test_pathr   r   r   s3_example_s3fs3   s     
 


r"   c           	      c   sT   ddl m} | d \}}}}d||||}||\}}|d |||fV  d S )Nr   )
FileSystemr	   zDs3://{}:{}@mybucket/data.parquet?scheme=http&endpoint_override={}:{}Zmybucket)Z
pyarrow.fsr#   r   Zfrom_uriZ
create_dir)	r   r#   r   r   r   r   urir!   pathr   r   r   s3_example_fsJ   s       
r&   )r   Zpyarrow.utilr   Zfixturer   r   r"   r&   r   r   r   r   <module>   s   



