o
    ñ²iû  ã                   @   st   d dl mZ ddlmZ erddlmZ ddlmZ ee	B e
B eB ZedB ZG dd„ dƒZG d	d
„ d
ƒZd
dgZdS )é    )ÚTYPE_CHECKINGé   )Ú
StructType)Ú	DataFrame)ÚSparkSessionNc                   @   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   ú[/home/ubuntu/.local/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                   @   sL   e Zd Zddd„Z			ddedB dedB d	eeB dB 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   ÚOptionalPrimitiveTyper    r   r   r   r   r      s     
üþý
üûúr   )Útypingr   Útypesr   r   r   r   r   ÚboolÚfloatÚintr   ÚPrimitiveTyper!   r   r   Ú__all__r   r   r   r   Ú<module>   s    	