U ){f>@sdZddlZddlmZmZmZGdddeZGdddeZGdd d eZ Gd d d eZ Gd d d e eZ GdddeZ GdddeZGdddeZGdddZeeedddZGdddZdS)zR priority/tree ~~~~~~~~~~~~~ Implementation of the Priority tree data structure. N)ListTupleOptionalc@seZdZdZdS) PriorityErrorz9 The base class for all ``priority`` exceptions. N__name__ __module__ __qualname____doc__r r P/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/priority/priority.pyrsrc@seZdZdZdS) DeadlockErrorz_ Raised when there are no streams that can make progress: all streams are blocked. Nrr r r r r sr c@seZdZdZdS) PriorityLoopzM An unexpected priority loop has been detected. The tree is invalid. Nrr r r r rsrc@seZdZdZdS)DuplicateStreamErrorzE An attempt was made to insert a stream that already exists. Nrr r r r r%src@seZdZdZdS)MissingStreamErrorzQ An operation was attempted on a stream that is not present in the tree. Nrr r r r r-src@seZdZdZdS)TooManyStreamsErrorz An attempt was made to insert a dangerous number of streams into the priority tree at the same time. .. versionadded:: 1.2.0 Nrr r r r r5src@seZdZdZdS)BadWeightErrorze An attempt was made to create a stream with an invalid weight. .. versionadded:: 1.3.0 Nrr r r r r@src@seZdZdZdS)PseudoStreamErrorzN An operation was attempted on stream 0. .. versionadded:: 1.3.0 Nrr r r r rJsrc@seZdZdZd&eeddddZeeddd Zejedd d d Zddd d dZ ddd ddZ d'de ddddZ edddZ edddZee dddZee dddZde dddZde dd d!Zde dd"d#Zde dd$d%ZdS)(Streamz Priority information for a given stream. :param stream_id: The stream ID for the new stream. :param weight: (optional) The stream weight. Defaults to 16. N) stream_idweightreturncCs4||_||_g|_d|_g|_d|_d|_d|_dS)NTr)rrchildrenparent child_queueactive last_weight_deficit)selfrrr r r __init__\szStream.__init__rcCs|jSN)_weightrr r r rfsz Stream.weight)valuercCs<t|tstdnd|kr(dks2ntd||_dS)Nz"Stream weight should be an integerz3Stream weight must be between 1 and 256 (inclusive)) isinstanceintrr#)rr%r r r rjs   )childrcCs*||_|j|t|j|j|fdS)z} Add a stream that depends on this one. :param child: A ``Stream`` object that depends on this one. N)rrappendheapqheappushrr)rr*r r r add_childts zStream.add_childcCs:|j}g|_g|_d|_|||D]}||q&dS)z Add a stream that exclusively depends on this one. :param child: A ``Stream`` object that exclusively depends on this one. rN)rrrr.)rr*Z old_childrenZ old_childr r r add_child_exclusive~s zStream.add_child_exclusiveT)r*strip_childrenrcCsf|j|g}|jrBt|j\}}||kr0qt|||fq||_|rb|jD]}||qRdS)a" Removes a child stream from this stream. This is a potentially somewhat expensive operation. :param child: The child stream to remove. :param strip_children: Whether children of the removed stream should become children of this stream. N)rremoverr,heappopr-r.)rr*r0Z new_queuelevelstream new_childr r r remove_childs  zStream.remove_childc Cs|jr td}g}z`|dkrpt|j}| ||\}}|jrH|j }qz | }Wqt k rlYqYqXqW5|D]F\}}||_|d|j|j7}d|j|j|_t|j||fqxX|S)z Returns the stream ID of the next child to schedule. Potentially recurses down the tree of priorities. Nr') rAssertionErrorrrrr,r-rr2r+rschedule IndexError)rZ next_streamZpopped_streamsr3r*valr r r r8s(     zStream.schedulecCsd|j|jfS)NzStreamrrr$r r r __repr__szStream.__repr__)otherrcCst|tsdS|j|jkS)NF)r(rrrr=r r r __eq__s z Stream.__eq__cCs || Sr")r?r>r r r __ne__sz Stream.__ne__cCst|tstS|j|jkSr"r(rNotImplementedrr>r r r __lt__s z Stream.__lt__cCst|tstS|j|jkSr"rAr>r r r __le__s z Stream.__le__cCst|tstS|j|jkSr"rAr>r r r __gt__s z Stream.__gt__cCst|tstS|j|jkSr"rAr>r r r __ge__s z Stream.__ge__)r)T)rrr r r)r propertyrsetterr.r/boolr6r8strr<objectr?r@rCrDrErFr r r r rTs,    #(r) new_parentcurrentrcCsL|}tdD],}|j}|j|jkr(dS|jdkr dSq td|jdS)zG Reports whether the new parent depends on the current stream. dTrFz Stream %d is in a priority loop.N)rangerrr)rLrMr_r r r _stream_cycles   rQc@seZdZdZd"eddddZeeddd Zeedd d d Zd#ee eee ddddZ d$ee eee ddddZ eddddZ eddddZeddddZddddZedddZedd d!ZdS)% PriorityTreea  A HTTP/2 Priority Tree. This tree stores HTTP/2 streams according to their HTTP/2 priorities. .. versionchanged:: 1.2.0 Added ``maximum_streams`` keyword argument. :param maximum_streams: The maximum number of streams that may be active in the priority tree at any one time. If this number is exceeded, the priority tree will raise a :class:`TooManyStreamsError ` and will refuse to insert the stream. This parameter exists to defend against the possibility of DoS attack by attempting to overfill the priority tree. If any endpoint is attempting to manage the priority of this many streams at once it is probably trying to screw with you, so it is sensible to simply refuse to play ball at that point. While we allow the user to configure this, we don't really *expect* them too, unless they want to be even more conservative than we are by default. :type maximum_streams: ``int`` N)maximum_streamsrcCsNtddd|_d|j_d|ji|_t|ts4td|dkrDtd||_dS)Nrr&r;Fzmaximum_streams must be an int.z+maximum_streams must be a positive integer.) r _root_streamr_streamsr(r) TypeError ValueError_maximum_streams)rrTr r r r -s  zPriorityTree.__init__)parent_stream_idrcCsDz |j|WStk r>|||||j|YSXdS)a* When inserting or reprioritizing a stream it is possible to make it dependent on a stream that is no longer in the tree. In this situation, rather than bail out, we should insert the parent stream into the tree with default priority and mark it as blocked. N)rVKeyError insert_streamblock)rrZr r r _get_or_insert_parent:s    z"PriorityTree._get_or_insert_parent) parent_streaminserted_streamrcCs||dS)z} Insert ``inserted_stream`` beneath ``parent_stream``, obeying the semantics of exclusive insertion. N)r/)rr_r`r r r _exclusive_insertHs zPriorityTree._exclusive_insertrF)r depends_onr exclusivercCs||jkrtd|t|jd|jkrtd|YnX|rn||krXtd|||}t||}n|jd}d}||_|r|j ||j ||jj |dd|r| |n | |dS)a  Update the priority status of a stream already in the tree. :param stream_id: The stream ID of the stream being updated. :param depends_on: (optional) The ID of the stream that the stream now depends on. If ``None``, will be moved to depend on stream 0. :param weight: (optional) The new weight to give the stream. Defaults to 16. :param exclusive: (optional) Whether this stream should now be an exclusive dependency of the new parent. rzCannot reprioritize stream 0Stream %d not in treez#Stream %d must not depend on itselfF)r0N) rrVr[rrr^rQrrr6r.r/)rrrbrrcZcurrent_streamrLcycler r r reprioritizes8     zPriorityTree.reprioritize)rrcCsV|dkrtdz|j|}Wn tk r@td|YnX|j}||dS)zu Removes a stream from the priority tree. :param stream_id: The ID of the stream to remove. rzCannot remove stream 0reN)rrVpopr[rrr6)rrr*rr r r remove_streamszPriorityTree.remove_streamcCsF|dkrtdzd|j|_Wn tk r@td|YnXdS)z Marks a given stream as blocked, with no data to send. :param stream_id: The ID of the stream to block. rzCannot block stream 0FreNrrVrr[rrrr r r r]s zPriorityTree.blockcCsF|dkrtdzd|j|_Wn tk r@td|YnXdS)z Marks a given stream as unblocked, with more data to send. :param stream_id: The ID of the stream to unblock. rzCannot unblock stream 0TreNrjrkr r r unblocks zPriorityTree.unblockr!cCs|Sr"r r$r r r __iter__szPriorityTree.__iter__cCs.z |jWStk r(tdYnXdS)Nz!No unblocked streams to schedule.)rUr8r9r r$r r r __next__s zPriorityTree.__next__cCs|Sr")rnr$r r r nextszPriorityTree.next)rS)NrF)NrF)rrr r r)r rr^rarrIr\rgrir]rlrmrnror r r r rRsB   / ?rR)r r,typingrrr Exceptionrr rrr[rrrrrrIrQrRr r r r s    +