∋d!ddlmZddlZddlZddlmZddlmZd'd(d Zd)dZ d*dZ d+d,dZ d+d-dZ d'd.dZ d*dZd*dZd*dZd/d#Z d0d1d%Z d0d2d&ZdS)3) annotationsN)registry)get_ip_interfaceFaddrstrstrictboolreturntuple[str, str]ct|tstd|jjz|d\}}}|r|sd|z}t ||stj d}||fS)z Split address into its scheme and scheme-dependent location string. >>> parse_address('tcp://127.0.0.1') ('tcp', '127.0.0.1') If strict is set to true the address must have a scheme. zexpected str, got %r://zKInvalid url scheme. Must include protocol like tcp://localhost:8000. Got %szdistributed.comm.default-scheme) isinstancer TypeError __class____name__ rpartition ValueErrordaskconfigget)rrschemeseplocmsgs ;lib/python3.11/site-packages/distributed/comm/addressing.py parse_addressr s dC J.1HHIIIu--FC c   oo D!BCC 3;rrc|d|S)zb Undo parse_address(). >>> unparse_address('tcp', '127.0.0.1') 'tcp://127.0.0.1' r )rrs runparse_addressr #s    rc.tt|S)z Canonicalize address, adding a default scheme if necessary. >>> normalize_address('tls://[::1]') 'tls://[::1]' >>> normalize_address('[::1]') 'tcp://[::1]' )r r)rs rnormalize_addressr"-s M$// 00raddressstr | tuple[str, int] default_portstr | int | Nonetuple[str, int]cttrSfd}fd}dvrd\}drdddd\}}}|s ||s |}n`|ds ||dd}n6d\}}}|s |}|}nd|vr ||t |fS) zB Parse an endpoint address given in the form "host:port". c*tdd)Nzinvalid address z'; maybe: ipv6 needs brackets like [::1]r)r#sr_failzparse_host_port.._failBs# Qw Q Q Q   rc0tdS)Nzmissing port number in address r*)r#r%sr_defaultz!parse_host_port.._defaultGs&  JwJJKK Krr [N]:)rtuplesplit startswith partitionrint) r#r%r+r-_hostrtailports `` rparse_host_portr;9s^ '5!!       ]]5)) 7#"!""+//44c4  EGGG 8::DD??3'' 8DD",,S11c4 D8::DD D[[ EGGG T?rr8r: int | NonecVd|vr|dsd|d}||d|S|S)z! Undo parse_host_port(). r1r.r0)r4)r8r:s runparse_host_portr>fsI d{{4??3//{4{{{  rct||\}}tj|} ||S#t$rt d|wxYw)a[ Get a (host, port) tuple out of the given address. For definition of strict check parse_address ValueError is raised if the address scheme doesn't allow extracting the requested information. >>> get_address_host_port('tcp://1.2.3.4:80') ('1.2.3.4', 80) >>> get_address_host_port('tcp://[::1]:80') ('::1', 80) )rz4don't know how to extract host and port for address )rr get_backendget_address_host_portNotImplementedErrorr)rrrrbackends rrArArs{ V444KFC"6**G ,,S111     K4 K K    s ?Acxt|\}}tj|}||S)a  Return a hostname / IP address identifying the machine this address is located on. In contrast to get_address_host_port(), this function should always succeed for well-formed addresses. >>> get_address_host('tcp://1.2.3.4:80') '1.2.3.4' )rrr@get_address_hostrrrrCs rrErEs9 %%KFC"6**G  # #C ( ((rct|\}}tj|}t|||S)ah Get a local listening address suitable for reaching *addr*. For instance, trying to reach an external TCP address will return a local TCP address that's routable to that external address. >>> get_local_address_for('tcp://8.8.8.8:1234') 'tcp://192.168.1.68' >>> get_local_address_for('tcp://127.0.0.1:1234') 'tcp://127.0.0.1' )rrr@r get_local_address_forrFs rrHrHsC %%KFC"6**G 67#@#@#E#E F FFrct|\}}tj|}t|||S)a Apply scheme-specific address resolution to *addr*, replacing all symbolic references with concrete location specifiers. In practice, this can mean hostnames are resolved to IP addresses. >>> resolve_address('tcp://localhost:8786') 'tcp://127.0.0.1:8786' )rrr@r resolve_addressrFs rrJrJsC %%KFC"6**G 67#:#:3#?#? @ @@rhost_arg str | Noneport_argr6ct|pd\}}t|||n|\}}|dkr*|r(|t|krtd|d|t ||}t ||}|S)zF Process the *host* and *port* CLI options. Return a URI. Nrz)port number given twice in options: host z and port )rr;r6rr>r )rKrMr%rrr8r:rs ruri_from_host_portrPs B//KFC  -XX<JD$  qyyXy$#h--"7"7 3 3 3(0 3 3   D$ ' 'C 63 ' 'D Krlistc d}td||||fDr)fdtt|||||fDSt||||gS)aGet a list of addresses if the inputs are lists This is like ``address_from_user_args`` except that it also accepts lists for some of the arguments. If these arguments are lists then it will map over them accordingly. Examples -------- >>> addresses_from_user_args(host="127.0.0.1", protocol=["inproc", "tcp"]) ["inproc://127.0.0.1:", "tcp://127.0.0.1:"] cft|ttfr|Stj|SN)rr2rQ itertoolsrepeat)objs rlistifyz)addresses_from_user_args..listifys. cE4= ) ) )J#C(( (rc3NK|] }t|ttfV!dSrT)rr2rQ).0xs r z+addresses_from_user_args..s0 S SA:a% ' ' S S S S S Src Fg|]\}}}}t||||S))r8r: interfaceprotocolpeersecurityr%)address_from_user_args)rZhpiprr%r`ras r z,addresses_from_user_args..sW    1a #!)       r)anyzipmaprb)r8r:r^r_r`rar%rXs ``` raddresses_from_user_argsrks*)))  S S$i1R S S SSS         #C$i1R$S$ST     #dIxx    rc|r |jr|sd}|r.|dr|d\}}}|dkr|s|s|rtddS|r"|rtd||t |}|r|r d|vr|dz|z}|s|rt |||}nd}|r!|dz|ddz}|S) z?Get an address to listen on from common user provided argumentstlsr inplacez=Can not specify inproc protocol and host or port or interfacez inproc://z'Can not specify both interface and hostrO)require_encryptionendswithrrrrPr3) r8r:r^r_r`rar%r7rs rrbrbsBH/4H%%e,,4!,,U33!Q9  4 9 O ;/  /F SWXX X#I..D'D'U$..%$& t!$l;;8%$**U"3"3B"77 Kr)F)rrrr r r )rrrrr r)rrr rrT)r#r$r%r&r r')r8rr:r<r r)rrrr r r')rKrLrMrLr%r6r r)NNNNNNr)r rQ)r r) __future__rrUrdistributed.commrdistributed.utilsrrr r"r;r>rArErHrJrPrkrbrrrrus"""""" %%%%%%......0 1 1 1 1FJ*****Z          , ) ) ) ) GGGG" A A A A<    - - - - - b    *******r