o 6 f @sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z mZmZddl mZed ZdddZdddZdS)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict get_param is_ip_address get_param_def)split_attemptspsycopgparamsrreturnlist[ConnDict]c sd}g}t|D].}z |t|IdHWq ty7}ztd|dt||}WYd}~q d}~ww|sE|s>Jt t|t |ddkrPt ||S)aSplit a set of connection params on the single attempts to perform. A connection param can perform more than one attempt more than one ``host`` is provided. Also perform async resolution of the hostname into hostaddr. Because a host can resolve to more than one address, this can lead to yield more attempts too. Raise `OperationalError` if no host could be resolved. Because the libpq async function doesn't honour the timeout, we need to reimplement the repeated attempts. Nzfailed to resolve host %r: %shostload_balance_hostsrandom) r extend_resolve_hostnamesOSErrorloggerdebuggetstreOperationalErrorrr)r last_excattemptsattemptexrY/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/_conninfo_attempts_async.pyconninfo_attempts_asyncs"  r!cstd}|r|ds|dddkrgStd}|r"gSt|r/id|igStd}|sAtd}|r?|jp@d}t}|j||tj tj d Id H}fd d |DS) aD Perform async DNS lookup of the hosts and return a list of connection attempts. If a ``host`` param is present but not ``hostname``, resolve the host addresses asynchronously. :param params: The input parameters, for instance as returned by `~psycopg.conninfo.conninfo_to_dict()`. The function expects at most a single entry for host, hostaddr because it is designed to further process the input of split_attempts(). :return: A list of attempts to make (to include the case of a hostname resolving to more than one IP). r/r:hostaddrport5432)prototypeNcs$g|]}id|ddiqS)r%rr).0itemr rr \s$z&_resolve_hostnames..) r startswithrr compiledasyncioget_running_loop getaddrinfosocket IPPROTO_TCP SOCK_STREAM)r rr%r&port_defloopansrr-r r6s$     r)r rr r)__doc__ __future__rr4r1loggingrrrr_conninfo_utilsrrrr r getLoggerrr!rrrrr s      !