File: //opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyo
�
�A�[c @ s� d Z d d l m Z m Z m Z m Z m Z m Z d d l m Z m
Z
d d l m Z
m Z d d l m Z d d l m
Z
m Z m Z m
Z
m Z m Z d d l Z d � Z d d � Z d d
� Z d d � Z d � Z d
� Z d � Z d � Z d d d � Z e
j e
j e
j e
j! e
j" f Z# e
j$ e
j% f Z& d � Z' d d � Z( d � Z) d � Z* d � Z+ d � Z, d � Z- d � Z. d � Z/ d Z0 d Z1 d � Z2 d � Z3 d d � Z4 d S( s�
Utility functions from 2to3, 3to2 and python-modernize (and some home-grown
ones).
Licences:
2to3: PSF License v2
3to2: Apache Software License (from 3to2/setup.py)
python-modernize licence: BSD (from python-modernize/LICENSE)
i����( t
FromImportt Newlinet is_importt find_roott does_tree_importt Comma( t Leaft Node( t python_symbolst python_grammar( t token( R t Callt Namet symsR t NumberNc C s� d | k r | S| j d � r, | d } n g | D]$ } | j d j | � � r3 | ^ q3 } t | � d k r� t d d j d � | D� � � � n! t | � d k r� t d
� � n | d Sd S( sl
Examples:
>>> canonical_fix_name('fix_wrap_text_literals')
'libfuturize.fixes.fix_wrap_text_literals'
>>> canonical_fix_name('wrap_text_literals')
'libfuturize.fixes.fix_wrap_text_literals'
>>> canonical_fix_name('wrap_te')
ValueError("unknown fixer name")
>>> canonical_fix_name('wrap')
ValueError("ambiguous fixer name")
s .fix_t fix_i s fix_{0}i sO Ambiguous fixer name. Choose a fully qualified module name instead from these:
s
c s s | ] } d | Vq d S( s N( ( t .0t myf( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pys <genexpr>- s i s1 Unknown fixer. Use --list-fixes or -l for a list.N( t
startswitht endswitht formatt lent
ValueErrort join( t fixt avail_fixest ft found( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt canonical_fix_name s
$ c C s t t j d d | �S( Nu *t prefix( R R
t STAR( R ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt Star6 s c C s t t j d d | �S( Nu **R ( R R
t
DOUBLESTAR( R ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt
DoubleStar9 s c C s t t j d d | �S( Nu -R ( R R
t MINUS( R ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt Minus< s c C s? g } x+ | D]# } | j | � | j t � � q
W| d =| S( s{
Accepts/turns: (Name, Name, ..., Name, Name)
Returns/into: (Name, Comma, Name, Comma, ..., Name, Comma, Name)
i����( t appendR ( t leafst new_leafst leaf( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt commatize? s
c C s� x1 | j d k r3 | j j t j k r3 | j } q W| j d k rG d S| j t j k r` | j S| j d k r� | j j t j k r� | j j S| j d k r� d S| j Sd S( sf
Returns the indentation for this node
Iff a node is in a suite, then it has indentation.
u N(
t parentt Nonet typeR
t suiteR
t INDENTt valuet prev_siblingR ( t node( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt indentationK s '
$
c C s@ t | � } t d � | j � D� � } | s2 d St | � Sd S( s
Dirty little trick to get the difference between each indentation level
Implemented by finding the shortest indentation string
(technically, the "least" of all of the indentation strings, but
tabs and spaces mixed won't get this far, so those are synonymous.)
c s s* | ] } | j t j k r | j Vq d S( N( R+ R
R- R. ( R t i( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pys <genexpr>i s u N( R t sett pre_ordert min( R0 t rt all_indents( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt indentation_step` s
c C s� x' | j D] } | j t j k r
d Sq
Wx? t | j � D]" \ } } | j t j k r: Pq: q: Wt d � � t t j t � t
t j t | � t
| � � g � } | j | d } | j � d | _ | j | � | j | � d S( sj
Turn the stuff after the first colon in parent's children
into a suite, if it wasn't already
Nu No class suite and no ':'!i u ( t childrenR+ R
R, t enumerateR
t COLONR R R R R- R1 R8 t removeR t append_child( R) R0 R2 R, t one_node( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt suitifyp s 7
c C sw | d k r d } n t d d | �| g } | d k rg | j t d d d �t | d d �g � n t t j | � S( s�
Accepts a package (Name node), name to import it as (string), and
optional prefix and returns a node:
import <package> [as <as_name>]
u u importR u asu N( R* R t extendR R
t import_name( t packaget as_nameR R9 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt
NameImport� s c c sZ | j } xC | j t j k rN | j } | j t j k r= Pn | V| j } q W| j } | j } x | d k r� | V| j } qd W| j } | j t k r� | } x$ | j d k r� | j V| j } q� W| j } n | j } | d k r� d Sxg | j t k rU| j t j k r| Vn | j } | d k r� | j } | j } | d k rRPqRq� q� Wd S( s�
Generator yields all nodes for which a node (an import_stmt) has scope
The purpose of this is for a call to _find() on each of them
N(
t next_siblingR+ R
t SEMIt NEWLINER) R* t _compound_stmtsR
R, ( R0 t testt nxtR) t contextt ct p( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt import_binding_scope� s>
c C sg t | � } t d d d �} t | d d �} t t j | | | g � } | d k rc | | _ n | S( Nu asR u ( R R R
t import_as_nameR* R ( t nameRC R t new_namet new_ast new_as_namet new_node( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt ImportAsName� s c C s> | j t j k o= t | j � d k o= | j d j t j k S( s<
Returns True if the node appears to be a docstring
i ( R+ R
t simple_stmtR R9 R
t STRING( R0 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt is_docstring� s c C s2 t | � } t d | | � r" d Sd } xv t | j � D]e \ } } t | � s\ t | � re | } n t | � rw q8 n t | � } | s� Pn | | k r8 d Sq8 Wt d t
t j | d d �g � } | d k r| d k r| j d j
| _
d | j d _
n | t � g } | j | t t j | � � d S( s
This seems to work
u
__future__NR t i u ( R R R* R: R9 t is_shebang_commentt is_encoding_commentRX t check_future_importR R R
t NAMER R t insert_childR R
RV ( t featureR0 t roott shebang_encoding_idxt idxt namest import_R9 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt
future_import� s( $c C s6 t | � } t d | | � r" d Sd } x_ t | j � D]N \ } } | j t j k r8 | j r8 | j d j t j k r8 | d } Pq8 q8 WxP | j | D]; } | j t j k r� | d 7} q� n | j
} d | _
Pq� Wd } t d t t j
| d d �g � } | t � g } | j | t t j | d | �� d S( sD
An alternative to future_import() which might not work ...
u
__future__Ni i u R u ( R R R: R9 R+ R
RV R
RW RG R R R R] R R^ R ( R_ R0 R` t
insert_posRb t thing_afterR Rd R9 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt future_import2 s(
$c C s� g | D] } | j t j k r | ^ q } t g | D] } | d f ^ q5 � } x} t | � D]o \ } } | j t j k r� | j d j t j k r� | j d j
} | j d | | <q] | | } | | | <q] W| S( u/
Parse a list of arguments into a dict
i i i N( R+ R
t COMMAt dictR* R: R
t argumentR9 t EQUALR. ( t arglistt schemeR2 t kt ret_mappingt argt slot( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt
parse_args. s +%+
c C s, | j t j k o+ | j o+ t | j d � S( Ni ( R+ R
RV R9 R ( R0 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt is_import_stmtH s c C s� t | � } t | | | � r" d St } x6 d d d d g D]" } t d | | � r; t } Pq; q; W| r� d \ } } xo t | j � D]^ \ } } t | � r� | } | } x3 | r� | j } | d 7} t | � s� | } Pq� q� WPq� q� W| }
nL xC t | j � D]2 \ } } | j t
j k r Pn t | � s� Pq� q� W| }
| d k rt
t
j t t j d � t t j | d d
�g � } n� t | t t j | d d
�g � } | d k rKt
t
j t
t
j t t j d � t
t
j t t j d � t t j d
� g � t
t
j t t j d � t t j d � g � g � g � } | t � g }
n g }
| t � g } | j |
j } d | j |
_ | j |
t
t
j | d | �� t |
� d k r�| j |
d t
t
j |
� � n d S( s� Works like `does_tree_import` but adds an import statement at the
top if it was not imported (but below any __future__ imports) and below any
comments such as shebang lines).
Based on lib2to3.fixer_util.touch_import()
Calling this multiple times adds the imports in reverse order.
Also adds "standard_library.install_aliases()" after "from future import
standard_library". This should probably be factored into another function.
Nt absolute_importt divisiont print_functiont unicode_literalst
__future__i u importR u u standard_libraryu .u install_aliasesu (u )u i ( NN( R R t Falset TrueR* R: R9 R\ RE R+ R
RV RX R RA R R
R] R t powert trailert DOTt LPARt RPARR R R^ R ( RB t name_to_importR0 R` R RP t startt endRb t idx2Rf Rd t
install_hookst children_hookst children_importt
old_prefix( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt touch_import_topM sd
!$ $"c C s� | } | j t j k o | j s( t � S| j d } | j t j k op t | j d d � op | j d j d k sz t � S| j d j t j k r� | j d } n
| j d } | j t j
k r:t � } xh | j D]] } | j t j k r | j | j � q� | j t j
k r� | j d } | j | j � q� q� W| S| j t j
k ri| j d } t | j g � S| j t j k r�t | j g � Sd S( sZ If this is a future import, return set of symbols that are imported,
else return None.i i R. u
__future__i i N( R+ R
RV R9 R3 t import_fromt hasattrR. R
R t import_as_namesR] t addRO ( R0 t savenodet resultt n( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyR\ � s4
s ^#!.*pythons ^#.*coding[:=]\s*([-\w.]+)c C s t t j t | j � � S( s�
Comments are prefixes for Leaf nodes. Returns whether the given node has a
prefix that looks like a shebang line or an encoding line:
#!/usr/bin/env python
#!/usr/bin/python3
( t boolt ret matcht
SHEBANG_REGEXR ( R0 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyRZ � s c C s t t j t | j � � S( s
Comments are prefixes for Leaf nodes. Returns whether the given node has a
prefix that looks like an encoding line:
# coding: utf-8
# encoding: utf-8
# -*- coding: <encoding name> -*-
# vim: set fileencoding=<encoding name> :
( R� R� R� t ENCODING_REGEXR ( R0 ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyR[ � s
c C sR t | � d k r3 | \ } } | t � | g } n | } t t | � | d | �S( s�
Example:
>>> wrap_in_fn_call("oldstr", (arg,))
oldstr(arg)
>>> wrap_in_fn_call("olddiv", (arg1, arg2))
olddiv(arg1, arg2)
>>> wrap_in_fn_call("olddiv", [arg1, comma, arg2, comma, arg3])
olddiv(arg1, arg2, arg3)
i R ( R R R R ( t fn_namet argsR t expr1t expr2t newargs( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt wrap_in_fn_call� s
(5 t __doc__t lib2to3.fixer_utilR R R R R R t lib2to3.pytreeR R t lib2to3.pygramR R
R R
R R R R� R R* R R! R# R( R1 R8 R? RD t if_stmtt
while_stmtt for_stmtt try_stmtt with_stmtRH RA R� t
_import_stmtsRN RU RX Re Rh Rs Rt R� R\ R� R� RZ R[ R� ( ( ( sG /opt/alt/python27/lib/python2.7/site-packages/libfuturize/fixer_util.pyt <module> s> .. " $ ;
( c *