U ڲg@s<dZddlmZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z ddlmZz,ddlmZmZddlmZdd lmZWnek red YnXd d lmZd d lmZe rddlmZeZee_eZ ee _dddddZ!dddddZ"dddddZ#Gddde Z$GdddZ%dS)z DNS query support ) annotationsN)randint)Any DefaultDict NamedTupleSequence TYPE_CHECKING) defaultdict)ResolverCache)r ) DNSExceptionzBthe module psycopg._dns requires the package 'dnspython' installed)errors)conninfo)SRVdict[str, Any]paramsreturnc stdtg}g}g}t|IdHD]\}|ddk rH||d|ddk rd||d|ddk r(|t|dq(|}d |}|r||d<d |}|r||d<d |}|r||d<|S)a Perform async DNS lookup of the hosts and return a new params dict. .. deprecated:: 3.1 The use of this function is not necessary anymore, because `psycopg.AsyncConnection.connect()` performs non-blocking name resolution automatically. z@from psycopg 3.1, resolve_hostaddr_async() is not needed anymoreNhosthostaddrport,) warningswarnDeprecationWarningrZconninfo_attempts_asyncgetappendstrcopyjoin) rhostsZ hostaddrsZportsattemptoutZshostsZ shostaddrsZsportsr$0/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/_dns.pyresolve_hostaddr_async's2    r&cCs t|S)z/Apply SRV DNS lookup as defined in :RFC:`2782`.)Rfc2782Resolverresolverr$r$r% resolve_srvNsr*cst|IdHS)z$Async equivalent of `resolve_srv()`.N)r' resolve_asyncr)r$r$r%resolve_srv_asyncSsr,c@s6eZdZUded<ded<dZded<dZded <dS) HostPortrrrFbooltotryNz str | Nonetarget)__name__ __module__ __qualname____annotations__r/r0r$r$r$r%r-Xs  r-c@seZdZdZedZdddddZdddddZdd dd d Z d d d ddZ d d d ddZ d dd dddZ dd ddddZ dddddZdS)r'zImplement SRV RR Resolution as per RFC 2782 The class is organised to minimise code duplication between the sync and the async paths. z9^(?P_[^\.]+)\.(?P_[^\.]+)\.(?P.+)rrcCsN||}|s|Sg}|D]&}|jr6|||q||q|||S)5Update the parameters host and port after SRV lookup.) _get_attemptsr/extend _resolve_srvr_return_paramsselfrattemptshpshpr$r$r%r(hs  zRfc2782Resolver.resolvecsT||}|s|Sg}|D],}|jr<|||IdHq||q|||S)r5N)r6r/r7_resolve_srv_asyncrr9r:r$r$r%r+ws  zRfc2782Resolver.resolve_asynczlist[HostPort]c Cs,|dtjddrgS|dtjdd}|d}t|dtjdd}|d}t|d krv|t|9}t|t|krtd t|d t|d g}d }t||D]d\}} |j |} | s| dkrd}| r| dnd} t || d| d} n t || d} || q|r(|SgS)z Return the list of host, and for each host if SRV lookup must be tried. Return an empty list if no lookup is requested. rZ PGHOSTADDRrZPGHOSTrrZPGPORTr z cannot match z hosts with z port numbersFsrvTr0N)rrr/r0rr)rosenvironsplitrleneOperationalErrorzip re_srv_rrmatchlowergroupr-r) r;rZhost_argZhosts_inZport_argZports_inr#Z srv_foundrrmr0r>r$r$r%r6s.       zRfc2782Resolver._get_attemptsr-)r>rcCs8zt|jd}Wntk r*d}YnX|||SNrr$)resolverr(rr _get_solved_entriesr;r>ansr$r$r%r8s  zRfc2782Resolver._resolve_srvcs>zt|jdIdH}Wntk r0d}YnX|||SrO)async_resolverr(rr rQrRr$r$r%r?s  z"Rfc2782Resolver._resolve_srv_asyncz Sequence[SRV])r>entriesrcCsd|s.|jr*|jdkr*t|j|jdgSgSt|dkrPt|djdkrPgSdd||DS)NrArBr r.cSs*g|]"}tt|jdt|jdqS)rVrB)r-rr0rstripr).0entryr$r$r% sz7Rfc2782Resolver._get_solved_entries..)r0rrLr-rFr sort_rfc2782)r;r>rUr$r$r%rQsz#Rfc2782Resolver._get_solved_entries)rr=rcCsJ|std|}ddd|D|d<ddd|D|d<|S)Nz!no host found after SRV RR lookuprcss|] }|jVqdSN)rrXr>r$r$r% sz1Rfc2782Resolver._return_params..rcss|]}t|jVqdSr\)rrr]r$r$r%r^sr)rGrHrr )r;rr=r#r$r$r%r9s  zRfc2782Resolver._return_paramsz list[SRV])rSrc Cstt}g}|D]}||j|qt|D]\}}t|dkrV||dq2|jdddtdd|D}|r2t d|}d} t |D]\} } | | j 7} | |krqq|| || j 8}|| =qxq2|S)zM Implement the priority/weight ordering defined in RFC 2782. r rcSs|jSr\weight)entr$r$r%z.Rfc2782Resolver.sort_rfc2782..)keycss|] }|jVqdSr\r_)rXrar$r$r%r^sz/Rfc2782Resolver.sort_rfc2782..) r listpriorityrsorteditemsrFsortsumr enumerater`) r;rSZ prioritiesr#rYprirUZ total_weightrZcsumirar$r$r%r[s*      zRfc2782Resolver.sort_rfc2782N)r1r2r3__doc__recompilerJr(r+r6r8r?rQr9r[r$r$r$r%r'_s ' r')&ro __future__rrCrprrandomrtypingrrrrr collectionsr Z dns.resolverr r Zdns.asyncresolverZ AsyncResolverZ dns.exceptionr ImportErrorr@rrGrZdns.rdtypes.IN.SRVrrPcacherTr&r*r,r-r'r$r$r$r%s8        '