jdZddlmZddlZddlZddlmZddlmZ ddl m Z m Z ddl mZmZmZmZ ddlZejd ZddZddZdS)z8 Separate connection attempts from a connection string. ) annotationsN)shuffle)errors)ConnDict ConnMapping) get_param get_param_def is_ip_addresssplit_attemptsTpsycopgparamsrreturnlist[ConnDict]c&Kd}g}t|ddkx}rd|D}t|D]} |t |d{V,#t $rR}t jd|dd|}t d|Yd}~d}~wwxYw|s|sJ|t|d d krt||rd |D|z}|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-standbyc&i|]\}}|dk ||S)r).0kvs j/workspace/show-city-briefing-agent/.venv/lib/python3.11/site-packages/psycopg/_conninfo_attempts_async.py z+conninfo_attempts_async..'s)QQQ41aQ:P5P5P!Q5P5P5Pzfailed to resolve host hostz: z%sload_balance_hostsrandomcg|] }i|ddi S)rstandbyr)ras r z+conninfo_attempts_async..=s6   9: 4q 4() 4 4   r) r itemsr extend_resolve_hostnamesOSErroreOperationalErrorgetloggerdebugr)rlast_excattemptsprefer_standbyattemptexs rconninfo_attempts_asyncr0sHH"6+ABBFVVV~RQQ6<<>>QQQ!&)))) ) OO"4W"="======= > > > > ) ) ))G'++f*=*=GG2GGH LLx ( ( ( ( ( ( ( (  ) -..(::  >F    Os (A66 CAC  CrcKtd}|r#|ds|dddkrgStdrgSt|r id|igStdx}std}|r|jpd} t j}|||tj tj d {V}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:hostaddrport5432T)prototypeNc8g|]}id|ddiS)r5rr)ritemrs rr!z&_resolve_hostnames..ns1 ? ? ?4 .v .z471: . . ? ? ?r) r startswithr r compiledasyncioget_running_loop getaddrinfosocket IPPROTO_TCP SOCK_STREAM)rrr6port_defloopanss` rr$r$Ds: VV $ $D 4??3''4!9+;+;x$$xT.,6,:t,,--ff-- -D8 ((-H-7 '))$$ $f0v7I%         @ ? ? ?3 ? ? ??r)rrrr)rrrr)__doc__ __future__rrBloggingrrrr&abcrr_conninfo_utilsr r r r r? getLoggerr)r0r$rrrrOs #""""" &&&&&&&&TTTTTTTTTTTTNNN  9 % %****Z*@*@*@*@*@*@r