MhxdZddlmZddlZddlmZddlmZddlm Z GddZ e Z dd gZ dS) z(Tracker for zero-copy messages with 0MQ.) annotationsN)Event)Frame)NotDonecPeZdZUdZded<ded<ddZed Zddd ZdS)MessageTrackeraA class for tracking if 0MQ is done using one or more messages. When you send a 0MQ message, it is not sent immediately. The 0MQ IO thread sends the message at some later time. Often you want to know when 0MQ has actually sent the message though. This is complicated by the fact that a single 0MQ message can be sent multiple times using different sockets. This class allows you to track all of the 0MQ usages of a message. Parameters ---------- towatch : Event, MessageTracker, zmq.Frame This objects to track. This class can track the low-level Events used by the Message class, other MessageTrackers or actual Messages. z set[Event]eventszset[MessageTracker]peerstowatch%tuple[MessageTracker | Event | Frame]ct|_t|_|D]}t|tr|j|2t|t r|j|bt|tr6|jstd|j|jtdt|dS)a\Create a message tracker to track a set of messages. Parameters ---------- *towatch : tuple of Event, MessageTracker, Message instances. This list of objects to track. This class can track the low-level Events used by the Message class, other MessageTrackers or actual Messages. zNot a tracked messagez&Require Events or Message Frames, not N) setr r isinstanceraddrrtracker ValueError TypeErrortype)selfr objs Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/zmq/sugar/tracker.py__init__zMessageTracker.__init__#see UU  V VC#u%% V $$$$C00 V s####C'' V{>$%<=== s{++++ Tc T TUUU V Vcp|jD]}|sdS|jD] }|jsdS dS)z9Is 0MQ completely done with the message(s) being tracked?FT)r is_setr done)revtpms rrzMessageTracker.done;s];  C::<< uu *  B7 uu trtimeout float | intctj}|dus|dkrd}n|}|jD]]}|dkrt|||sttj}|||z z}|}^|jD]B}|dkrt||tj}|||z z}|}CdS)aWait for 0MQ to be done with the message or until `timeout`. Parameters ---------- timeout : float default: -1, which means wait forever. Maximum time in (s) to wait before raising NotDone. Returns ------- None if done before `timeout` Raises ------ NotDone if `timeout` reached before I am done. Fri: )r N)timer rwaitrr )rr tic remainingrtocpeers rr$zMessageTracker.waitFs&ikk e  w{{%III;  C1}} HHYH ' ' '::<<  )++C s "ICCJ  D1}} IIiI ( ( ()++C s "ICC   rN)r r )r)r r!) __name__ __module__ __qualname____doc____annotations__rpropertyrr$rrrrs VVVV0X)))))))rr_FINISHED_TRACKER) r, __future__rr# threadingr zmq.backendr zmq.errorrrr0__all__r/rrr6s.. #""""" ````````F#N$$ 0 1r