o DDi@sdZddlmZddlZddlZddlmZddlmZ ddl m Z m Z ddl mZmZmZmZ ddlZed ZdddZdddZdS)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict ConnMapping) get_param get_param_def is_ip_addresssplit_attemptsTpsycopgparamsrreturnlist[ConnDict]c sd}g}t|ddk}rdd|D}t|D]4}z |t|IdHWqtyO}ztd|dd|}t d |WYd}~qd}~ww|sX|sVJ|t|d d krct ||rnd d |D|}|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. Ntarget_session_attrszprefer-standbycSsi|] \}}|dkr||qS)r).0kvrrY/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/_conninfo_attempts_async.py 'sz+conninfo_attempts_async..zfailed to resolve host hostz: z%sload_balance_hostsrandomcSsg|] }i|ddiqS)rstandbyr)rarrr =sz+conninfo_attempts_async..) ritemsr extend_resolve_hostnamesOSErroreOperationalErrorgetloggerdebugr)r last_excattemptsprefer_standbyattemptexrrrconninfo_attempts_asyncs6  r+rcstd}|r|ds|dddkrgStdr gSt|r-id|igStd}s?td}|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:hostaddrport5432T)prototypeNcs$g|]}id|ddiqS)r/rr)ritemr rrrns$z&_resolve_hostnames..) r startswithr r compiledasyncioget_running_loop getaddrinfosocket IPPROTO_TCP SOCK_STREAM)r rr0port_defloopansrr6rrDs"    r)r rrr)r rrr)__doc__ __future__rr<loggingrrrr!abcrr_conninfo_utilsrr r r r9 getLoggerr$r+rrrrrs     -