o 6 f @sUdZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z d d lmZd d lmZd Zd ed<d%ddZd&ddZe GdddZifd'dd Zed(d#d$ZdS))z5 Internal utilities to manipulate connection strings ) annotationsN)Any) lru_cache) ip_address) dataclass) TypeAlias)pq)errorszdict[str, Any]rConnDictparamsreturnlist[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]gSt|d krg||9}g}t|D]#}}|r{|||d<|r|||d<|r|||d <||qm|S)zV Split connection parameters with a sequence of hosts into separate attempts. keystrr list[str]cst|}|r |dSgS)N,) get_paramsplit)rvalr P/home/ertert/spirit/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)leneOperationalErrormaxrangecopyappend) 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,paramdefenvrrrrFs rc@s*eZdZUdZded<ded<ded<dS)ParamDefzG Information about defaults and env vars for connection params rkeywordr2r-compiledN)__name__ __module__ __qualname____doc____annotations__rrrrr5\s  r5r6_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)r6r2r7) r Conninfo get_defaultsr5r6decoder2r7r1)r6r=defsdcdrrrr.gs   r.sboolcCs$zt|WdStyYdSw)z7Return True if the string represent a valid ip address.FT)r ValueError)rGrrr is_ip_addressxs   rJ)r r r r)r r r,rr r-)r6rr=r>r r?)rGrr rH)r; __future__rr/typingr functoolsr ipaddressr dataclassesrtyping_extensionsrr@r r rr r<r+rr5r.rJrrrrs$           0