pj dZddlmZddlZddlZddlmZddlmZ ddl m Z m Z ddl mZmZmZdd l mZejd ZddZddZdS)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict ConnMapping) get_param is_ip_address get_param_def)split_attemptspsycopgparamsrreturnlist[ConnDict]cd}g}t|D]m} |t|&#t$r;}td|d||}Yd}~fd}~wwxYw|s%|sJtjt|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_hostnamesOSErrorloggerdebuggeteOperationalErrorstrr r)rlast_excattemptsattemptexs j/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/psycopg/_conninfo_attempts.pyconninfo_attemptsr#sHH!&))  OO.w77 8 8 8 8    LL8'++f:M:Mr R R RHHHHHH  0x X///-..(:: Os": A?1A::A?rctd}|r#|ds|dddkrgStd}|rgSt|r id|igStd}|std}|r|jpd}t j||t jt j }fd |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)prototypec8g|]}id|ddiS)r(r).0itemrs r" z&_resolve_hostnames..`s1 ? ? ?4 .v .z471: . . ? ? ?) r startswithr r compiledsocket getaddrinfo IPPROTO_TCP SOCK_STREAM)rrr(r)port_defanss` r"rr:s VV $ $D 4??3''4!9+;+;x,,HxT.,6,:t,,-- VV $ $D 8 ((-H-7   d&,63E   C @ ? ? ?3 ? ? ??r3)rrrr)rrrr)__doc__ __future__rr6loggingrrrrabcrr_conninfo_utilsr r r r getLoggerrr#rr/r3r"rCs #""""" &&&&&&&&DDDDDDDDDD++++++  9 % %B&@&@&@&@&@&@r3