o Ni @s~UddlZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZgd ZdZeeeed <z&ddlZejejjejd kradd lmZndd lmZed Wn eefyzdZYnwia e!e"Z#deeddfddZ$e eee fD]Z%e$e%qedure$edde&deee&e&fdefddZ'dee&fddZ(dS)N)DictOptionalSetType)GsClient)hf_get_from_cache) HttpClient)R2Client)S3Client) SchemeClient)rr r r r r BeakerClientz2.0)r clientreturncCsTt|jtr|jD]}|t|<q dSt|jtr|t|j<dStd|d|j)z Add a new :class:`SchemeClient`. This can be used to extend :func:`cached_path.cached_path()` to handle custom schemes, or handle existing schemes differently. zUnexpected type for z scheme: N) isinstanceschemetuple_SCHEME_TO_CLIENTstr ValueError)rrrP/home/ubuntu/.local/lib/python3.10/site-packages/cached_path/schemes/__init__.pyadd_scheme_client"s    rresourceheaderscCsR|dd}t|t}|r%t|tr|||dSd|j}t|||S)a Get the right client for the given resource. Parameters ---------- resource : str The URL or path to the resource. headers : Optional[Dict[str, str]], optional Custom headers to add to HTTP requests, by default None. Example: {"Authorization": "Bearer YOUR_TOKEN"} for private resources. Only used for HTTP/HTTPS resources. Returns ------- SchemeClient The appropriate client for the resource. z://r)rz9Headers are only supported for HTTP/HTTPS resources, got )splitrgetr issubclass__name__loggerwarning)rr maybe_scheme client_clsmsgrrrget_scheme_client9s     r$cCsttdhBS)z+ Return all supported URL schemes. hf)setrkeysrrrrget_supported_schemesXsr()N))loggingtypingrrrrpackaging.version packaginggsrr%rhttpr r2r s3r scheme_clientr __all__r __annotations__beaker.versionbeakerversionparseVERSION beaker_v1append ImportErrorModuleNotFoundErrorr getLoggerrrrrrr$r(rrrrs:          $