U
    -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)scopec                 C   s   | d S )NZparquet )Zbase_datadirr   r   _/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/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   



