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	 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 d dlmZ d dlmZ d dl m!Z! G dd„ de"ƒZ#dS )é    )ÚInboxFileClient)ÚBenchmarksApiClient)ÚBlobApiClient)ÚOperationsClient)ÚCompetitionApiClient)ÚDatasetApiClient)ÚEducationApiClient)ÚKernelsApiClient)ÚModelApiClient)ÚModelClient)ÚSearchApiClient)Ú	IamClient)ÚOAuthClient)ÚAccountClient)ÚGroupApiClient)Ú	KaggleEnv)ÚKaggleHttpClientc                   @   s  e Zd ZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG d	d
„ d
eƒZG dd„ deƒZ	G dd„ deƒZ
G dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZd-dedededed ed!efd"d#„Zd$efd%d&„Zd$efd'd(„Zd)d*„ Zd+d,„ ZdS ).ÚKaggleClientc                   @   ó   e Zd Zdefdd„ZdS )zKaggleClient.AdminÚhttp_clientc                 C   ó   t |ƒ| _d S ©N)r   Úinbox_file_client©Úselfr   © r   úS/home/jeff/fluffinator/venv/lib/python3.10/site-packages/kagglesdk/kaggle_client.pyÚ__init__   ó   zKaggleClient.Admin.__init__N©Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   ÚAdmin   ó    r#   c                   @   r   )zKaggleClient.Benchmarksr   c                 C   r   r   )r   Úbenchmarks_api_clientr   r   r   r   r      r   z KaggleClient.Benchmarks.__init__Nr   r   r   r   r   Ú
Benchmarks   r$   r&   c                   @   r   )zKaggleClient.Blobsr   c                 C   r   r   )r   Úblob_api_clientr   r   r   r   r      r   zKaggleClient.Blobs.__init__Nr   r   r   r   r   ÚBlobs   r$   r(   c                   @   r   )zKaggleClient.Commonr   c                 C   r   r   )r   Úoperations_clientr   r   r   r   r   "   r   zKaggleClient.Common.__init__Nr   r   r   r   r   ÚCommon!   r$   r*   c                   @   r   )zKaggleClient.Competitionsr   c                 C   r   r   )r   Úcompetition_api_clientr   r   r   r   r   &   r   z"KaggleClient.Competitions.__init__Nr   r   r   r   r   ÚCompetitions%   r$   r,   c                   @   r   )zKaggleClient.Datasetsr   c                 C   r   r   )r   Údataset_api_clientr   r   r   r   r   *   r   zKaggleClient.Datasets.__init__Nr   r   r   r   r   ÚDatasets)   r$   r.   c                   @   r   )zKaggleClient.Educationr   c                 C   r   r   )r   Úeducation_api_clientr   r   r   r   r   .   r   zKaggleClient.Education.__init__Nr   r   r   r   r   Ú	Education-   r$   r0   c                   @   r   )zKaggleClient.Kernelsr   c                 C   r   r   )r	   Úkernels_api_clientr   r   r   r   r   2   r   zKaggleClient.Kernels.__init__Nr   r   r   r   r   ÚKernels1   r$   r2   c                   @   r   )zKaggleClient.Modelsr   c                 C   ó   t |ƒ| _t|ƒ| _d S r   )r
   Úmodel_api_clientr   Úmodel_clientr   r   r   r   r   6   ó   
zKaggleClient.Models.__init__Nr   r   r   r   r   ÚModels5   r$   r7   c                   @   r   )zKaggleClient.Searchr   c                 C   r   r   )r   Úsearch_api_clientr   r   r   r   r   ;   r   zKaggleClient.Search.__init__Nr   r   r   r   r   ÚSearch:   r$   r9   c                   @   r   )zKaggleClient.Securityr   c                 C   r3   r   )r   Ú
iam_clientr   Úoauth_clientr   r   r   r   r   ?   r6   zKaggleClient.Security.__init__Nr   r   r   r   r   ÚSecurity>   r$   r<   c                   @   r   )zKaggleClient.Usersr   c                 C   r3   r   )r   Úaccount_clientr   Úgroup_api_clientr   r   r   r   r   D   r6   zKaggleClient.Users.__init__Nr   r   r   r   r   ÚUsersC   r$   r?   NFÚ ÚenvÚverboseÚusernameÚpasswordÚ	api_tokenÚ
user_agentc           	   	   C   sÂ   t |||||||d | _}t |¡| _t |¡| _t |¡| _t 	|¡| _
t |¡| _t |¡| _t |¡| _t |¡| _t |¡| _t |¡| _t |¡| _t |¡| _|| _|| _|| _d S )N)rC   rD   rE   rF   Úresponse_processor)r   Ú_http_clientr   r#   Úadminr&   Ú
benchmarksr(   Úblobsr*   Úcommonr,   Úcompetitionsr.   Údatasetsr0   Ú	educationr2   Úkernelsr7   Úmodelsr9   Úsearchr<   Úsecurityr?   ÚusersrC   rD   rE   )	r   rA   rB   rC   rD   rE   rF   rG   r   r   r   r   r   H   s    
zKaggleClient.__init__Úreturnc                 C   s   | j S r   )rH   ©r   r   r   r   r   Z   s   zKaggleClient.http_clientc                 C   s   | j j ¡ S r   )rI   Úadmin_clientÚrenew_iap_tokenrV   r   r   r   Ú_renew_iap_token]   s   zKaggleClient._renew_iap_tokenc                 C   s   | j  ¡  | S r   )rH   Ú	__enter__rV   r   r   r   rZ   `   s   
zKaggleClient.__enter__c                 C   s   | j  |||¡ d S r   )rH   Ú__exit__)r   Úexc_typeÚ	exc_valueÚtbr   r   r   r[   d   s   zKaggleClient.__exit__)NFNNNr@   N)r    r!   r"   Úobjectr#   r&   r(   r*   r,   r.   r0   r2   r7   r9   r<   r?   r   ÚboolÚstrr   r   r   rY   rZ   r[   r   r   r   r   r      s$    $r   N)$Ú+kagglesdk.admin.services.inbox_file_servicer   Ú4kagglesdk.benchmarks.services.benchmarks_api_servicer   Ú)kagglesdk.blobs.services.blob_api_servicer   Ú,kagglesdk.common.services.operations_servicer   Ú7kagglesdk.competitions.services.competition_api_servicer   Ú/kagglesdk.datasets.services.dataset_api_servicer   Ú2kagglesdk.education.services.education_api_servicer   Ú.kagglesdk.kernels.services.kernels_api_servicer	   Ú+kagglesdk.models.services.model_api_servicer
   Ú'kagglesdk.models.services.model_servicer   Ú,kagglesdk.search.services.search_api_servicer   Ú'kagglesdk.security.services.iam_servicer   Ú)kagglesdk.security.services.oauth_servicer   Ú(kagglesdk.users.services.account_servicer   Ú*kagglesdk.users.services.group_api_servicer   Úkagglesdk.kaggle_envr   Úkagglesdk.kaggle_http_clientr   r_   r   r   r   r   r   Ú<module>   s$    