o DDi\ @sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZmZd%ddZd&ddZe GdddZifd'ddZed(d d!Zd)d#d$ZdS)*z5 Internal utilities to manipulate connection strings ) annotationsN) lru_cache) ip_address) dataclass)errors)pq)ConnDict ConnMappingparamsr returnlist[ConnDict]c s*dfdd }|d}|d}|d }|r/|r/t|t|kr/td t|d t|d tt|t|}d t|krD|krVnntd t|dt|d|d kr_igSt|d kri||9}g}t|D]#}i}|r}|||d<|r|||d<|r|||d <||qo|S)zV Split connection parameters with a sequence of hosts into separate attempts. keystrr list[str]cst|}|r |dSgS)N,) get_paramsplit)rvalr P/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/_conninfo_utils.py split_vals z!split_attempts..split_valhosthostaddrportzcould not match z host names with z hostaddr valuesrz port numbers to z hostsN)rrr r)leneOperationalErrormaxrangeappend) r rhosts hostaddrsportsnhostsrviattemptrrrsplit_attemptss<        r)namer str | NonecCsB||vr t||St|}sdStj|j}dur|SdS)zi Return a value from a connection string. The value may be also specified in a PG* env var. N)r get_param_defosenvirongetenvvar)r r*paramdefenvrrrrCs  rc@s*eZdZUdZded<ded<ded<dS)ParamDefzG Information about defaults and env vars for connection params rkeywordr0r+compiledN)__name__ __module__ __qualname____doc____annotations__rrrrr3Ws  r3r4_cachedict[str, ParamDef]ParamDef | NonecCsd|s-tj}|D]#}t|j|jr|jnd|jdur#|jndd}|||j<q ||S)z? Return the ParamDef of a connection string parameter. N)r4r0r5) rConninfo get_defaultsr3r4decoder0r5r/)r4r;defsdcdrrrr,bs   r,sboolcCs$zt|WdStyYdSw)z7Return True if the string represent a valid ip address.FT)r ValueError)rErrr is_ip_addressss   rHr cCstt|dS)z7Return `true` if `gssencmode` was specified explicitly. gssencmode)rFrrrrrgssapi_requested}srJ)r r r r )r r r*rr r+)r4rr;r<r r=)rErr rF)r r r rF)r9 __future__rr- functoolsr ipaddressr dataclassesrr>rrrabcr r r)rr3r,rHrJrrrrs"        0