>dZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z dZdZeefZd ZGd d eZGd d eZGddeZdfdZy)z=General implementation of walking commits and their contents.N)chain)RENAME_CHANGE_TYPES tree_changestree_changes_for_mergeRenameDetector)MissingCommitError)Tagdatetopoc$eZdZdZdZddZdZy) WalkEntryz1Object encapsulating a single result from a walk.c||_|j|_|j|_i|_|j |_yN)commitstore_store get_parents _get_parents_changesrename_detector_rename_detector)selfwalkerrs lib/third_party/dulwich/walk.py__init__zWalkEntry.__init__5s7 ll ".. & 6 6Nc$|jj|}||j}|j|s t}d}n&t |j|dk(rlt}|j |j|dj}|r|j|j j|\}}|j |}nt}|j|Dcgc]}|j |j}}|r`|Dcgc]}|j |} }g}| D]=} |j|j j|\}} |j| ?|j} |r8|j | } | j|j j|\}} t||j || |j}||j|<|j|Scc}wcc}w#t$rYwxYw)a-Get the tree changes for this entry. Args: path_prefix: Portion of the path in the repository to use to filter changes. Must be a directory name. Must be a full, valid, path reference (no partial names or wildcards). Returns: For commits with up to one parent, a list of TreeChange objects; if the commit has no parents, these will be relative to the empty tree. For merge commits, a list of lists of TreeChange objects; see dulwich.diff.tree_changes_for_merge. Nr)r)rgetrrrlenrtree lookup_path __getitem__rappendKeyErrorlistr) r path_prefixcachedr changes_funcparentmode subtree_shap parent_treesstcommit_tree_sha commit_trees rchangeszWalkEntry.changes<s"";/ >[[F$$V,+ T&&v./14+ T%6%6v%>q%ABGG(.(:(: //#)%D+"[[5F5 7;7H7H7PQ7P!$++a.--7PQ0)H HHcXd|jjd|jdS)Nz)ridr3rs r__repr__zWalkEntry.__repr__ys KKNN LLN  rr)__name__ __module__ __qualname____doc__rr3r8rrrr2s;7;*z rrc,eZdZdZdZdZdZdZeZy)_CommitTimeQueuez3Priority queue of WalkEntry objects by commit time.c||_|j|_|j|_|j |_g|_t|_ t|_ t|_ |j|_ d|_t|_d|_t%|j&|j D]}|j)|y)NF)_walkerrrrrexcluded _excluded_pqset_pq_set_seen_donesince _min_time_last_MAX_EXTRA_COMMITS_extra_commits_left _is_finishedrinclude_push)rr commit_ids rrz_CommitTimeQueue.__init__s ll "..u U U  #5 !v~~v?I JJy !@rc |j|}t|tr|j |j dy|}|j|jvr|j|jvrxtj|j|j |f|jj|j|jj|jyyy#t$r t|wxYwNr)rr&r isinstancer rPobjectr6rFrHheapqheappushrD commit_timeaddrG)r object_idobjrs rrPz_CommitTimeQueue._pushs 0++i(C c3  JJszz!} %  99DLL (VYYdjj-H NN488v'9'9&96%B C LL  VYY ' JJNN699 %.I ( 0$Y/ / 0s C,,Dc|j}|j}|g}|ra|j}|j|D]9}||vr"||vr|j |j ||j |;|r`yyr)rCrGpoprr%rrY)rrrBseentodor+s r_exclude_parentsz!_CommitTimeQueue._exclude_parentsst>>zzxXXZF++F3)fn KK F 34 V$4rcjryjrtjj\}}|j}j j ||jvrejj|j|D]}j|d}|jv}|rj|jrdtfdjDrFjd\}}jr&|jjjk\rd}nd}j |jj krd}|r t"_n*xj$dzc_j$s d_y|s|_t'j(|Sjrd_y)NTc3TK|]\}}|jjv!ywr)r6rC).0_crs r z(_CommitTimeQueue.next..s##Pxtq!ADDDNN$:xs%(rFr)rNrDrVheappopr6rFremoverHrYrrPrCr`allrKrXrJrLrMrrA)rrdrsha parent_idreset_extra_commits is_excludedns` rnextz_CommitTimeQueue.nexts   hh dhh/IAv))C LL   $djj  JJNN3 !..v6  9%7#' /K%%f-88#Ptxx#P P88A;DAqzzammtzz7M7M&M /3+.3+~~)f.@.@4>>.Q ',#"+=(((A-(// ! #  v66YhhZ!rN) r9r:r;r<rrPr`ro__next__r=rrr?r?s!="" & %1fHrr?c TeZdZdZdeddddddddef dZdZdZdZ d Z d Z d Z y) WalkerzObject for performing a walk of commits in a store. Walker objects are initialized with a store and other options and can then be treated as iterators of Commit objects. NFc|jSrparentsrs rzWalker.s6>>rc|tvrtd|z||_t|tr|g}||_t |xsg|_||_||_ ||_ |xr t |xsd|_ | r |s t|}||_ | |_| |_| |_| |_d|_| ||_t)j*|_y)aConstructor. Args: store: ObjectStore instance for looking up objects. include: Iterable of SHAs of commits to include along with their ancestors. exclude: Iterable of SHAs of commits to exclude along with their ancestors, overriding includes. order: ORDER_* constant specifying the order of results. Anything other than ORDER_DATE may result in O(n) memory usage. reverse: If True, reverse the order of output, requiring O(n) memory. max_entries: The maximum number of entries to yield, or None for no limit. paths: Iterable of file or subtree paths to show entries for. rename_detector: diff.RenameDetector object for detecting renames. follow: If True, follow path across renames/copies. Forces a default rename_detector. since: Timestamp to list commits after. until: Timestamp to list commits before. get_parents: Method to retrieve the parents of a commit queue_cls: A class to use for a queue of commits, supporting the iterator protocol. The constructor takes a single argument, the Walker. zUnknown walk order %sNr) ALL_ORDERS ValueErrorrrTbytesrOrErBorderreverse max_entriespathsrrrfollowrIuntil _num_entries_queue collectionsdeque _out_queue)rrrOexcluder|r}r~rrrrIrr queue_clss rrzWalker.__init__sX  "4u<= = gu %iG GMr*   &)s5z1T /,U3O.&   o %++-rc|y|jD].}||k(ry|j|s|t|dk(s.yy)NFT/)r startswithr!)r changed_path followed_paths r _path_matcheszWalker._path_matches4sL  !ZZM},'' 6 ]!34?(rcR|sy|jj}|jj}|j|rU|jrH|j t vr6|jj||jj|y|j|ryy)NFT) oldpathnewrrtyperrrYrh)rchangeold_pathnew_paths r_change_matcheszWalker._change_matchesAs|::??::??   h '{{v{{.AA x( !!(+    )rc|j}|j|j|jkry|j|j|jkDry|j|j vry|j yt|j|dkDr1|jD]}|D]}|j|syy|jD]}|j|syy)a Determine if a walk entry should be returned.. Args: entry: The WalkEntry to consider. Returns: True if the WalkEntry should be returned by this walk, or False otherwise (e.g. if it doesn't match any requested paths). NFTr) rrIrXrr6rBrr!rr3r)rentryr path_changesrs r_should_returnzWalker._should_returnPs :: !f&8&84::&E :: !f&8&84::&E 99 % ::  t' (1 , %  +F++F3#+ !0 --/''/*rc|j}||j|krt|j}||jj ||t |jtkDrO|jsy|jj}|j|r|xjdz c_|S||j|kryrS) r~rrorrr%r!rLpopleftr)rr~rs r_nextz Walker._nextqs&& !T%6%6%D%E &&u-}DOO 47I I//1&&u-%%*% L!T%6%6%Drc|jtk(rt||j}|jrt t |}|S)aPossibly reorder a results iterator. Args: results: An iterator of WalkEntry objects, in the order returned from the queue_cls. Returns: An iterator or list of WalkEntry objects, in the order required by the Walker. )r| ORDER_TOPO _topo_reorderrr}reversedr')rresultss r_reorderzWalker._reorders< :: ##GT-=-=>G <<tG}-Grc^t|jt|jdSr)iterrrr7s r__iter__zWalker.__iter__s!DMM$tzz4"89::r) r9r:r;r< ORDER_DATEr?rrrrrrrr=rrrrrrsP1"C.J  B  ;rrrc|jSrrtrvs rrwrwsfnnrc#Ktj}i}tjt}|D]7}|j |||j D]}||xxdz cc<9|r|j }|j }|j}||r|||<5||D];} || xxdzcc<|| r|j| d} | s+|j| =||ryyw)aReorder an iterable of entries topologically. This works best assuming the entries are already in almost-topological order, e.g. in commit time order. Args: entries: An iterable of WalkEntry objects. get_parents: Optional function for getting the parents of a commit. Returns: iterator over WalkEntry objects from entries in FIFO order, except where a parent would be yielded before any of its children. rN) rr defaultdictintr%rrr6r] appendleft) entriesrr_pending num_childrenrr.rrQrk parent_entrys rrrs    DG**3/L EU\\*A Oq O+  II  "!&GI  $V,I  #q ( # *&{{9d; OOL1 -   sB>C2C2C20C2)r<rrV itertoolsrdulwich.diff_treerrrrdulwich.errorsrdulwich.objectsr rrryrLrUrr?rrrr=rrrs*D    * % K K \evePh;Vh;V(E!r