o
    ?D©iÿ  ã                   @   s|   d dl mZmZmZ ddlmZ erddlmZ ddlm	Z	 ee
eeef Zee ZG dd„ dƒZG dd	„ d	ƒZd	dgZd
S )é    )ÚTYPE_CHECKINGÚOptionalÚUnioné   )Ú
StructType©Ú	DataFrame)ÚSparkSessionc                   @   s(   e Zd Zd
dd„Zdeddfdd	„ZdS )ÚDataStreamWriterÚ	dataframer   c                 C   ó
   || _ d S ©N)r   )Úselfr   © r   ú`/var/www/Datamplify/venv/lib/python3.10/site-packages/duckdb/experimental/spark/sql/streaming.pyÚ__init__   ó   
zDataStreamWriter.__init__Ú
table_nameÚreturnNc                 C   s   t ‚r   )ÚNotImplementedError)r   r   r   r   r   ÚtoTable   s   zDataStreamWriter.toTable)r   r   )Ú__name__Ú
__module__Ú__qualname__r   Ústrr   r   r   r   r   r
      s    
r
   c                   @   sN   e Zd Zddd„Z			ddee dee deeedf d	ed
df
dd„Z	dS )ÚDataStreamReaderÚsessionr	   c                 C   r   r   )r   )r   r   r   r   r   r      r   zDataStreamReader.__init__NÚpathÚformatÚschemaÚoptionsr   r   c                 K   s   ddl m} t‚)Nr   r   )Ú'duckdb.experimental.spark.sql.dataframer   r   )r   r   r   r   r    r   r   r   r   Úload   s   zDataStreamReader.load)r   r	   )NNN)
r   r   r   r   r   r   r   r   ÚOptionalPrimitiveTyper"   r   r   r   r   r      s     
üþýüûúr   N)Útypingr   r   r   Útypesr   r   r   r   r	   ÚboolÚfloatÚintr   ÚPrimitiveTyper#   r
   r   Ú__all__r   r   r   r   Ú<module>   s    	