
    XIg                     V    d dl mZ d dlZd dlmZ d dlmZ d dlmZ  G d d      ZdgZ	y)    )OptionalN)DuckDBPyConnectionContributionsAcceptedError)	SparkConfc            	       
   e Zd ZdefdZedefd       Zd*dZe	d+de
e   dd fd       Ze	d	ed
eddfd       Zedefd       Zedefd       Zedefd       Zedefd       Zedefd       Zedefd       ZdefdZdeddfdZd,dededdfdZdeddfdZd*dZdeddfdZdeddfdZde
e   fdZdefdZd	ede
e   fdZdeddfd Z d
eddfd!Z!d,ded"ed#eddfd$Z"d	ed
eddfd%Z#d&eddfd'Z$d*d(Z%defd)Z&y)-SparkContextmasterc                 8    t        j                  d      | _        y )Nz:memory:)duckdbconnect_connection)selfr
   s     ^/var/www/html/answerous/venv/lib/python3.12/site-packages/duckdb/experimental/spark/context.py__init__zSparkContext.__init__
   s    !>>*5    returnc                     | j                   S N)r   r   s    r   
connectionzSparkContext.connection   s    r   Nc                 8    | j                   j                          y r   )r   closer   s    r   stopzSparkContext.stop   s     r   confc                     t         r   r   )clsr   s     r   getOrCreatezSparkContext.getOrCreate       ((r   keyvaluec                     t         r   r   )r   r    r!   s      r   setSystemPropertyzSparkContext.setSystemProperty   r   r   c                     t         r   r   r   s    r   applicationIdzSparkContext.applicationId   r   r   c                     t         r   r   r   s    r   defaultMinPartitionsz!SparkContext.defaultMinPartitions    r   r   c                     t         r   r   r   s    r   defaultParallelismzSparkContext.defaultParallelism$   r   r   c                     t         r   r   r   s    r   	startTimezSparkContext.startTime,   r   r   c                     t         r   r   r   s    r   uiWebUrlzSparkContext.uiWebUrl0   r   r   c                     t         r   r   r   s    r   versionzSparkContext.version4   r   r   c                     t         r   r   r   s    r   __repr__zSparkContext.__repr__8       ((r   pathc                     t         r   r   r   r3   s     r   
addArchivezSparkContext.addArchive>   r2   r   	recursivec                     t         r   r   )r   r3   r7   s      r   addFilezSparkContext.addFileA   r2   r   c                     t         r   r   r5   s     r   	addPyFilezSparkContext.addPyFileD   r2   r   c                     t         r   r   r   s    r   cancelAllJobszSparkContext.cancelAllJobsP   r2   r   groupIdc                     t         r   r   )r   r>   s     r   cancelJobGroupzSparkContext.cancelJobGroupS   r2   r   c                     t         r   r   r5   s     r   dump_profileszSparkContext.dump_profilesV   r2   r   c                     t         r   r   r   s    r   getCheckpointDirzSparkContext.getCheckpointDir\   r2   r   c                     t         r   r   r   s    r   getConfzSparkContext.getConf_   r2   r   c                     t         r   r   )r   r    s     r   getLocalPropertyzSparkContext.getLocalPropertyb   r2   r   dirNamec                     t         r   r   )r   rI   s     r   setCheckpointDirzSparkContext.setCheckpointDir   r2   r   c                     t         r   r   )r   r!   s     r   setJobDescriptionzSparkContext.setJobDescription   r2   r   descriptioninterruptOnCancelc                     t         r   r   )r   r>   rN   rO   s       r   setJobGroupzSparkContext.setJobGroup   r2   r   c                     t         r   r   )r   r    r!   s      r   setLocalPropertyzSparkContext.setLocalProperty   r2   r   logLevelc                     t         r   r   )r   rT   s     r   setLogLevelzSparkContext.setLogLevel   r2   r   c                     t         r   r   r   s    r   show_profileszSparkContext.show_profiles   r2   r   c                     t         r   r   r   s    r   	sparkUserzSparkContext.sparkUser   r2   r   )r   Nr   )F)'__name__
__module____qualname__strr   propertyr   r   r   classmethodr   r   r   r#   r%   intr'   r)   r+   r-   r/   r1   r6   boolr9   r;   r=   r@   rB   rD   rF   rH   rK   rM   rQ   rS   rV   rX   rZ    r   r   r	   r	   	   sQ   6s 6  .    ! )x	2 )n ) ) )C ) ) ) ) )s ) ) )c ) ) )C ) ) )3 ) ) )# ) ) ) ) ))# ))s )t ))C )D )T ))c )d )))c )d ))# )$ ))(3- )) ))C )HSM )<) ) ))s )t ))3 )S )T )^b ))C ) ) ))C )D )))3 )r   r	   )
typingr   r   r   #duckdb.experimental.spark.exceptionr   duckdb.experimental.spark.confr   r	   __all__rc   r   r   <module>rh      s+      % J 4J) J)r 
r   