g0 \ddlZddlZddlZddlZddlmZGddeZGddeZy)N)loggerc*eZdZdZddZdZdZdZy) TimedLocka Enforce request rate limit if applicable. This uses the lock so there is synchronized access to the API. When N threads enter this method, the first will pass trough, since there there was no last request recorded. The last request time will be set. Then, the second thread will unlock, and see that the last request was X seconds ago. It will sleep (request_limit - X) seconds, and then continue. Then the third one will unblock, and so on. After all threads finish, the total time will at least be (N * request_limit) seconds. If some request takes longer than request_limit seconds, the next unblocked thread will wait less. ctj|_d|_||_t j |_y)z! Set up the lock rN) threadingLocklock last_used minimum_deltaqueueQueue)selfr s /opt/Tautulli/plexpy/lock.py__init__zTimedLock.__init__'s.NN$ *[[] cX|jjtj|jz }|j|z }|dk\r+t j d|tj||jjs} |jjd}t j d|tj||jj|jjs|yy#tj$rYwxYw)z3 Called when with lock: is invoked rzSleeping %s (interval)FzSleeping %s (queued)N) r acquiretimer r rdebugsleepr emptygetEmpty task_done)rdelta sleep_amountsecondss r __enter__zTimedLock.__enter__0s  dnn,))E1 1  LL1< @ JJ| $**""$ **../ 3W= 7# JJ "**""$ ;;  sADD)(D)cjtj|_|jjy)z5 Called when exiting the with block. N)rr r releasertypevalue tracebacks r__exit__zTimedLock.__exit__Ds! rcftjd||jj|y)z Asynchronously add time to the next request. Can be called outside of the lock context, but it is possible for the next lock holder to not check the queue until after something adds time to it. zAdding %s to queueN)rinfor put)rrs rsnoozezTimedLock.snoozeKs#  ('2 wrN)r)__name__ __module__ __qualname____doc__rrr%r)rrrrs ##(  rrceZdZdZdZdZy)FakeLockzA If no locking or request throttling is needed, use this cy)z% Do nothing on enter Nr.)rs rrzFakeLock.__enter__\ rcy)z$ Do nothing on exit Nr.r!s rr%zFakeLock.__exit__br2rN)r*r+r,r-rr%r.rrr0r0Ws  rr0)r rrplexpyrobjectrr0r.rrr6s/$  : : z v r