*JfR dZddlmZddlZddlZddlmZddlmZ ddl m Z m Z m Z mZddl mZejd ZddZddZdS)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict get_param is_ip_address get_param_def)split_attemptspsycopgparamsrreturnlist[ConnDict]c d}g}t|D]z} |t|&#t$rH}td|dt||}Yd}~sd}~wwxYw|s%|sJtj t|t|ddkrt||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_excattemptsattemptexs ]/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/_conninfo_attempts.pyconninfo_attemptsr"sHH!&))  OO.w77 8 8 8 8    LL8'++f:M:MsSUww W W WHHHHHH  0 X///-..(:: Os": B >BB ctd}|r#|ds|dddkrgStd}|rgSt|r id|igStd}|std}|r|jpd}t j||t jt j }fd |DS) a> Perform 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)prototypec8g|]}id|ddiS)r'r).0itemr s r! z&_resolve_hostnames..Zs1 ? ? ?4 .v .z471: . . ? ? ?) r startswithr r compiledsocket getaddrinfo IPPROTO_TCP SOCK_STREAM)r rr'r(port_defanss` r!rr5s VV $ $D 4??3''4!9+;+;x,,HxT.,6,:t,,-- VV $ $D 8 ((-H-7   d&,63E   C @ ? ? ?3 ? ? ??r2)r rrr)__doc__ __future__rr5loggingrrrr_conninfo_utilsrrr r r getLoggerrr"rr.r2r!rAs #""""" NNNNNNNNNNNN++++++  9 % %B%@%@%@%@%@%@r2