
    ei                         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        Z ej                  d      d        Zej                  d        Z	ej                  d        Z
ej                  d	        Zy)
    N)guidmodule)scopec                     | dz  S )Nparquet )base_datadirs    `/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/pyarrow/tests/parquet/conftest.pydatadirr      s    )##    c                      t         j                  dk(  rt        j                  d       t        j
                  j                  d      } | st        d      t        j                  |       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)results    r
   parquet_test_datadirr       sU    
|||#:;ZZ^^/0F A B 	B<<r   c           	      n   t        j                  d      }t        j                  d      }d}| d   \  }}}}|j                  dd| d| |||j                  j                  d	      d
      }	 |j	                  |       |j                          |S # t
        $ r Y w xY w# |j                          w xY 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Configcreate_bucket	Exceptionclose)		s3_serverr   r   s3_bucket_namehostport
access_key
secret_key	s3_clients	            r
   	s3_bucketr8   +   s    (E"":.H N)2<)@&D$
JtfAdV,$(%%%?  I~6 	   	s$   /B 	BB" BB" "B4c           	   #     K   t        j                  d      }| d   \  }}}}|j                  ||dd| d| i      }| dt                }|j	                  |       ||f 	 |j                  |d	       y # t        $ r Y y w xY ww)
Ns3fsr   r%   r!   r"   )keysecretclient_kwargs/T)	recursive)r   r+   S3FileSystemr   mkdirrmFileNotFoundError)	r1   r8   r:   r3   r4   r5   r6   fs	test_paths	            r
   s3_example_s3fsrF   E   s     v&D)2<)@&D$
J			gdV1TF3
 
 
B +Qtvh'IHHY
i-
i4( s*   A"B%A9 8B9	BBBBc           	   #      K   ddl m} | d   \  }}}}d| d| d| d| d	}|j                  |      \  }}|j                  d       |||f y w)	Nr   )
FileSystemr   zs3://r"   z5@mybucket/data.parquet?scheme=http&endpoint_override=z&allow_bucket_creation=Truemybucket)
pyarrow.fsrH   from_uri
create_dir)	r1   rH   r3   r4   r5   r6   urirD   paths	            r
   s3_example_fsrO   \   su     %)2<)@&D$
J

|1ZL )"V1TF*E	G  ""3'HBMM*
c4-s   AA)r   r   r   r   pyarrow.utilr   fixturer   r   r8   rF   rO   r   r   r
   <module>rR      s   $ 
  
   h$  $ h      2  ,  r   