File manager

U

e5dg;�
@s�
dZd
dl
mZ
d
dlmZmZ
d
dlmZ
d
dl	m
Z

dd�Zdd	�Zd
d�Z
dd
�ZdWdd�
Zdd�Zdd�Zdd�Ze�e
�fdd�
ZdXdd�
Zdd�Zdd�ZdYdd �
Zd!d"�ZdZd#d$�
Zd[d%d&�
Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2d3d4d5d6d7d8d9d:h
Z d;d<�Z!d=a"d>a#d?a$d@a%dAdB�Z&dCdD�Z'dEdF�Z(dGdH�Z)dIdJ�Z*dKdL�Z+dMdN�Z,dOdP�Z-ej.ej/hZ0d\dQdR�
Z1ej/ej.ej2hZ3dSdT�Z4d]dUdV�
Z5dS)^z1Utility functions, node construction macros, etc.�
)
�token)�Leaf�Node)
�python_symbols)
�patcompcCstt
j|ttjd
�|
g�S)N�
=)r�symsZargumentrr�EQUAL)�keyword�value�r�*/usr/lib64/python3.8/lib2to3/fixer_util.py�
KeywordArgs

�rcCstt
jd
�S)N�
()rr�LPARrrrr
�LParens
rcCstt
jd
�S)N�
))rr�RPARrrrr
�RParens
rcCsHt|t
�s|g
}t|
t
�s&d
|
_|
g
}
ttj|ttjdd
d�g
|
�S)zBuild an assignment statement�
 r�
�prefix)	�
isinstance�listrrr�atomrrr	)�target�sourcerrr
�Assigns






�rNcCstt
j||
d
�S)zReturn a NAME leafr)rr�NAME)�namerrrr
�Name$sr cCs|tt
jt�|
g�gS)
zA node tuple for obj.attr)rr�trailer�Dot)�obj�attrrrr
�Attr(sr%cCstt
jd
�S)zA comma leaf�
,)rr�COMMArrrr
�Comma,sr(cCstt
jd
�S)zA period (.) leaf�
.)rr�DOTrrrr
r"0sr"cCs4tt
j|
��|��g�}|r0|�d
tt
j|��
|S)z-A parenthesised argument list, used by Call()r
)rrr!�clone�insert_child�arglist)�argsZlparenZrparen�noderrr
�ArgList4s


r0cCs&tt
j|t|
�
g�}|d
k	r"||_|S)zA function callN)rr�powerr0r)Z	func_namer.rr/rrr
�Call;s


r2cCstt
jd
�S)zA newline literal�

�rr�NEWLINErrrr
�NewlineBsr6cCstt
jd
�S)zA blank line�r4rrrr
�	BlankLineFsr8cCstt
j||
d
�S)Nr)rr�NUMBER)�
nrrrr
�NumberJs
r;c

Cs"tt
jttjd
�|ttjd�g�S)zA numeric or string subscript�
[�
])rrr!rr�LBRACE�RBRACE)
Z
index_noderrr
�	SubscriptMs


�r@cCstt
j||
d
�S)z
A string leafr)rr�STRING)�stringrrrr
�StringSsrCc	Cs�d
|_d|
_d|_t
tjd�}d|_t
tjd�}d|_||
||g}|rtd|_t
tjd�}d|_|�ttj||g��

ttj|ttj	|�g�}ttj
t
tjd�|t
tjd�g�S)zuA list comprehension of the form [xp for fp in it if test].

    If test is None, the "if test" part is omitted.
    r7r�for�in�ifr<r=)
rrrr�appendrrZcomp_ifZ	listmakerZcomp_forrr>r?)	Zxp�fp�itZtestZfor_leafZin_leafZ
inner_argsZif_leaf�innerrrr
�ListCompWs(


















��rKcCsV|
D]}|��
qt
tjd
�t
tj|dd�t
tjddd�ttj|
�g}ttj|�}|S)zO Return an import statement in the form:
        from package import name_leafs�fromrr�import)�removerrrrr�import_as_names�import_from)Zpackage_nameZ
name_leafsZleaf�children�imprrr
�
FromImportos





�
rSc	Cs�|
d
��}|j
tjkr"|��}nttj|��g
�}|
d}|rNdd�|D�
}ttjtt|d�
t|d�
�ttj|
d��||
d��g�g
|�}|j	|_	|S)	zfReturns an import statement and calls a method
    of the module:

    import module
    module.name()r#�afterc
Ssg|]}
|
���qSr)
r+)�.0r:rrr
�
<listcomp>�sz!ImportAndCall.<locals>.<listcomp>�r
ZlparZrpar)
r+�typerr-rr1r%r r!r)r/�results�namesr#Z
newarglistrT�newrrr
�
ImportAndCall�s*













�����
r\c

Cs�t|t
�r |jt�t�gkr d
St|t
�o�t|j�
dko�t|jdt�o�t|jdt
�o�t|jdt�o�|jdjdko�|jdjdkS)z(Does the node represent a tuple literal?T�rWr
�rr)rrrQrr�lenrr�
r/rrr
�is_tuple�s



������rac

CsXt|t
�oVt|j�
d
koVt|jdt�oVt|jdt�oV|jdjdkoV|jdjdkS)z'Does the node represent a list literal?r
rW���r<r=)rrr_rQrrr`rrr
�is_list�s

�����rcc

Cstt
jt�|t�g�S�
N)rrrrrr`rrr
�parenthesize�s
re�sortedr�set�any�all�tuple�sum�min�max�	enumerateccs$t||
�}|r |V
t||
�}q
d
S)al
Follow an attribute chain.

    If you have a chain of objects where a.foo -> b, b.foo-> c, etc,
    use this to iterate over all objects in the chain. Iteration is
    terminated by getattr(x, attr) is None.

    Args:
        obj: the starting object
        attr: the name of the chaining attribute

    Yields:
        Each successive object in the chain.
    N)
�getattr)r#r$�nextrrr
�
attr_chain�s



rqzefor_stmt< 'for' any 'in' node=any ':' any* >
        | comp_for< 'for' any 'in' node=any any* >
     z�
power<
    ( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' |
      'any' | 'all' | 'enumerate' | (any* trailer< '.' 'join' >) )
    trailer< '(' node=any ')' >
    any*
>
z`
power<
    ( 'sorted' | 'enumerate' )
    trailer< '(' arglist<node=any any*> ')' >
    any*
>
Fc
Cspts&t
�t�
at
�t�
at
�t�
ad
atttg}
t|
t|d��D]*\}}i}|�||�r@|d|kr@
d
Sq@dS)a
 Returns true if node is in an environment where all that is required
        of it is being iterable (ie, it doesn't matter if it returns a list
        or an iterator).
        See test_map_nochange in test_fixers.py for some examples and tests.
        T�parentr/F)	�
pats_builtrZcompile_pattern�p0�p1�p2�ziprq�match)r/Zpatterns�patternrrrYrrr
�in_special_context�s













rzc
Cs�|j}
|
d
k	r|
j
tjkrdS|j}|j
tjtjfkr:dS|j
tjkrX|j	d|krXdS|j
tj
ks�|j
tjkr�|
d
k	r�|
j
tjks�|j	d|kr�dSdS)zG
    Check that something isn't an attribute or function name etc.
    NFrWT)
Zprev_siblingrXrr*rrr�funcdef�classdef�	expr_stmtrQZ
parametersZ
typedargslistr')r/�prevrrrrr
�is_probably_builtin�s&







��
��
rc
CsJ|d
k	rF|jt
jkr>t|j�
dkr>|jd}
|
jtjkr>|
jS|j}qdS)zFind the indentation of *node*.Nr^r
r7)	rXr�suiter_rQr�INDENTrrr)r/�indentrrr
�find_indentation
s






r�c
Cs>|jt
jkr|S|��}|jd}
|_tt
j|g
�}|
|_|Srd)rXrr�r+rrr)r/rrr�rrr
�
make_suite
s






r�c

Cs$|jt
jkr |j}|std
�
�
q|S)zFind the top level namespace.z,root found before file_input node was found.)rXrZ
file_inputrr�
ValueErrorr`rrr
�	find_root&
s





r�cCst|
t
|�
|�}t|�
S)
z� Returns true if name is imported from package at the
        top level of the tree which node belongs to.
        To cover the case of an import like 'import foo', use
        None for the package and 'foo' for the name. )�find_bindingr��bool)�packagerr/Zbindingrrr
�does_tree_import/
s
r�c

Cs|jt
jt
jfkS)
z0Returns true if the node is an import statement.)rXr�import_namerPr`rrr
�	is_import7
sr�cCs.
d
d�}t|�
}t
||
|�r dSd}}t|j�
D]F\}}||�
sDq2t|j|d��
D]\}}||�
sV
qlqV||}
qzq2|dkr�t|j�
D]8\}}|jtjkr�|jr�|jdjtjkr�|d}
q�q�|dkr�t	tj
ttjd�ttj|
dd�g�}	nt
|ttj|
dd�g
�}	|	t�g}
|�|t	tj|
��
dS)	z\ Works like `does_tree_import` but adds an import statement
        if it was not imported. c

Ss |jt
jko|jot|jd
�
S)NrW)rXr�simple_stmtrQr�r`rrr
�is_import_stmt>
s

�z$touch_import.<locals>.is_import_stmtNrWr
rMrr)r�r�rnrQrXrr�rrArr�rrrSr6r,)r�rr/r��rootZ
insert_pos�offset�idxZnode2�import_rQrrr
�touch_import;
s8











�




�

r�cCs�
|
jD�
]�}d
}|j
tjkrVt||jd�r4|
St|t|jd�
|�}|rR|}�
n0|j
tjtjfkr�t|t|jd�
|�}|r�|}�n�|j
tj	k�
rt|t|jd�
|�}|r�|}nTt
|jdd
��
D]@\}}|j
tjkr�|j
dkr�t|t|j|d�
|�}|r�|}q�nx|j
tk�
r2|jdj
|k�
r2|}nTt|||��
rF|}n@|j
tjk�
rbt|||�}n$|j
tjk�
r�t||jd��
r�|}|r|�
s�|
St|�
r|
Sqd
S)	z� Returns the node which binds variable name, otherwise None.
        If optional argument package is supplied, only imports will
        be returned.
        See test cases for examples.Nr
rbr^r]�
:�rW)rQrXrZfor_stmt�_findr�r�Zif_stmtZ
while_stmtZtry_stmtrnr�COLONr�	_def_syms�_is_import_bindingr�r}r�)rr/r��childZretr:�
iZkidrrr
r�i
sH





























r�cCsT|
g
}|rP|��}
|
j
d
kr4|
j
tkr4|�|
j�

q|
j
tjkr|
j|kr|
SqdS)N�
)�poprX�_block_syms�extendrQrrr)rr/Znodesrrr
r��
s







r�cCs�
|jt
jkr�|s�|jd
}|jt
jkrx|jD]H}|jt
jkrV|jdj|
krt|
Sq,|jtjkr,|j|
kr,|
Sq,nL|jt
jkr�|jd}|jtjkr�|j|
kr�|Sn|jtjkr�|j|
kr�|Sn�|jt
j	k�
r�|r�t
|jd
�
��|kr�dS|jd}|�
rtd|��
rdS|jt
j
k�
r0t|
|��
r0|S|jt
jk�
rh|jd}|jtjk�
r�|j|
k�
r�|Sn6|jtjk�
r�|j|
k�
r�|S|�
r�|jtjk�
r�|SdS)z� Will reuturn node if node will import name, or node
        will import * from package.  None is returned otherwise.
        See test cases for examples. r
r^rbNr]�as)rXrr�rQZdotted_as_namesZdotted_as_namerrrrP�str�stripr�rOZimport_as_name�STAR)r/rr�rRr�Zlastr:rrr
r��
s@


































r�)
N)NN)
N)
N)
N)
N)
N)6�__doc__Zpgen2rZpytreerrZpygramrrr7rrrrrr r%r(r"r0r2r6r8r;r@rCrKrSr\rarcreZconsuming_callsrqrtrurvrsrzrr�r�r�r�r�r�r|r{r�r�r!r�r�r�rrrr
�<module>
s`








�
		-

*