o oĎi@sddlZddlZddlZddlmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZddlmZmZmZddlmZddlmZed ed Zd ed e fd dZGdddeje eZdS)N)Path)TemporaryDirectory)GenericCallableIterableOptionalTupleTypeVarUnion)CloudImplementation CloudPathimplementation_registry) FileCacheMode)InvalidConfigurationExceptionBoundedCloudPath)boundkeyreturncsdtdtffdd }|S)Nclsrcs4t|ts td|t_t_t|_|S)Nz,Only subclasses of Client can be registered.) issubclassClient TypeErrorr _client_classname _cloud_metarrO/home/jeff/fluffinator/venv/lib/python3.10/site-packages/cloudpathlib/client.py decorators    z(register_client_class..decorator)type)rr rrrregister_client_classsr"c @seZdZUeed<dZddejfdee e e fdee e e j fdeefddZd/d d Zed0d d Zd/d dZde e efde defddZddZejdede e e j fdefddZejdedefddZejdededeeeeffddZej d1deded edefd!d"Z ejd1d#ed$eddfd%d&Z!ejde e e j fdedefd'd(Z"ejdede fd)d*Z#ej +d2ded,e$de fd-d.Z%dS)3rrNfile_cache_modelocal_cache_dircontent_type_methodcCsd|_d|_|jt|trt|}|durt}|dur-tj dd}|dkr-d}|r6|dur6tj }|tj krH|durHt dtj d|dur[t |_|jj}|dur[tj}t||_||_|durjtj}||_dS)NCLOUDPATHLIB_LOCAL_CACHE_DIRzIf you use the 'zP' cache mode, you must pass a `local_cache_dir` when you instantiate the client.)r#_cache_tmp_dirrvalidate_completeness isinstancestrrfrom_environmentosenvironget persistentrrrtmp_dirr_local_cache_dirr%)selfr#r$r%rrr__init__!s6      zClient.__init__rcCsDt|ddtjtjtjfvr||jr |jdSdSdS)Nr#) getattrrr1 close_filecloudpath_object clear_cacher2existsrmdirr3rrr__del__Us  zClient.__del__cCs|jdur ||_|jS)zGet the default client, which the one that is used when instantiating a cloud path instance for this cloud without a client specified. N)_default_clientrrrrget_default_clientbs zClient.get_default_clientcCs ||j_dS)zSet this client instance as the default one used when instantiating cloud path instances for this cloud without a client specified.N) __class__r=r;rrrset_as_default_clientks zClient.set_as_default_client cloud_pathpartscGs|jj|g|Rd|iS)Nclient)r path_class)r3rArBrrrr pszClient.CloudPathcCs>|jr|jD]}|r|q t|q dSdS)zqClears the contents of the cache folder. Does not remove folder so it can keep being written to. N)r2r9iterdiris_fileunlinkshutilrmtree)r3prrrr8ss   zClient.clear_cache local_pathcCdSNr)r3rArKrrr_download_file~zClient._download_filecCrLrMrr3rArrr_existszClient._exists recursivecCdS)aList all the files and folders in a directory. Parameters ---------- cloud_path : CloudPath The folder to start from. recursive : bool Whether or not to list recursively. Returns ------- contents : Iterable[Tuple] Of the form [(CloudPath, is_dir), ...] for every child of the dir. Nr)r3rArSrrr _list_dirszClient._list_dirTsrcdst remove_srccCrLrMr)r3rVrWrXrrr _move_filerOzClient._move_filepath missing_okcCrT)zRemove a file or folder from the server. Parameters ---------- path : CloudPath The file or folder to remove. Nr)r3rZr[rrr_removes zClient._removecCrLrMr)r3rKrArrr _upload_filerOzClient._upload_filecCrLrMrrPrrr_get_public_urlrRzClient._get_public_urlexpire_secondscCrLrMr)r3rAr`rrr_generate_presigned_urlrOzClient._generate_presigned_url)rN)rr)T)r_)&__name__ __module__ __qualname__r __annotations__r= mimetypes guess_typerr r+rr-PathLikerr4r< classmethodr>r@rr r8abcabstractmethodrrNboolrQrrrUrYr\r]r^intrarrrrrs  4          r)rjrfr-pathlibrrHtempfilertypingrrrrrr r cloudpathr r renumsr exceptionsrrr+r"ABCrrrrrs  $