o wi@sddlZddlZddlZddlmZgdZGdddeZGdddeZddd d fd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZdS)-N)parser) get_env get_envbool get_envint get_envfloatget_envdecimal get_envdateget_envdatetime get_envlist get_envdict CoercionErrorRequiredSettingMissingErrorc eZdZdZfddZZS)r z3Custom error raised when a value cannot be coerced.cs$d|||j}tt||dS)Nz"Unable to coerce '{}={}' using {}.)format__name__superr __init__)selfkeyvaluefuncmsg __class__W/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/utils/env_var_parsing.pyrBszCoercionError.__init__r __module__ __qualname____doc__r __classcell__rrrrr ?r cr)r z7Custom error raised when a required env var is missing.csd|}tt||dS)Nz!Required env var '{}' is missing.)rrr r)rrrrrrrJs z$RequiredSettingMissingError.__init__rrrrrr Gr!r cC|SNrxrrrOr&FcCs\ztj|}Wnty|durt||YSwz||WSty-t|||w)a Return env var coerced into a type other than string. This function extends the standard os.getenv function to enable the coercion of values into data types other than string (all env vars are strings by default). Args: key: string, the name of the env var to look up Kwargs: default: the default value to return if the env var does not exist. NB the default value is **not** coerced, and is assumed to be of the correct type. coerce: a function that is used to coerce the value returned into another type required: bool, if True, then a RequiredSettingMissingError error is raised if the env var does not exist. Returns the env var, passed through the coerce function T)osenvironKeyErrorr Exceptionr )rdefaultcoercerequiredrrrr_get_envOs    r/cCs$t|tr|S|dup|dv S)N)false0nonfnone) isinstanceboollowerrrrr_boolos r:cCt|Sr#)intr9rrr_intvr=cCr;r#)floatr9rrr_floatzr>r@cC t|Sr#)decimalDecimalr9rrr_decimal~ rDcCrAr#)jsonloadsr9rrr_dictrErHcCrAr#)rparser9rrr _datetimerErJcCst|Sr#)rrIdater9rrr_datesrLcOsPt|dvs Jd|ddd}t|dk}|s|dnd}t||||dS) a Return env var. This is the parent function of all other get_foo functions, and is responsible for unpacking args/kwargs into the values that _get_env expects (it is the root function that actually interacts with environ). Args: key: string, the env var name to look up. default: (optional) the value to use if the env var does not exist. If this value is not supplied, then the env var is considered to be required, and a RequiredSettingMissingError error will be raised if it does not exist. Kwargs: coerce: a func that may be supplied to coerce the value into something else. This is used by the default get_foo functions to cast strings to builtin types, but could be a function that returns a custom class. Returns the env var, coerced if required, and a default if supplied. )rzToo many args supplied.r-cSr"r#rr$rrrr&r'zget_env..rN)r,r-r.)lengetr/)rr,kwargsrr.rrrrs  rcGt|g|RdtiS)zReturn env var cast as boolean.r-)rr:rr,rrrrrcGrQ)zReturn env var cast as integer.r-)rr=rRrrrrrSrcGrQ)zReturn env var cast as float.r-)rr@rRrrrrrSrcGrQ)zReturn env var cast as Decimal.r-)rrDrRrrrrrSrcGrQ)zReturn env var as a date.r-)rrLrRrrrrrSrcGrQ)zReturn env var as a datetime.r-)rrJrRrrrr rSr cs*|ddt|g|RdfddiS)zReturn env var as a list. separator r-cs |Sr#)splitr$rTrrr&s zget_envlist..)rOr)rr,rPrrWrr s r cGrQ)zReturn env var as a dict.r-)rrHrRrrrr rSr )rBrFr(dateutilr__all__r+r r r/r:r=r@rDrHrJrLrrrrrrr r r rrrrs0)