3
+��hB: � + @ s� d Z ddlZejdFkr ddlZnddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
yddlmZmZ W n ek
r� dZeZY nX ejdd � dGk r�ddlZeje�Zejej� ee
j�ee
j� ee
j� ee
j � Z d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-g#Z G d.d/� d/�Z d0d1� Z d2d3� Z!d4d5� Z"d6d7� Z#G d8d9� d9e�Z$G d:d;� d;e�Z%e%� Z&f f fd<d=�Z'd>d?� Z(d@dA� Z)G dBdC� dCe�Z*G dDdE� dEe�Z+e'Z,dS )Ha'
past.translation
==================
The ``past.translation`` package provides an import hook for Python 3 which
transparently runs ``futurize`` fixers over Python 2 code on import to convert
print statements into functions, etc.
It is intended to assist users in migrating to Python 3.x even if some
dependencies still only support Python 2.x.
Usage
-----
Once your Py2 package is installed in the usual module search path, the import
hook is invoked as follows:
>>> from past.translation import autotranslate
>>> autotranslate('mypackagename')
Or:
>>> autotranslate(['mypackage1', 'mypackage2'])
You can unregister the hook using::
>>> from past.translation import remove_hooks
>>> remove_hooks()
Author: Ed Schofield.
Inspired by and based on ``uprefix`` by Vinay M. Sajip.
� N� � )�
ParseError)�RefactoringTool)�fixes)�
PathFinder�SourceFileLoader� � zlib2to3.fixes.fix_applyzlib2to3.fixes.fix_exceptzlib2to3.fixes.fix_execfilezlib2to3.fixes.fix_exitfunczlib2to3.fixes.fix_funcattrszlib2to3.fixes.fix_filterzlib2to3.fixes.fix_has_keyzlib2to3.fixes.fix_idiomszlib2to3.fixes.fix_importzlib2to3.fixes.fix_internz lib2to3.fixes.fix_isinstancez lib2to3.fixes.fix_methodattrszlib2to3.fixes.fix_nez lib2to3.fixes.fix_numliteralszlib2to3.fixes.fix_parenzlib2to3.fixes.fix_printzlib2to3.fixes.fix_raisezlib2to3.fixes.fix_renameszlib2to3.fixes.fix_reducezlib2to3.fixes.fix_reprz lib2to3.fixes.fix_standarderrorzlib2to3.fixes.fix_sys_exczlib2to3.fixes.fix_throwz lib2to3.fixes.fix_tuple_paramszlib2to3.fixes.fix_typeszlib2to3.fixes.fix_ws_commaz lib2to3.fixes.fix_xreadlinesz lib2to3.fixes.fix_basestringzlib2to3.fixes.fix_execzlib2to3.fixes.fix_getcwduzlib2to3.fixes.fix_longzlib2to3.fixes.fix_nextzlib2to3.fixes.fix_nonzerozlib2to3.fixes.fix_raw_inputzlib2to3.fixes.fix_xrangec @ s8 e Zd ZdZdZdZdZdZedd� �Z edd� �Z
dS )�RTsa&
A namespace for the refactoring tools. This avoids creating these at
the module level, which slows down the module import. (See issue #117).
There are two possible grammars: with or without the print statement.
Hence we have two possible refactoring tool implementations.
Nc C s. dt jt jgkr*tt�t _ttddi�t _dS )zj
Call this before using the refactoring tools to create them on demand
if needed.
N�print_functionT)r �_rt�_rtpr �myfixes� r r �@/root/tmp/pip-build-yeh9mt8d/future/past/translation/__init__.py�setup� s
z RTs.setupc C s. dt jt jgkr*tt�t _ttddi�t _dS )zj
Call this before using the refactoring tools to create them on demand
if needed.
Nr T)r �_rt_py2_detect�_rtp_py2_detectr �py2_detect_fixersr r r r �setup_detect_python2� s
zRTs.setup_detect_python2) �__name__�
__module__� __qualname__�__doc__r
r r r � staticmethodr r r r r r r � s r c C sr g }xht jj| �}|d | kr2|jd|d � P q|d | krR|jd|d � P q|d } |jd|d � qW |S )zA
Split a path into all components. From Python Cookbook.
r � )�os�path�split�insert)r Zallparts�partsr r r �splitall� s r" c C sL g }t | �}t |�}x*t||�D ] \}}||kr2P |j|� q W tjj|� S )z^
Returns the longest common substring to the two strings, starting from the
left.
)r" �zip�appendr r �join)�s1�s2�chunks�path1�path2Zdir1Zdir2r r r �common_substring� s r+ c C s� t j� yt jj| |�}W nF tk
r` } z*|jdks@|jdkrB� t jj| |�}W Y dd}~X nX | t|�dd � kr�t j
dj |�� dS t j
dj |�� dS dS )
zD
Returns a bool indicating whether we think the code is Py2
z bad input�=Nr zDetected Python 2 code: {0}TzDetected Python 3 code: {0}F���) r r r �refactor_stringr �msg�valuer �str�logger�debug�format)�source�pathname�tree�er r r �detect_python2� s r9 c C sz t j� | d7 } yt jj| |�}W nF tk
rh } z*|jdksH|jdkrJ� t jj| |�}W Y d d }~X nX t|�d d� S )N�
z bad inputr, r r- ) r r r
r. r r/ r0 r r1 )r5 r6 r7 r8 r r r � transform� s r; c s@ e Zd Zg Zg Zdd� Zdd� Z� fdd�Z� fdd�Z� Z S ) �PastSourceFileLoaderc sJ | j � t� fdd�| jD ��r$d}n"t� fdd�| jD ��rBd}nd}|S )Nc 3 s | ]}� j |�V qd S )N)�
startswith)�.0r )�fullnamer r � <genexpr> s z7PastSourceFileLoader._convert_needed.<locals>.<genexpr>Fc 3 s | ]}� j |�V qd S )N)r= )r> r )r? r r r@
s T)�name�any�
exclude_paths�
include_paths)�self�convertr )r? r �_convert_needed s z$PastSourceFileLoader._convert_neededc C sB | j | j�}| j}t||�r&t||�}t||d�}t||j� d S )N�exec)�
get_sourcerA r r9 r; �compilerH �__dict__)rE �moduler5 r6 �coder r r �_exec_transformed_module s
z-PastSourceFileLoader._exec_transformed_modulec s� t jd|� |tjkr"tj| }nv| j� r�t jd|� tj|�}|tj|< | j|_| |_ | j
|�rpg |_ ||_ n|j
d�d |_ | j|� n t� j|�}|S )NzRunning load_module for %szAutoconverting %s�.r )r2 r3 �sys�modulesrG �imp�
new_moduler �__file__�
__loader__�
is_package�__path__� __package__�
rpartitionrN �super� load_module)rE r? �mod)� __class__r r r[ s
z PastSourceFileLoader.load_modulec s> t jd|� | j� r.t jd| j� | j|� n t� j|� d S )NzRunning exec_module for %szAutoconverting %s)r2 r3 rG rA rN rZ � exec_module)rE rL )r] r r r^ 4 s
z PastSourceFileLoader.exec_module)
r r r rC rD rG rN r[ r^ �
__classcell__r r )r] r r< s
r< c @ s@ e Zd ZdZdZdd� Zdd� Zdd� Zdd
d �Zdd d
�Z d S )�Py2Fixerzi
An import hook class that uses lib2to3 for source-to-source translation of
Py2 code to Py3.
Tc C s( d | _ ddg| _tj| j�| _g | _d S )N�futureZpast)�foundZbase_exclude_paths�copyrC rD )rE r r r �__init__H s
zPy2Fixer.__init__c C s | j |7 _ dS )z�
Pass in a sequence of module names such as 'plotrique.plotting' that,
if present at the leftmost side of the full package name, would
specify the module to be transformed from Py2 to Py3.
N)rD )rE �pathsr r r �includeN s zPy2Fixer.includec C s | j |7 _ dS )z�
Pass in a sequence of strings such as 'mymodule' that, if
present at the leftmost side of the full package name, would cause
the module not to undergo any source transformation.
N)rC )rE re r r r �excludeV s zPy2Fixer.excludeNc C sH t jd||� tj||�}|s.t jd|� d S t|_| j|_| j|_|S )Nz Running find_module: (%s, %s)zPy2Fixer could not find %s)r2 r3 r � find_moduler< r] rC rD )rE r? r �loaderr r r rh _ s zPy2Fixer.find_modulec C sR t jd|||� tj|||�}|s2t jd|� d S t|j_| j|j_| j|j_|S )Nz Running find_spec: (%s, %s, %s)zPy2Fixer could not find %s) r2 r3 r � find_specr< ri r] rC rD )rE r? r �target�specr r r rj k s
zPy2Fixer.find_spec)N)NN)
r r r r ZPY2FIXERrd rf rg rh rj r r r r r` = s
r` c C s~ t | t�r| f} t |t�r |f}t| �t|� dks<td��tj| � tj|� tjd dk}|rzttj krztj j
dt� d S )Nr zPass at least one argumentr ) �
isinstancer1 �len�AssertionError�_hookrf rg rP � version_info� meta_pathr )rD rC �enabler r r �
install_hooksz s
rt c C s t tjkrtjjt � d S )N)rp rP rr �remover r r r � remove_hooks� s
rv c C s
t tjkS )zG
Returns True if the import hooks are installed, False if not.
)rp rP rr r r r r � detect_hooks� s rw c @ s e Zd ZdZdd� Zdd� ZdS )�hooksz�
Acts as a context manager. Use like this:
>>> from past import translation
>>> with translation.hooks():
... import mypy2module
>>> import requests # py2/3 compatible anyway
>>> # etc.
c C s t � | _t� | S )N)rw �hooks_were_installedrt )rE r r r � __enter__� s zhooks.__enter__c G s | j s t� d S )N)ry rv )rE �argsr r r �__exit__� s zhooks.__exit__N)r r r r rz r| r r r r rx � s rx c @ s e Zd ZdZdd� Zdd� ZdS )�
suspend_hooksax
Acts as a context manager. Use like this:
>>> from past import translation
>>> translation.install_hooks()
>>> import http.client
>>> # ...
>>> with translation.suspend_hooks():
>>> import requests # or others that support Py2/3
If the hooks were disabled before the context, they are not installed when
the context is left.
c C s t � | _t� | S )N)rw ry rv )rE r r r rz � s zsuspend_hooks.__enter__c G s | j r t� d S )N)ry rt )rE r{ r r r r| � s zsuspend_hooks.__exit__N)r r r r rz r| r r r r r} � s
r} )r r )r r
)-r rP rq � importlibrR �loggingr rc Zlib2to3.pgen2.parser �lib2to3.refactorr Z libfuturizer �importlib.machineryr r � ImportError�object� getLoggerr r2 �setLevel�DEBUG�listZ libfuturize_fix_names_stage1Zlib2to3_fix_names_stage1Z libfuturize_fix_names_stage2Zlib2to3_fix_names_stage2r r r r" r+ r9 r; r< r` rp rt rv rw rx r} Z
autotranslater r r r �<module>! s�
58: |