qj dZddlmZddlZddlZddlmZddlmZ ddl m Z m Z ddl mZmZmZdd l mZej$d Zd d Zdd Zy)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict ConnMapping) get_param is_ip_address get_param_def)split_attemptspsycopgc\d}g}t|D]} |jt||s"|sJtjt|t|ddk(r t||S#t$r2}tj d|j d||}Yd}~d}~wwxYw)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_hostnamesOSErrorloggerdebuggeteOperationalErrorstrr r)paramslast_excattemptsattemptexs [/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_conninfo_attempts.pyconninfo_attemptsr!sHH!&)  OO.w7 8* x  X//-.(: O  LL8'++f:Mr RH sA00 B+9(B&&B+ct|d}|r|jds|dddk(r|gSt|d}|r|gSt|r i|d|igSt|d}|std}|xr |jxsd}t j ||t jt j }|Dcgc]}i|d|d d ic}Scc}w) 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)prototyper) r startswithr r compiledsocket getaddrinfo IPPROTO_TCP SOCK_STREAM)rrr&r'port_defansitems r rr:s VV $D 4??3'4!9+;x,HxT,6,:t,-- VV $D  (-H--7    d&,,63E3E Cr@sG # &DD+   9 %B&@r?