jdZddlmZddlZddlZddlmZddlmZ ddl m Z m Z ddl mZmZmZmZ ddlZej&d Zd d Zd d Zy)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict ConnMapping) get_param get_param_def is_ip_addresssplit_attemptspsycopgc0Kd}g}t|ddk(x}r)|jDcic]\}}|dk7s ||}}}t|D]%} |jt |d{'|s|sJ|t|dd k(r t||r|Dcgc] }i|dd i c}|z}|Scc}}w7O#t $rJ}t jd|jdd|}tjd|Yd}~d}~wwxYwcc}ww) 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-standbyzfailed to resolve host hostz: z%sload_balance_hostsrandomstandby) r itemsr extend_resolve_hostnamesOSErroreOperationalErrorgetloggerdebugr) paramslast_excattemptsprefer_standbykvattemptexas g/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_conninfo_attempts_async.pyconninfo_attempts_asyncr's>HH"6+ABFVVV~V#)<<>Q>41aQ:P5P!Q$>Q!&) ) OO"4W"== >* x-.(:>F >F 4q 4() 4h  O5R> ))))'++f*=)@2$GH LLx ( (  )" s])D B3B3DB;)B9 *B;2*DD+D9B;; DAD D DDcKt|d}|r|jds|dddk(r|gSt|dr|gSt|r i|d|igSt|dx}std}|xr |jxsd} t j }|j||tjtj d {}|Dcgc]}i|d|d d ic}S7 cc}ww) 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)prototypeNr) r startswithr r compiledasyncioget_running_loop getaddrinfosocket IPPROTO_TCP SOCK_STREAM)rrr-port_defloopansitems r&rrDs  VV $D 4??3'4!9+;x$xT,6,:t,--ff- -D - (-H--7 '')$$ $f00v7I7I%  r?)__doc__ __future__rr7loggingrrrrabcrr_conninfo_utilsr r r r r4 getLoggerrr'rr&rIsH # &TT   9 %*Z*@rH