gJdZddlmZddlmZddlmZGddeZdS)a Notification Management Tools Primarily this module houses the `NotificationManager` class which provides an iterator for a NOTIFY event loop against a set of connections. >>> import postgresql >>> db = postgresql.open(...) >>> from postgresql.notifyman import NotificationManager >>> nm = NotificationManager(db, timeout = 10) # idle events every 10 seconds >>> for x in nm: ... if x is None: ... # idle event ... ... ... db, notifies = x ... for channel, payload, pid in notifies: ... ... )time)select)chaincleZdZdZdZdddZeefdZdZ dZ d Z d Z efd Z d Zd ZdZefdZdS)NotificationManagera_ A class for managing the asynchronous notifications received by a set of connections. Instances provide the iterator for an event loop that responds to NOTIFYs received by the connections being watched. There is no thread safety, so when a connection is being managed, it should not be used concurrently in other threads while being managed. ) connectionsgarbageincomingtimeout _last_time_pulledNr c||t||_t|_d|_d|_t |_dSN) settimeoutsetrr r r dictr )selfr rs J/granary/edev/quantum/lib/python3.11/site-packages/postgresql/notifyman.py__init__zNotificationManager.__init__-sL//'%%$$,$-$/$,,,c4|jdkrd}nD|j%|jpd||jz z }|dkrd}n||_|jpd}|j|jz }|D]=}|jr4|j||j|>|j|jz }||d||\}}}|D]=} |d#t$r| |Y:wxYw| |dS)Nr ) r r rr closedremoveaddexecute Exceptionappendtrash) rrr max_durationcheckdbrwxs r_wait_on_wiresz"NotificationManager._wait_on_wires9sV \Q<< o!L&B4466DO+CDLa\dffDO<%2L  T\ )% biBLR  T\ )% F5"e\ 2 2'!Q  bJJrNNNN HHRLLLLL**Q-----sCD?Dcn|j||j|dS)z Remove the given connections from the set of good connections, and add them to the `garbage` set. This method can be overridden by subclasses to take a callback approach to connection failures. N)r updaterdifference_update)rrs rr"zNotificationManager.trash_s7,k"""$$[11111rc~|j|t}||dS)z Queue the notifies for the specified connection. Upon success, the This method can be overridden by subclasses to take a callback approach to notification management. N)r setdefaultlistextend)rr%notifiesls rqueuezNotificationManager.queueks7 lb$&&))!((8rc|jD]V}|js |jjfd|jD}||||jdt |=WdS)Nc\g|](}|j|j|jf)Sr)channelpayloadpid).0r(decodes r z>NotificationManager._pull_from_connections..}sJ  VAIqy))151r)r _notifiestypior:r3len)rr%r1r:s @r_pull_from_connectionsz*NotificationManager._pull_from_connectionsvs   $ $b ,  HO6 \8::b( |NS]]N## $ $rctg}|jD]0}|jj}|||j|f1|rV|jr1t |jt |ic|_|_dSt |ic|_|_dS|jt dic|_|_dSdS)Nr)r r=r:r!r riter)rnew_seqsr%r:s r _queue_nextzNotificationManager._queue_nexts ( L++b HO6 ??B R()**** . m5#( tH~~"F"FDM4<<<"&x.."DM4<<< }!%b24=$,,,rc|}|j||_n*|jr||j|jzkr d|_dSd|_dS)NTF)r r )rrnows r _timedoutzNotificationManager._timedoutsY # _4?? |$, >??4? $4? rcB||dkrtd||_dS)a Set the maximum duration, in seconds, for waiting for NOTIFYs on the set of managed connections. The given `seconds` argument can be a number or `None`. A timeout of `None` means no timeout, and "idle" events will never occur. A timeout of `0` means to never wait for NOTIFYs. This has the effect of a StopIteration being raised by `__next__` when there are no more Notifications available for any of the connections in the set. "Idle" events will never occur in this situation as well. A timeout greater than zero means to emit `None` as "idle" events into the loop at the specified interval. Idle events are guaranteed to occur. Nrz!cannot set timeout less than zero) ValueErrorr )rsecondss rrzNotificationManager.settimeouts," Wq[[ 7 8 88$,,,rc|jS)zGet the timeout.rrs r gettimeoutzNotificationManager.gettimeouts rc|SrrrKs r__iter__zNotificationManager.__iter__s +rcVd} |j? t|jS#t$rd|_|jdks|jsYnwxYw|rdS|s|jr|d}nd}||)NTrF) r next StopIterationr rrFr)r?rC)rr checked_wires r__next__zNotificationManager.__next__s, m        T] !1    nn 4 4+LLL   7s $AA)__name__ __module__ __qualname____doc__ __slots__rrrr)r"r3r?rCrFrrLrNrSrrrrrs-1     "&$$$$L 2 2 2 $ $ $..."!"* rrN)rWrr itertoolsrobjectrrrrr[s$GGGGG&GGGGGr