o
    ̳i                     @   s   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	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ G dd deZdS )	    )FileDownload)HttpRedirect)	Operation)KaggleHttpClient)ApiCancelKernelSessionRequestApiCancelKernelSessionResponseApiCreateKernelSessionRequestApiDeleteKernelRequestApiDeleteKernelResponseApiDownloadKernelOutputRequest!ApiDownloadKernelOutputZipRequestApiGetKernelRequestApiGetKernelResponse ApiGetKernelSessionStatusRequest!ApiGetKernelSessionStatusResponseApiListKernelFilesRequestApiListKernelFilesResponse!ApiListKernelSessionOutputRequest"ApiListKernelSessionOutputResponseApiListKernelsRequestApiListKernelsResponseApiSaveKernelRequestApiSaveKernelResponsec                   @   s   e Zd ZdefddZddedefddZddede	fd	d
Z
ddedefddZddedefddZddedefddZddedefddZddedefddZddedefddZddedefddZdde de!fddZ"dde#de$fddZ%dS )KernelsApiClientclientc                 C   s
   || _ d S N)_client)selfr    r   b/home/ubuntu/.local/lib/python3.10/site-packages/kagglesdk/kernels/services/kernels_api_service.py__init__	   s   
zKernelsApiClient.__init__Nrequestreturnc                 C       |du rt  }| jdd|tS )z
    Args:
      request (ApiListKernelsRequest):
        The request object; initialized to empty instance if not specified.
    Nkernels.KernelsApiServiceListKernels)r   r   callr   r   r!   r   r   r   list_kernels      zKernelsApiClient.list_kernelsc                 C   r#   )z
    Args:
      request (ApiListKernelFilesRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   ListKernelFiles)r   r   r&   r   r'   r   r   r   list_kernel_files   r)   z"KernelsApiClient.list_kernel_filesc                 C   r#   )z
    Args:
      request (ApiGetKernelRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   	GetKernel)r   r   r&   r   r'   r   r   r   
get_kernel$   r)   zKernelsApiClient.get_kernelc                 C   r#   )z
    Args:
      request (ApiSaveKernelRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   
SaveKernel)r   r   r&   r   r'   r   r   r   save_kernel0   r)   zKernelsApiClient.save_kernelc                 C   r#   )z
    Args:
      request (ApiListKernelSessionOutputRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   ListKernelSessionOutput)r   r   r&   r   r'   r   r   r   list_kernel_session_output<   r)   z+KernelsApiClient.list_kernel_session_outputc                 C   r#   )z
    Args:
      request (ApiGetKernelSessionStatusRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   GetKernelSessionStatus)r   r   r&   r   r'   r   r   r   get_kernel_session_statusH   r)   z*KernelsApiClient.get_kernel_session_statusc                 C   r#   )z
    Meant for use by Kaggle Hub (http bindings and terminology align with that)

    Args:
      request (ApiDownloadKernelOutputRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   DownloadKernelOutput)r   r   r&   r   r'   r   r   r   download_kernel_outputT      	z'KernelsApiClient.download_kernel_outputc                 C   r#   )z
    Meant for use by Kaggle Hub (and DownloadKernelOutput above)

    Args:
      request (ApiDownloadKernelOutputZipRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   DownloadKernelOutputZip)r   r   r&   r   r'   r   r   r   download_kernel_output_zipb   r6   z+KernelsApiClient.download_kernel_output_zipc                 C   r#   )z
    Args:
      request (ApiDeleteKernelRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   DeleteKernel)r	   r   r&   r
   r'   r   r   r   delete_kernelp   r)   zKernelsApiClient.delete_kernelc                 C   r#   )z
    Args:
      request (ApiCancelKernelSessionRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   CancelKernelSession)r   r   r&   r   r'   r   r   r   cancel_kernel_session|   r)   z&KernelsApiClient.cancel_kernel_sessionc                 C   r#   )z
    Args:
      request (ApiCreateKernelSessionRequest):
        The request object; initialized to empty instance if not specified.
    Nr$   CreateKernelSession)r   r   r&   r   r'   r   r   r   create_kernel_session   r)   z&KernelsApiClient.create_kernel_sessionr   )&__name__
__module____qualname__r   r    r   r   r(   r   r   r+   r   r   r-   r   r   r/   r   r   r1   r   r   r3   r   r   r5   r   r   r8   r	   r
   r:   r   r   r<   r   r   r>   r   r   r   r   r      s    r   N)$kagglesdk.common.types.file_downloadr   $kagglesdk.common.types.http_redirectr   !kagglesdk.common.types.operationsr   kagglesdk.kaggle_http_clientr   +kagglesdk.kernels.types.kernels_api_servicer   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   objectr   r   r   r   r   <module>   s    T