B %Q[37@sdZddlmZmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZdd lmZmZmZdd lmZdd lmZmZmZmZmZe e e gZGd d d eZGdddeZdS) )datetimeMINYEAR)linesep)randint)sleep)FIRST ROUND_ROBINRANDOMSEQUENCE_TYPES STRING_TYPESget_config_parameter)LDAPUnknownStrategyErrorLDAPServerPoolErrorLDAPServerPoolExhaustedError)Server)log log_enabledERRORBASICNETWORKc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)ServerPoolStatecCsDg|_|j|_||_d|_|t|_tt r@t t d|dS)Nrz"instantiated ServerPoolState: <%r>) serversstrategy server_poollast_used_serverrefreshrnowZinitialize_timerrr)selfrr!UC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6i93dh7p\ldap3\ldap3\core\pooling.py__init__(s zServerPoolState.__init__cCsdt}|jr4x0|jD]}|t|dt7}qWn |dt7}|dt|jt7}|d|jdkrhdnt|j|jd7}|S)Nz servers: rNonezPool strategy: z - Last used server: )rrstrrr)r sserverr!r!r"__str__3s  *zServerPoolState.__str__cCsLg|_x*|jjD]}|j|ttdddgqWtdt|jd|_dS)NrTr)rrappendrrrlenr)r r(r!r!r"r?szServerPoolState.refreshcCs|j|jdS)Nr)rr)r r!r!r"get_current_serverEsz"ServerPoolState.get_current_servercCs"|jr|jjtkr4|jjr,|jdd|_qd|_n|jjtkr|jjr\||jd|_q|jdt|jkrz|jdnd|_nX|jjt kr|jjr| |_qt dt|jd|_n t t rtt d|jjtdt trttd|j|j|jdSt t rtt d|tddS) Nr)startingrz$unknown server pooling strategy <%s>zunknown server pooling strategyz&server returned from Server Pool: <%s>zno servers in Server Pool <%s>zno servers in server pool)rrrr activefind_active_serverrr r+r find_active_random_serverrrrrrrr)r r!r!r" get_serverHs,  &    zServerPoolState.get_servercCsL|jj}x|r(ttr&ttd||jdd}x|r|tdt|d}|dst |jj t rr|jj st |dj|jj krttr6ttd|dq6ttrttd|dt |d<ttrttd|d|drd|d<|j|Sd |d<q6Wt |jjt s |d8}q Wttr@ttd |td dS) Nz4entering loop for finding active server in pool <%s>rrrz8server <%s> excluded from checking because it is offlinezserver <%s> reinserted in poolz%checking server <%s> for availabilityTFzSno random active server available in Server Pool <%s> after maximum number of trieszNno random active server available in server pool after maximum number of tries)rr.rrrrpoprr+ isinstanceexhaustboolrrsecondscheck_availabilityindexrr)r counterZ temp_listr(r!r!r"r0js4  0      z)ServerPoolState.find_active_random_servercCstd}|jj}|t|jkr"d}x|rttrBttd||d}t|j}xn||dkr|d7}|||kr~||n |||}|j|dsXt|jj t r|jj st |j|dj |jj kr8ttrTt|jj t rttd|j|dqTttd|j|d|jj t |j|dj qTttrXttd |j|dt |j|d<ttrttd |j|d|j|drd |j|d<|Sd |j|d<qTWt|jjt s|d8}ttrttd |t|q&Wttrttd|tddS)NZPOOLING_LOOP_TIMEOUTrz@entering loop number <%s> for finding active server in pool <%s>r%rrz5server <%s> excluded from checking because is offlinezDserver <%s> excluded from checking because is offline for %d secondszserver <%s> reinserted in poolz%checking server <%s> for availabilityTFz9waiting for %d seconds before retrying pool servers cyclezLno active server available in Server Pool <%s> after maximum number of trieszGno active server available in server pool after maximum number of tries)rrr.r+rrrrr3r4r5rrr6r7rrr)r r-Zconf_pool_timeoutr9r8Z pool_sizeoffsetr!r!r"r/sH   84       z"ServerPoolState.find_active_servercCs t|jS)N)r+r)r r!r!r"__len__szServerPoolState.__len__N) __name__ __module__ __qualname__r#r)rr,r1r0r/r;r!r!r!r"r's  " +rc@sneZdZdeddfddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS) ServerPoolNTFcCs|tkr$ttrttd|td|rF|sFttr>ttdtdg|_t|_||_ ||_ t |t t fr|||nt |tr|t |||_ttrttd|dS)Nzunknown pooling strategy <%s>zunknown pooling strategyz3cannot instantiate pool with exhaust and not activez)POOLING_STRATEGIESrrrrrrdict pool_statesr.r4r3r raddr rr)r rZ pool_strategyr.r4r!r!r"r#s&    zServerPool.__init__cCsdt}|jr0x,|jD]}|t|t7}qWn |dt7}|dt|j7}|d|jrbt|jnd7}|d|jr~t|jnd7}|S)Nz servers: r$zPool strategy: z - active: Falsez - exhaust pool: )rrr&rr.r4)r r'r(r!r!r"r)s  zServerPool.__str__cCsd}|jrD|d7}x|jD]}||d7}qW|ddd}n|d7}|d|7}|d|7}|d |7}|d 7}|S) NzServerPool(servers=[z, ]r$z, pool_strategy={0.strategy!r}z, active={0.active!r}z, exhaust={0.exhaust!r}))r__repr__format)r rr(r!r!r"rIs zServerPool.__repr__cCs t|jS)N)r+r)r r!r!r"r;szServerPool.__len__cCs |j|S)N)r)r itemr!r!r" __getitem__szServerPool.__getitem__cCs |jS)N)r__iter__)r r!r!r"rNszServerPool.__iter__cCst|tr"||jkr|j|nt|tr>|jt|nt|trxz|D]T}t|trj|j|qNt|tr|jt|qNttrttd|t dqNWnttrttd|t dx|j D]}|j | qWdS)Nz,element must be a server in Server Pool <%s>z%server in ServerPool must be a ServerzLserver must be a Server of a list of Servers when adding to Server Pool <%s>z+server must be a Server or a list of Server) r3rrr*r r rrrrrBr)r rr( connectionr!r!r"rCs&          zServerPool.addcCsX||jkr|j|nttr.ttd||tdx|jD]}|j|q>WdS)Nz/server %s to be removed not in Server Pool <%s>zserver not in server pool)rremoverrrrrBr)r r(rOr!r!r"rPs  zServerPool.removecCst|}||j|<dS)N)rrB)r rOZ pool_stater!r!r" initializeszServerPool.initializecCs:||jkr|j|Sttr.ttd||tddS)Nz-connection <%s> not in Server Pool State <%s>z!connection not in ServerPoolState)rBr1rrrr)r rOr!r!r"r1$s  zServerPool.get_servercCs:||jkr|j|Sttr.ttd||tddS)Nz-connection <%s> not in Server Pool State <%s>z!connection not in ServerPoolState)rBr,rrrr)r rOr!r!r"r,,s  zServerPool.get_current_server)r<r=r>r r#r)rIr;rMrNrCrPrQr1r,r!r!r!r"r?s   r?N) __doc__rrosrrandomrtimerr r r r r r exceptionsrrrr(rZ utils.logrrrrrr@objectrr?r!r!r!r"s