o
    Zh-
                     @   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)scopec                 C   s   | d S )NZparquet )Zbase_datadirr   r   U/var/www/html/lang_env/lib/python3.10/site-packages/pyarrow/tests/parquet/conftest.pydatadir   s   r   c           	      C   s   t d}t d}d}| d \}}}}|jdd|||||jjddd	d
}z z|j|d W n	 ty:   Y nw W |  |S W |  |S |  w )Nboto3botocorez	test-s3fs
connectionZs3http://{}:{}Zs3v4)Zsignature_versionz	us-east-1)endpoint_urlZaws_access_key_idZaws_secret_access_keyconfigZregion_name)ZBucket)pytestimportorskipclientformatZConfigZcreate_bucket	Exceptionclose)		s3_serverr   r	   Zs3_bucket_namehostport
access_key
secret_keyZ	s3_clientr   r   r   	s3_bucket   s0   


	
r   c           	      c   s    t d}| d \}}}}|j||dd||id}d|t }|| ||fV  z
|j|dd W d S  tyA   Y d S w )	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_s3fs6   s"   


r!   c           	      c   sV    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_fsM   s   
r%   )r   Zpyarrow.utilr   Zfixturer   r   r!   r%   r   r   r   r   <module>   s   



