U /e!@sddlmZddlZddlZddlmZddlmZd/dddd d d Zdddd d dZ dddddZ d0ddddddZ d1ddddddZ d2dddd ddZ dddddZdddd d!Zdddd"d#Zd$d$d%dd&d'd(Zd3d)d*d+d,Zd4dd*d-d.ZdS)5) annotationsN)registry)get_ip_interfaceFstrboolztuple[str, str])addrstrictreturncCsZt|tstd|jj|d\}}}|rB|sBd|}t||sRtj 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)rrschemeseplocmsgr?/tmp/pip-unpacked-wheel-g426oqom/distributed/comm/addressing.py parse_address s  r)rrr cCs|d|S)zb Undo parse_address(). >>> unparse_address('tcp', '127.0.0.1') 'tcp://127.0.0.1' r r)rrrrrunparse_address#sr)rr cCs tt|S)z Canonicalize address, adding a default scheme if necessary. >>> normalize_address('tls://[::1]') 'tls://[::1]' >>> normalize_address('[::1]') 'tcp://[::1]' )rr)rrrrnormalize_address-s rzstr | tuple[str, int]zstr | int | Noneztuple[str, int])address default_portr c sttrSfdd}fdd}dkr>d\}drddd \}}}|sj||sv|}q|d s||dd}n.d \}}}|s|}|}nd |kr||t|fS) zB Parse an endpoint address given in the form "host:port". cstdddS)Nzinvalid address z'; maybe: ipv6 needs brackets like [::1]rr)rrr_failBs zparse_host_port.._failcsdkrtdS)Nzmissing port number in address rrrrrr_defaultGsz!parse_host_port.._defaultr [N]:)r tuplesplit startswith partitionrint) rrr r"_hostrtailportrr!rparse_host_port9s,    r0z int | None)r-r/r cCs<d|kr|dsd|d}|dk r4|d|S|SdS)z! Undo parse_host_port(). r&r#r%N)r))r-r/rrrunparse_host_portfs  r1cCsNt||d\}}t|}z ||WStk rHtd|YnXdS)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 N)rr get_backendget_address_host_portNotImplementedErrorr)rrrrbackendrrrr3rs   r3cCs t|\}}t|}||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' )rrr2get_address_hostrrrr5rrrr6s  r6cCs&t|\}}t|}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' )rrr2rget_local_address_forr7rrrr8s  r8cCs&t|\}}t|}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' )rrr2rresolve_addressr7rrrr9s  r9z str | Noner+)host_argport_argrr cCsnt|pd\}}t||dk r |n|\}}|dkrV|rV|t|krVtd|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 )rr0r+rr1r)r:r;rrrr-r/rrrruri_from_host_ports   r=list)r csddd}tdd||||fDrJfddtt|||||fDSt||||gSdS)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:"] cSs t|ttfr|St|SdSN)r r'r> itertoolsrepeat)objrrrlistifysz)addresses_from_user_args..listifycss|]}t|ttfVqdSr?)r r'r>).0xrrr sz+addresses_from_user_args..c s*g|]"\}}}}t||||dqS))r-r/ interfaceprotocolpeersecurityr)address_from_user_args)rDhpiprrrIrJrr s z,addresses_from_user_args..N)anyzipmaprK)r-r/rGrHrIrJrrCrrPraddresses_from_user_argss rUc Cs|r|jr|sd}|r0|dr0|d\}}}|dkrR|sD|sD|rNtdndS|rp|rhtd||nt|}|r|rd|kr|d|}|s|rt|||}nd}|r|d|dd}|S) z?Get an address to listen on from common user provided argumentsZtlsr Zinplacez=Can not specify inproc protocol and host or port or interfacez inproc://z'Can not specify both interface and hostr<)Zrequire_encryptionendswithrrrr=r() r-r/rGrHrIrJrr,rrrrrKs,   rK)F)N)N)F)NNNNNNr)NNNNNNr) __future__rr@rZdistributed.commrZdistributed.utilsrrrrr0r1r3r6r8r9r=rUrKrrrrs>     - 1