o ]<0@sxdZddlZddlZddlZddlZddlZdZdZzddlmZej ej Bej Bej BZWn e y9dZYnwzddlmZWn e yMeZYnwddlmZddlmZdd lmZmZdd lmZdd lmZmZdd lmZmZm Z m!Z!dd l"m#Z#e$dj%Z&          dddZ'e$dj%Z(efddZ)esddZ*nddZ*ddZ+Gddde,Z-dS)z&Internal network layer helper methods.NT)pollF)error)_decode_all_selective)PY3)helpersmessage)MAX_MESSAGE_SIZE) decompress_NO_COMPRESSION) AutoReconnectNotMasterErrorOperationFailure ProtocolError) _UNPACK_REPLYztd||7}||ks|S)Nrrw) rx memoryview recv_intorzr{r|r}r~r )r?rormv bytes_read chunk_lengthr_rcrcrdrjs"  cCs$t|dr|jS|jr|jdSdS)Nr}r)hasattrr}args)r_rcrcrdr| s  r|c@seZdZddZddZdS) SocketCheckercCs*tr t|_t|_dSd|_d|_dS)N) _HAS_POLL threadingLock_lockr_poller)selfrcrcrd__init__s    zSocketChecker.__init__c Cs zB|jr6|j%|j|tz|jd}W|j|n|j|wWdn1s0wYn t|gggd\}}}Wn@ttfyMt yVYdSt t fyz}zt |t jt jfvroWYd}~qWYd}~dSd}~wtyYdSwt|dkS)zHReturn True if we know socket has been closed, False otherwise. TrN)rrregister _EVENT_MASKr unregisterselect RuntimeErrorrl ValueError _SELECT_ERRORrzr|r}r~EAGAINr7r)rr?rdrpr_rcrcrd socket_closeds2    zSocketChecker.socket_closedN)__name__ __module__ __qualname__rrrcrcrcrdrs r) TNNFNNNFNNFFN).__doc__r'r}rstructrrrrPOLLINPOLLPRIPOLLERRPOLLHUP ImportErrorrrr{bsonrbson.py3compatrpymongorrpymongo.commonrpymongo.compression_supportr r pymongo.errorsr r r rpymongo.messagerStructunpackrirerkr3rjr|objectrrcrcrcrds^          %