a 1$eR@sndZddlmZmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZmZmZmZmZmZmZmZddlmZmZmZddlmZmZGdd d e jZeZee Z!Gd d d Z"d d Z#ddZ$ddZ%ddZ&ddZ'ddZ(GdddZ)ddZ*ddZ+ddZ,e j-d d!eGd"d#d#Z.d2d&d'Z/d(ej0vre 1Z2e2re!3d)e2ed(e2n@e 4d*re!3d+n&e 5Z2e2red(e2e!3d,e2d-ej0vrej0d-Z2ee2nd.ej0vrej0d.Z2ee2nhe1r.e1Z2e!3d/e2ee2n>> with Env(CPL_DEBUG=True, CPL_CURL_VERBOSE=True): ... with fiona.open("zip+https://example.com/a.zip") as col: ... print(col.profile) For access to secured cloud resources, a Fiona Session or a foreign session object may be passed to the constructor. >>> import boto3 >>> from fiona.session import AWSSession >>> boto3_session = boto3.Session(...) >>> with Env(AWSSession(boto3_session)): ... with fiona.open("zip+s3://example/a.zip") as col: ... print(col.profile aws_access_key_idNzaPassing abstract session keyword arguments is deprecated. Pass a Fiona AWSSession object instead.aws_secret_access_keyaws_session_token region_nameAWS_ACCESS_KEY_IDAWS_SECRET_ACCESS_KEYzdGDAL's AWS config options can not be directly set. AWS credentials are handled exclusively by boto3.zNPassing a boto3 session is deprecated. Pass a Fiona AWSSession object instead.session)r#r$r%r& profile_name aws_unsigned)popwarningswarnr isdisjointr isinstancer aws_or_dummyr*issubsetosenvironkeysZ from_environrcopyoptionscontext_options) rr*r,r+Z session_classr8r#r$r%r&rrrrtsF5          z Env.__init__cOs&t}|jfi|t|i|S)auCreate an environment with default config options Parameters ---------- args : optional Positional arguments for Env() kwargs : optional Keyword arguments for Env() Returns ------- Env Notes ----- The items in kwargs will be overlaid on the default values. )rr"update)r!argskwargsr8rrr from_defaultsszEnv.from_defaultscCs.|j}|jjfi|tfi|dS)zGet credentials and configure GDAL Note well: this method is a no-op if the GDAL environment already has credentials, unless session is not None. Returns ------- None N)r*Zget_credential_optionsr8r:setenv)rZ cred_optsrrr credentializes zEnv.credentializecCs tjS)z'Return a mapping of registered drivers.)localrdriversrrrrrAsz Env.driverscCs tjS)zdWrites descriptions of open datasets to stderr For debugging and testing purposes. )r@r_dump_open_datasetsrrrrrBszEnv._dump_open_datasetscCstjdurjd|_it_dd|jD}|D]"}t|dd}|dur.|tj|<q.tfi|ji|_nd|_t |_t fi|j| |S)NFcSsh|]}|dkr|qS)Z RASTERIO_ENVr).0krrr z Env.__enter__.. normalizeT) r@r_has_parent_envrr8r6r defenvr9getenvr>r?)rZ probe_envkeyvalrrr __enter__s   z Env.__enter__cCsRt|jr$ttfi|jn*tjrHtj\}}t||ddq$dt_dS)NFrG) delenvrIrJr>r9r@rpopitemr )rexc_typeexc_valexc_tbrLrMrrr__exit__&sz Env.__exit__)NNN)rrr__doc__ classmethodr"rr2rr=r?rArBrNrTrrrrr@s   h rcKs.tjs tt_tjjfi|tjdS)z*Create a default environment if necessary.N)r@rrupdate_config_optionsstartr8rrrrJ5srJcCs tjstdn tjjSdS)z!Get a mapping of current options.No GDAL environment existsN)r@rrr8r7rrrrrK>s rKcCs ttjSr)boolr@rrrrrhasenvFsr\cKs&tjstdntjjfi|dS)z(Set options in the existing environment.rZN)r@rrrWrYrrrr>Js r>cCs(tdttjduo&tdddDS)Nz)Please use Env.session.hascreds() insteadcss|]}|tjvVqdSr)r@rZget_config_options)rCrLrrr Tszhascreds..)r'r()r.r/rr@rallrrrrhascredsRs r_cCs.tjstdn tjtjdt_dS)z+Delete options in the existing environment.rZN)r@rrZclear_config_optionsstoprrrrrOZs    rOc@s$eZdZddZddZddZdS)NullContextManagercCsdSrrrrrrrfszNullContextManager.__init__cCs|SrrrrrrrNiszNullContextManager.__enter__cGsdSrr)rr;rrrrTlszNullContextManager.__exit__N)rrrrrNrTrrrrraesracCstjr tStSdS)zjReturn an Env if one does not exist Returns ------- Env or a do-nothing context manager N)r@rrarr=rrrrenv_ctx_if_neededpsrbcstfdd}|S)ahA decorator that ensures an env exists before a function calls any GDAL C functions. Parameters ---------- f : function A function. Returns ------- A function wrapper. Notes ----- If there is already an existing environment, the wrapper does nothing and immediately calls f with the given arguments. csNtjr|i|St|i|WdS1s@0YdSr)r@rrr=)r;r<frrwrappers zensure_env..wrapperrrdrerrcr ensure_env~srhcstfdd}|S)aEnsures a config environment exists and has credentials. Parameters ---------- f : function A function. Returns ------- A function wrapper. Notes ----- The function wrapper checks the first argument of f and credentializes the environment if the first argument is a URI with scheme "s3". If there is already an existing environment, the wrapper does nothing and immediately calls f with the given arguments. cstjr t}ntj}|ddp$|d}t|trXt|}tjrP| t rPt }|}nt }||d|i|WdS1s0YdS)Nfprr)) r@rrr=getr1strrZ cls_from_pathr_rKr)r;kwdsZenv_ctorZfp_argZ session_clsr*rcrrres   z,ensure_env_with_credentials..wrapperrfrgrrcrensure_env_with_credentialssrmT)slotsc@s|eZdZdZejdejedZ ejdejedZ ddZ ddZ dd Z d d Zed d ZeddZddZdS) GDALVersiona Convenience class for obtaining GDAL major and minor version components and comparing between versions. This is highly simplistic and assumes a very normal numbering scheme for versions and ignores everything except the major and minor components. r)default validatorcCs|j|jft|j|jkSrmajorminortuplerotherrrr__eq__szGDALVersion.__eq__cCs|j|jft|j|jkSrrrrvrrr__lt__szGDALVersion.__lt__cCsd|jd|jdS)NzGDALVersion(major=z, minor=)rsrtrrrr__repr__szGDALVersion.__repr__cCs|jd|jS)N.r{rrrr__str__szGDALVersion.__str__cCs|t||r|St|tr ||St|trptd|}|sHtd|dd|dD\}}|||dSt ddS) a+ Parses input tuple or string to GDALVersion. If input is a GDALVersion instance, it is returned. Parameters ---------- input: tuple of (major, minor), string, or instance of GDALVersion Returns ------- GDALVersion instance z ^\d+\.\d+z;value does not appear to be a valid GDAL version number: {}css|]}t|VqdSr)int)rCcrrrr]rFz$GDALVersion.parse..r}r{z5GDALVersion can only be parsed from a string or tupleN) r1rurkresearch ValueErrorformatgroupsplit TypeError)r!inputmatchrsrtrrrparses     zGDALVersion.parsecCs |tS)z*Return GDALVersion of current GDAL runtime)rr r rrrruntimeszGDALVersion.runtimecCs|j|}||kSr) __class__rrvrrrat_least s zGDALVersion.at_leastN)rrrrUattrib validators instance_ofrrsrtrxryr|r~rVrrrrrrrros  roFcsdur0durtdttttfs0tdttkrNdndr`dnfdd}|S) aA decorator that ensures the called function or parameters are supported by the runtime version of GDAL. Raises GDALVersionError if conditions are not met. Examples:  @require_gdal_version('2.2') def some_func(): calling `some_func` with a runtime version of GDAL that is < 2.2 raises a GDALVersionErorr.  @require_gdal_version('2.2', param='foo') def some_func(foo='bar'): calling `some_func` with parameter `foo` of any value on GDAL < 2.2 raises a GDALVersionError.  @require_gdal_version('2.2', param='foo', values=('bar',)) def some_func(foo=None): calling `some_func` with parameter `foo` and value `bar` on GDAL < 2.2 raises a GDALVersionError. Parameters ------------ version: tuple, string, or GDALVersion param: string (optional, default: None) If `values` are absent, then all use of this parameter with a value other than default value requires at least GDAL `version`. values: tuple, list, or set (optional, default: None) contains values that require at least GDAL `version`. `param` is required for `values`. is_max_version: bool (optional, default: False) if `True` indicates that the version provided is the maximum version allowed, instead of requiring at least that version. reason: string (optional: default: '') custom error message presented to user in addition to message about GDAL version. Use this to provide an explanation of what changed if necessary context to the user. Returns --------- wrapped function Nz8require_gdal_version: param must be provided with valuesz:require_gdal_version: values must be a tuple, list, or setz>=z<= c s&tfdd}|S)Ncskr rrkrdur6tdtt}|}|jrn|tt|jdt |||j rttt |jt |j }ni}|vrdur҈|vs||krtdn&|vrtd||i|S)NzGDAL version must be {} {}{}z-usage of parameter "{}" requires GDAL {} {}{}z'parameter "{}={}" requires GDAL {} {}{}) rrrkrr7r;r:dictziplendefaultsreversed)r;rlZargspecZ full_kwdsr)rd inequalityis_max_versionparamreasonrvaluesversionrrreUsJ  " z8require_gdal_version..decorator..wrapperrfrgrrrrrrrrcr decoratorTs0z'require_gdal_version..decorator)rr1rulistsetrorr)rrrrrrrrrrequire_gdal_versions4 4rZ GDAL_DATAz$GDAL data found in package: path=%r.z header.dxfz0GDAL data files are available at built-in paths.z,GDAL data found in other locations: path=%r.Z PROJ_DATAZPROJ_LIBz$PROJ data found in package: path=%r.z0PROJ data files are available at built-in paths.z,PROJ data found in other locations: path=%r.)NNFr)7rU functoolsrrinspectrloggingr4r threadingr.rZ fiona._envrrrrr r r r r Z fiona.errorsrrrZ fiona.sessionrrr@r getLoggerrlogrrJrKr\r>r_rOrarbrhrmsrorr5Z search_wheelpathdebugZ find_filerhas_datarrrrst ,  v   1 C {