B .zf@sUddlZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z e rdd l mZdd l mZe ejejejfZejfZe e ed fed <ejfZe e ed fed<ydd l mZdd l mZWnek r dZYn&Xeejejf7Zeejejf7ZdZeedddZdedddZd(dee ed fdddZd)deedddZd*e ddefe eefeeeddd Zeedd!d"d#Zd+eeeed$d%d&Ze d'kre!edS),N)Any) Generator)Optional)Sequence)Tuple)Type) TYPE_CHECKING)Union)ast27)ast3.AST expr_contextFT)nodereturncCst|tot|t S)N) isinstancer r )rrL/home/ankuromar296_gmail_com/.local/lib/python3.7/site-packages/astpretty.py _is_sub_node srASTTypecCsVxP|jD]B}t||}t|r"dSt|ttfrx|D]}t|r6dSq6WqWdSdS)NFT)_fieldsgetattrrrlisttuple)rfieldattrvalrrr_is_leaf$s    r)n show_offsetsrcCs|r|j|jS|jSdS)N) _attributesr)rrrrrr1s r)rrrc snttr:dtjdfddtdDSttrbddfddDStSdS)Nz{}({})z, c3s(|] }d|tt|dVqdS)z{}={})rN)format_leafr).0r)rrrr =sz_leaf..)rz[{}]c3s|]}t|dVqdS))rN)r!)r"x)rrrr#Fs) rr r type__name__joinrrrepr)rrr)rrrr!8s   r! )rindentr_indentrc s|dkrt|St|tr"t|St|r6t|dSttrJdnGfdddtjtddfdd }tdfd d }dt d dtfttd fdd }t |j d}| xt |dD]}t ||} | gkrd} nt| trJt| dkrJt| d trJt| d rJd|| d d} n~t| trd} |0x(| D] } | d||| j7} qhWWdQRX| |d7} n"t| tr|| j} nt| } |||d| d7}qWWdQRX||d7}|SdS)N)r cseZdZZdS)zpformat..stateN)r& __module__ __qualname__r*r)r+rrstate^sr/)NNN)rc3s&jd7_dVjd8_dS)N)r*r)r/rrindentedaszpformat..indentedcs jS)N)r*r)indent_sr/rr indentstrgszpformat..indentstrrr)elr+rcst||dS)N)r*rr+)pformat)r4r+)r*rrr_pformatjszpformat.._pformatz( z[]r0[]z[ z{}{}, =z, ))r)r(rstrrr!int contextlibcontextmanagerrr r%r&rrrlenr r r*) rr*rr+r1r3r6outrrZrepresentationr4r)r+r*r2rr/rr5LsL    $      $ (r5)argskwargsrcOstt||dS)N)printr5)rArBrrrpprintsrD)argvrc Cst}|d|jdddd|}|jdtjddtjd d trt|jd ddtjd d |jdddtjdd | |}|j tjkot j dk}|rddi}ni}t |jd}|}WdQRXt|j |f||jddS)Nfilenamez--no-show-offsetsr store_false)destactionz --untyped parse_func store_constz&(default) Use the stdlib `ast` parser.)defaultrHrIconsthelpz --typed-27z%Use typed_ast.ast27 to parse the ast.)rHrIrMrNz --typed-3z$Use typed_ast.ast3 to parse the ast.) type_commentsTrb)rr)argparseArgumentParser add_argumentadd_mutually_exclusive_groupastparse typed_supportr r parse_argsrJsys version_infoopenrFreadrDr)rEparsergrprArQrBfcontentsrrrmains8   rc__main__)T)T)r)Tr)N)"rSrWr=r[typingrrrrrrrr Z typed_astr r r r__annotations__r ImportErrorrYboolrrr;rr!r<r5rDrcr&exitrrrrsL             =&