o
    ?D©iA  ã                   @   sP   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„ dƒZdgZ	dS )é    )ÚOptionalN)ÚDuckDBPyConnection©ÚContributionsAcceptedError)Ú	SparkConfc                	   @   s  e Zd Zdefdd„Zedefdd„ƒZdGdd	„Ze	dHd
e
e dd fdd„ƒZe	dededdfdd„ƒZedefdd„ƒZedefdd„ƒZedefdd„ƒZedefdd„ƒZedefdd„ƒZedefdd„ƒZdefdd„Zdeddfd d!„ZdIded#eddfd$d%„Zdeddfd&d'„ZdGd(d)„Zd*eddfd+d,„Zdeddfd-d.„Zde
e fd/d0„Zdefd1d2„Zdede
e fd3d4„Zd5eddfd6d7„Z deddfd8d9„Z!dId*ed:ed;eddfd<d=„Z"dededdfd>d?„Z#d@eddfdAdB„Z$dGdCdD„Z%defdEdF„Z&dS )JÚSparkContextÚmasterc                 C   s   t  d¡| _| j d¡ d S )Nz:memory:z8set default_null_order='nulls_first_on_asc_last_on_desc')ÚduckdbÚconnectÚ_connectionÚexecute)Úselfr   © r   úZ/var/www/Datamplify/venv/lib/python3.10/site-packages/duckdb/experimental/spark/context.pyÚ__init__
   s   zSparkContext.__init__Úreturnc                 C   s   | j S ©N)r   ©r   r   r   r   Ú
connection   s   zSparkContext.connectionNc                 C   s   | j  ¡  d S r   )r   Úcloser   r   r   r   Ústop   s   zSparkContext.stopÚconfc                 C   ó   t ‚r   r   )Úclsr   r   r   r   ÚgetOrCreate   ó   zSparkContext.getOrCreateÚkeyÚvaluec                 C   r   r   r   )r   r   r   r   r   r   ÚsetSystemProperty   r   zSparkContext.setSystemPropertyc                 C   r   r   r   r   r   r   r   ÚapplicationId   r   zSparkContext.applicationIdc                 C   r   r   r   r   r   r   r   ÚdefaultMinPartitions"   r   z!SparkContext.defaultMinPartitionsc                 C   r   r   r   r   r   r   r   ÚdefaultParallelism&   r   zSparkContext.defaultParallelismc                 C   r   r   r   r   r   r   r   Ú	startTime.   r   zSparkContext.startTimec                 C   r   r   r   r   r   r   r   ÚuiWebUrl2   r   zSparkContext.uiWebUrlc                 C   r   r   r   r   r   r   r   Úversion6   r   zSparkContext.versionc                 C   r   r   r   r   r   r   r   Ú__repr__:   ó   zSparkContext.__repr__Úpathc                 C   r   r   r   ©r   r'   r   r   r   Ú
addArchive@   r&   zSparkContext.addArchiveFÚ	recursivec                 C   r   r   r   )r   r'   r*   r   r   r   ÚaddFileC   r&   zSparkContext.addFilec                 C   r   r   r   r(   r   r   r   Ú	addPyFileF   r&   zSparkContext.addPyFilec                 C   r   r   r   r   r   r   r   ÚcancelAllJobsR   r&   zSparkContext.cancelAllJobsÚgroupIdc                 C   r   r   r   )r   r.   r   r   r   ÚcancelJobGroupU   r&   zSparkContext.cancelJobGroupc                 C   r   r   r   r(   r   r   r   Údump_profilesX   r&   zSparkContext.dump_profilesc                 C   r   r   r   r   r   r   r   ÚgetCheckpointDir^   r&   zSparkContext.getCheckpointDirc                 C   r   r   r   r   r   r   r   ÚgetConfa   r&   zSparkContext.getConfc                 C   r   r   r   )r   r   r   r   r   ÚgetLocalPropertyd   r&   zSparkContext.getLocalPropertyÚdirNamec                 C   r   r   r   )r   r4   r   r   r   ÚsetCheckpointDir‚   r&   zSparkContext.setCheckpointDirc                 C   r   r   r   )r   r   r   r   r   ÚsetJobDescription…   r&   zSparkContext.setJobDescriptionÚdescriptionÚinterruptOnCancelc                 C   r   r   r   )r   r.   r7   r8   r   r   r   ÚsetJobGroupˆ   r&   zSparkContext.setJobGroupc                 C   r   r   r   )r   r   r   r   r   r   ÚsetLocalProperty‹   r&   zSparkContext.setLocalPropertyÚlogLevelc                 C   r   r   r   )r   r;   r   r   r   ÚsetLogLevelŽ   r&   zSparkContext.setLogLevelc                 C   r   r   r   r   r   r   r   Úshow_profiles‘   r&   zSparkContext.show_profilesc                 C   r   r   r   r   r   r   r   Ú	sparkUser”   r&   zSparkContext.sparkUser)r   Nr   )F)'Ú__name__Ú
__module__Ú__qualname__Ústrr   Úpropertyr   r   r   Úclassmethodr   r   r   r   r   Úintr    r!   r"   r#   r$   r%   r)   Úboolr+   r,   r-   r/   r0   r1   r2   r3   r5   r6   r9   r:   r<   r=   r>   r   r   r   r   r   	   sL    


r   )
Útypingr   r	   r   Ú#duckdb.experimental.spark.exceptionr   Úduckdb.experimental.spark.confr   r   Ú__all__r   r   r   r   Ú<module>   s     
