U ÁùÌ^!ã@sRddlmZddlmZmZddlmZmZdd„Zdd„Z d d „Z dd d „Z d S)é©Úunpack)ÚBytesIOÚbé)ÚdumpsÚloadscCstd|ƒdS)NzdSt| ¡ƒ}|S)zÑ Atomic read of a BSON message. This function either returns a dict, None, or raises a socket error. If the return value is None, it means the socket is closed by the other side. éN)Ú recvbytesr Úgetvaluer)rÚsock_bufZmessage_lengthÚretvalr r r Úrecvobjs   rNcCsn|dkrtƒ}d}||krj| t||dƒ¡}t|ƒ}t|ƒtkrJt|ƒ}|dkrVdS||7}| |¡q|S)a Atomic read of bytes_needed bytes. This function either returns exactly the nmber of bytes requested in a StringIO buffer, None, or raises a socket error. If the return value is None, it means the socket is closed by the other side. Nri€r)rÚrecvÚminÚlenÚtypeÚstrrÚwrite)rZ bytes_neededrZ bytes_countÚchunkÚ part_countr r r r*s   r)N) ÚstructrÚsixrrÚrrr rrrr r r r Ús