o
    䯪g
                     @   s   d dl Z d dlZd dlZd dlZd dlmZ ejdddd 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 )Nparquet )base_datadirr   r   Z/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/pyarrow/tests/parquet/conftest.pydatadir   s   r	   c                  C   s6   t jdkr
td tjd} | stdt	| S )N
emscriptenz$needs PARQUET_TEST_DATA files accessPARQUET_TEST_DATAzRPlease point the PARQUET_TEST_DATA environment variable to the test data directory)
sysplatformpytestskiposenvirongetRuntimeErrorpathlibPath)resultr   r   r   parquet_test_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
connections3http://{}:{}s3v4)signature_versionz	us-east-1)endpoint_urlaws_access_key_idaws_secret_access_keyconfigregion_name)Bucket)r   importorskipclientformatConfigcreate_bucket	Exceptionclose)		s3_serverr   r   s3_bucket_namehostport
access_key
secret_key	s3_clientr   r   r   	s3_bucket+   s0   


	
r3   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secretclient_kwargsz{}/{}T)	recursive)r   r%   S3FileSystemr'   r   mkdirrmFileNotFoundError)	r,   r3   r4   r.   r/   r0   r1   fs	test_pathr   r   r   s3_example_s3fsE   s"   


r?   c           	      c   sV    ddl m} | d \}}}}d||||}||\}}|d |||fV  d S )Nr   )
FileSystemr   z_s3://{}:{}@mybucket/data.parquet?scheme=http&endpoint_override={}:{}&allow_bucket_creation=Truemybucket)
pyarrow.fsr@   r'   from_uri
create_dir)	r,   r@   r.   r/   r0   r1   urir=   pathr   r   r   s3_example_fs\   s   
rG   )r   r   r   r   pyarrow.utilr   fixturer	   r   r3   r?   rG   r   r   r   r   <module>   s   






