B %d!@s^dZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZe rdddlmZddlmZe eeefZe edfZdZe eed ge edffdd d d Zeeed ddZe eeeeedddZededdddZeeeedddZeeedddZde e edfeee edfdddZdeeeedddZdS) zBModule with functions which are supposed to be as fast as possible)S_ISDIR) safe_decodedefenc)CallableListMutableSequenceSequenceTuple TYPE_CHECKINGUnionoverload)ReadableBuffer)GitCmdObjectDBN)tree_to_streamtree_entries_from_datatraverse_trees_recursivetraverse_tree_recursiver )entrieswritereturnc Cstd}d}x|D]\}}}d}x.tdD]"}t||d?|@|g|}q*W|d|krh|dd}t|tr~|t} n|} |d|d | d |fqWdS) zWrite the give list of entries into a stream using its write method :param entries: **sorted** list of tuples with (binsha, mode, name) :param write: write method which takes a data string0rN )ordrangebytes isinstancestrencoderjoin) rrord_zeroZbit_maskZbinshamodenameZmode_stri name_bytesr*\/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/git/objects/fun.pyr&s"    r)datarc Cstd}td}t|}d}g}x||krd}x,|||krZ|d>|||}|d7}q0W|d7}|}x||dkr|d7}qjW|||}t|} |d7}|||d} |d}|| || fq"W|S)zReads the binary representation of a tree and returns tuples of Tree items :param data: data block with tree data (as bytes) :return: list(tuple(binsha, mode, tree_relative_path), ...)r rrr)rlenrappend) r,r%Z space_ordZlen_datar(outr&nsr)r'shar*r*r+rFs*    r) tree_datar'is_dirstart_atrcCsy8||}|r6|d|kr6t|d|kr6d||<|SWntk rLYnXx@t|D]4\}}|rX|d|krXt|d|krXd||<|SqXWdS)zreturn data entry matching the given name and tree mode or None. Before the item is returned, the respective data item is set None in the tree_data list to mark it donerN)r IndexError enumerate)r4r'r5r6itemindexr*r*r+ _find_by_nameus  r<)r: path_prefixrcCsdS)Nr*)r:r=r*r*r+ _to_full_pathsr>cCsdS)Nr*)r:r=r*r*r+r>scCs"|s|S|d|d||dfS)z$Rebuild entry with given path prefixrrr7r*)r:r=r*r*r+r>sr.)odb tree_shasr=rc sBg}t|}x:|D]2}|dkr$g}ntt||}||qWg}xt|D]\}} xt| D]\} } | svqhddt|D} | | |<| \} }}t|}x8t|d||D]"}||}t ||||| | |<qW|r| t |dd| D|dn|t fdd| Dd| | <qhW| dd=qVW|S) a :return: list of list with entries according to the given binary tree-shas. The result is encoded in a list of n tuple|None per blob/commit, (n == len(tree_shas)), where * [0] == 20 byte sha * [1] == mode as int * [2] == path relative to working tree root The entry tuple is None if the respective blob/commit did not exist in the given tree. :param tree_shas: iterable of shas pointing to trees. All trees must be on the same level. A tree-sha may be None in which case None :param path_prefix: a prefix to be added to the returned paths on this level, set it '' for the first iteration :note: The ordering of the returned items will be partially lostNcSsg|]}dqS)Nr*).0_r*r*r+ sz,traverse_trees_recursive..rcSsg|]}|r|dpdqS)rNr*)rAeir*r*r+rCs/c3s|]}t|VqdS)N)r>)rAe)r=r*r+ sz+traverse_trees_recursive..) r/listrstreamreadr0r9rrr<extendrtuple)r?r@r=Z trees_datanttree_shar,r1tir4iir:rZ_shar&r'r5Ztior*)r=r+rs:    r)r?rNr=rcCsfg}t||}xJ|D]B\}}}t|rJ|t||||dq|||||fqW|S)a& :return: list of entries of the tree pointed to by the binary tree_sha. An entry has the following format: * [0] 20 byte sha * [1] mode as int * [2] path relative to the repository :param path_prefix: prefix to prepend to the front of all returned pathsrE)rrIrJrrKrr0)r?rNr=rr,r3r&r'r*r*r+rsr) __doc__statrZ git.compatrrtypingrrrrr r r r Z _typeshedr gitrr intr"ZEntryTupZEntryTupOrNone__all__rrboolr<r>rrr*r*r*r+s( (    ( /L