o
    iQ                     @   s   d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lm	Z	 ddlm
Z
 ddlmZ ddlmZ G dd dZdS )
    N   )	SSEClient   )ApiGetStatusRequest)ApiGetStatusResponse)ApiFetchContentsRequest)ApiFetchContentsResponse)ApiCreateReloadRequest)ApiCreateReloadResponse)ApiGetReloadEventSourceData)ApiGetReloadRequestc                   @   sX   e Zd ZdejfddZdefddZdefdd	Zded
efddZ	defddZ
dS )ReloadClienthttp_clientc                 C   s
   || _ d S )N)client)selfr    r   K/home/ubuntu/.local/lib/python3.10/site-packages/spaces/reloading/client.py__init__   s   
zReloadClient.__init__revisionc                 C   @   t |d}| jjd| d}|jdksJ |jt| S )N)r   z/get-statusjson   )r   r   post
model_dumpstatus_coder   model_validater   )r   r   reqresr   r   r   
get_status      
zReloadClient.get_statusfilepathc                 C   r   )N)r"   z/fetch-contentsr   r   )r   r   r   r   r   r	   r   r   )r   r"   r   r   r   r   r   fetch_contents   r!   zReloadClient.fetch_contentscontentsc                 C   sB   t ||d}| jjd| d}|jdksJ |jt| S )N)r"   r$   z/create-reloadr   r   )r
   r   r   r   r   r   r   r   )r   r"   r$   r   r   r   r   r   create_reload"   s   zReloadClient.create_reload	reload_idc                 c   s    t |d}| jjdd| d*}|jdksJ |jt|  D ]}|jdkr2t	
|jV  q$W d    d S 1 s>w   Y  d S )N)reloadIdPOSTz/get-reloadr   r   message)r   r   streamr   r   r   
iter_byteseventseventr   model_validate_jsondata)r   r&   r   r   r-   r   r   r   
get_reload(   s   

"zReloadClient.get_reloadN)__name__
__module____qualname__httpxClientr   strr    r#   r%   r0   r   r   r   r   r      s    r   )__doc__r4   _vendor.sseclientr   typesr   r   r   r	   r
   r   r   r   r   r   r   r   r   <module>   s    