3 uQh@sVddlZddlZddlZddlZddlmZddlmZddlm Z GdddZ dS)N) TLRequest)MessageContainer) TLMessagec@s0eZdZdZddZddZddZdd Zd S) MessagePackeraN This class packs `RequestState` as outgoing `TLMessages`. The purpose of this class is to support putting N `RequestState` into a queue, and then awaiting for "packed" `TLMessage` in the other end. The simplest case would be ``State -> TLMessage`` (1-to-1 relationship) but for efficiency purposes it's ``States -> Container`` (N-to-1). This addresses several needs: outgoing messages will be smaller, so the encryption and network overhead also is smaller. It's also a central point where outgoing requests are put, and where ready-messages are get. cCs(||_tj|_tj|_|t|_dS)N) _state collectionsdeque_dequeasyncioEvent_ready__name___log)selfstateloggersrE/tmp/pip-build-2nz6shyl/telethon/telethon/extensions/messagepacker.py__init__s  zMessagePacker.__init__cCs|jj||jjdS)N)r appendr set)rrrrrrs zMessagePacker.appendcCs|jj||jjdS)N)r extendr r)rZstatesrrrr#s zMessagePacker.extendcs|js |jj|jjIdHtj}g}d}x|joDt|tjkr|jj }|t|j t j 7}|tj kr|jj||j t|jt|jr|jjndd|_|j||jjd|j|jjjt|jq2|r|jj|P|jjd|jjjt|j |jjtdd}q2q2W|sd St|dkrvt j!dtj"t||j#}tj}|jj||d d }x|D] }||_$qfW|j#}||fS) z Returns (batch, data) if one or more items could be retrieved. If the cancellation occurs or only invalid items were in the queue, (None, None) will be returned instead. Nr)Zafter_idzAssigned msg_id = %d to %s (%x)z:Message payload for %s is too long (%d) and cannot be sentzRequest payload is too bigzs