B ݁[@sddlZddlmZddlZddlZyddlmZWn ek rTddlmZYnXdddgZGddde Z ej d krd d Z nd d Z dddZ dddZddZddZddadddZdddZdS)N)partial) monotonic)timeNoWayToWaitForSocketError wait_for_readwait_for_writec@s eZdZdS)rN)__name__ __module__ __qualname__r r R/Users/kashii/Desktop/Projects/Workshops/worker-safety/lambda/urllib3/util/wait.pyr s)cCs||S)Nr )fntimeoutr r r _retry_on_intr)src Cs|dkrtd}n t|}xry||Sttjfk r}z@|jdtjkrRn(|t}|dkrhd}|tdkrxd}wWdd}~XYqXqWdS)Ninfr)floatrOSErrorselecterrorargserrnoZEINTR)rrZdeadlineer r r r-s    Fc Csd|s|stdg}g}|r&|||r4||ttj|||}t||\}}} t|p`|p`| S)Nz2must specify at least one of read=True, write=True) RuntimeErrorappendrrrbool) sockreadwriterZrcheckZwcheckrZrreadyZwreadyZxreadyr r r select_wait_for_socketDs  r cs^|s|stdd}|r"|tjO}|r0|tjO}t||fdd}tt||S)Nz2must specify at least one of read=True, write=Truercs|dk r|d9}|S)Ni)poll)t)poll_objr r do_pollcsz%poll_wait_for_socket..do_poll)rrPOLLINPOLLOUTr!registerrr)rrrrmaskr$r )r#r poll_wait_for_socketWs    r)cOs tddS)Nzno select-equivalent available)r)rkwargsr r r null_wait_for_socketksr+c Cs:yt}t|jdWnttfk r0dSXdSdS)NrFT)rr!rAttributeErrorr)r#r r r _have_working_pollos r-cOs*tr tanttdrtantat||S)Nr)r-r)wait_for_sockethasattrrr r+)rr*r r r r.|s  r.cCst|d|dS)z Waits for reading to be available on a given socket. Returns True if the socket is readable, or False if the timeout expired. T)rr)r.)rrr r r rscCst|d|dS)z Waits for writing to be available on a given socket. Returns True if the socket is readable, or False if the timeout expired. T)rr)r.)rrr r r rs)FFN)FFN)N)N)r functoolsrrsysrr ImportError__all__ Exceptionr version_inforr r)r+r-r.rrr r r r s&