o
    Ďi                     @   s   d dl mZ d dlmZ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mZmZmZmZmZ d dlmZ G dd de Z!dS )    )HttpRedirect)ApiDirectoryContentApiFilesSummary)ApiCreateDatasetRequestApiCreateDatasetResponse"ApiCreateDatasetVersionByIdRequestApiCreateDatasetVersionRequest
ApiDatasetApiDeleteDatasetRequestApiDeleteDatasetResponseApiDownloadDatasetRawRequestApiDownloadDatasetRequest ApiGetDatasetFilesSummaryRequestApiGetDatasetMetadataRequestApiGetDatasetMetadataResponseApiGetDatasetRequestApiGetDatasetStatusRequestApiGetDatasetStatusResponseApiListDatasetFilesRequestApiListDatasetFilesResponseApiListDatasetsRequestApiListDatasetsResponseApiListTreeDatasetFilesRequestApiUpdateDatasetMetadataRequest ApiUpdateDatasetMetadataResponseApiUploadDatasetFileRequestApiUploadDatasetFileResponse)KaggleHttpClientc                   @   sF  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%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 )&DatasetApiClientclientc                 C   s
   || _ d S N)_client)selfr    r#   k/home/jeff/fluffinator/venv/lib/python3.10/site-packages/kagglesdk/datasets/services/dataset_api_service.py__init__   s   
zDatasetApiClient.__init__Nrequestreturnc                 C       |du rt  }| jdd|tS )z
    Args:
      request (ApiListDatasetsRequest):
        The request object; initialized to empty instance if not specified.
    Ndatasets.DatasetApiServiceListDatasets)r   r!   callr   r"   r&   r#   r#   r$   list_datasets      zDatasetApiClient.list_datasetsc                 C   r(   )z
    Args:
      request (ApiGetDatasetRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   
GetDataset)r   r!   r+   r	   r,   r#   r#   r$   get_dataset   r.   zDatasetApiClient.get_datasetc                 C   r(   )z
    Args:
      request (ApiListDatasetFilesRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   ListDatasetFiles)r   r!   r+   r   r,   r#   r#   r$   list_dataset_files#   r.   z#DatasetApiClient.list_dataset_filesc                 C   r(   )z
    Args:
      request (ApiListTreeDatasetFilesRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   ListTreeDatasetFiles)r   r!   r+   r   r,   r#   r#   r$   list_tree_dataset_files/   r.   z(DatasetApiClient.list_tree_dataset_filesc                 C   r(   )z
    Args:
      request (ApiGetDatasetMetadataRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   GetDatasetMetadata)r   r!   r+   r   r,   r#   r#   r$   get_dataset_metadata;   r.   z%DatasetApiClient.get_dataset_metadatac                 C   r(   )z
    Args:
      request (ApiUpdateDatasetMetadataRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   UpdateDatasetMetadata)r   r!   r+   r   r,   r#   r#   r$   update_dataset_metadataG   r.   z(DatasetApiClient.update_dataset_metadatac                 C   r(   )z
    Args:
      request (ApiDownloadDatasetRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   DownloadDataset)r   r!   r+   r   r,   r#   r#   r$   download_datasetS   r.   z!DatasetApiClient.download_datasetc                 C   r(   )aA  
    Note: This API method has extremely low usage, and can be considered for
    deprecation. The existing DownloadDataset RPC, with `raw=true` set on the
    request, is equivalent.

    Args:
      request (ApiDownloadDatasetRawRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   DownloadDatasetRaw)r   r!   r+   r   r,   r#   r#   r$   download_dataset_raw_   s   z%DatasetApiClient.download_dataset_rawc                 C   r(   )z
    Args:
      request (ApiCreateDatasetVersionRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   CreateDatasetVersion)r   r!   r+   r   r,   r#   r#   r$   create_dataset_versiono   r.   z'DatasetApiClient.create_dataset_versionc                 C   r(   )z
    Args:
      request (ApiCreateDatasetVersionByIdRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   CreateDatasetVersionById)r   r!   r+   r   r,   r#   r#   r$   create_dataset_version_by_id{   r.   z-DatasetApiClient.create_dataset_version_by_idc                 C   r(   )z
    Args:
      request (ApiCreateDatasetRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   CreateDataset)r   r!   r+   r   r,   r#   r#   r$   create_dataset   r.   zDatasetApiClient.create_datasetc                 C   r(   )z
    Args:
      request (ApiGetDatasetStatusRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   GetDatasetStatus)r   r!   r+   r   r,   r#   r#   r$   get_dataset_status   r.   z#DatasetApiClient.get_dataset_statusc                 C   r(   )z
    Args:
      request (ApiGetDatasetFilesSummaryRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   GetDatasetFilesSummary)r   r!   r+   r   r,   r#   r#   r$   get_dataset_files_summary   r.   z*DatasetApiClient.get_dataset_files_summaryc                 C   r(   )z
    Deprecated. Use the new unified BlobApiService#StartBlobUpload rpc.

    Args:
      request (ApiUploadDatasetFileRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   UploadDatasetFile)r   r!   r+   r   r,   r#   r#   r$   upload_dataset_file   s   	z$DatasetApiClient.upload_dataset_filec                 C   r(   )z
    Args:
      request (ApiDeleteDatasetRequest):
        The request object; initialized to empty instance if not specified.
    Nr)   DeleteDataset)r
   r!   r+   r   r,   r#   r#   r$   delete_dataset   r.   zDatasetApiClient.delete_datasetr    )/__name__
__module____qualname__r   r%   r   r   r-   r   r	   r0   r   r   r2   r   r   r4   r   r   r6   r   r   r8   r   r   r:   r   r<   r   r   r>   r   r@   r   rB   r   r   rD   r   r   rF   r   r   rH   r
   r   rJ   r#   r#   r#   r$   r      s"    r   N)"$kagglesdk.common.types.http_redirectr   9kagglesdk.datasets.databundles.types.databundle_api_typesr   r   ,kagglesdk.datasets.types.dataset_api_servicer   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   kagglesdk.kaggle_http_clientr   objectr   r#   r#   r#   r$   <module>   s
    h