U ڲg @sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dd l m Z mZdd l mZd d d ddZd dddddZe GdddZifddddddZedddddZdS) z5 Internal utilities to manipulate connection strings ) annotationsN) lru_cache) ip_address) dataclass)pq)ConnDict ConnMapping)errorsr zlist[ConnDict])paramsreturnc s0dddfdd }|d}|d}|d}|rd|rdt|t|krdtd t|d t|d tt|t|}d t|kr|krnn td t|d t|d|d krˆgSt|d kr||9}g}t|D]H}}|r|||d<|r|||d<|r |||d<||q|S)zV Split connection parameters with a sequence of hosts into separate attempts. strz list[str])keyr cst|}|r|dSgS)N,) get_paramsplit)rvalr ;/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/_conninfo_utils.py split_vals z!split_attempts..split_valhostZhostaddrportzcould not match z host names with z hostaddr valuesrz port numbers to z hosts)leneZOperationalErrormaxrangeappend) r rhostsZ hostaddrsZportsZnhostsrviattemptrrrsplit_attemptss8      r"r str | None)r namer cCsB||krt||St|}|s$dStj|j}|dk r>|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$ZparamdefenvrrrrCs rc@s*eZdZUdZded<ded<ded<dS)ParamDefzG Information about defaults and env vars for connection params r keywordr)r#compiledN)__name__ __module__ __qualname____doc____annotations__rrrrr+Ys r+zdict[str, ParamDef]zParamDef | None)r,_cacher cCsd|sZtj}|D]F}t|j|jr0|jnd|jdk rF|jndd}|||j<q||S)z? Return the ParamDef of a connection string parameter. N)r,r)r-) rZConninfoZ get_defaultsr+r,decoder)r-r()r,r3ZdefsdZcdrrrr%ds  r%bool)sr cCs(z t|Wntk r"YdSXdS)z7Return True if the string represent a valid ip address.FT)r ValueError)r8rrr is_ip_addressus  r:)r1 __future__rr& functoolsr ipaddressrZ dataclassesrr4rabcrr r rr"rr+r%r:rrrrs      0