o
    '¦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   ÚreturnNc                 C   ó
   || _ d S ©N)r
   )Úselfr
   © r   úe/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/duckdb/experimental/spark/sql/streaming.pyÚ__init__   ó   
zDataStreamWriter.__init__Ú
table_namec                 C   ó   t ‚r   ©ÚNotImplementedError)r   r   r   r   r   ÚtoTable   s   zDataStreamWriter.toTable)r
   r   r   N)Ú__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   r   Nc                 C   r   r   )r   )r   r   r   r   r   r      r   zDataStreamReader.__init__ÚpathÚformatÚschemaÚoptionsr   c                 K   r   r   r   )r   r   r   r    r!   r   r   r   Úload   s   zDataStreamReader.load)r   r   r   N)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    	