U
�.W�v � @ s\ d Z ddlmZmZmZ ddlZddlZddlZddlZzddl m
Z
W n e k
rh ddl m
Z
Y nX ddl
mZ dddd gZeej�Zeej�ZeeejfZd"d
d �Zd d
� Zdd � ZG dd� de�ZG dd� de�Z G dd� de �Z e ddd�Z e ddd�Z e ddd�Z!e dd d�Z"e d d d�Z#G d d� de�Z$G d d� de�Z%G d!d� de�Z&dS )#z�Function signature objects for callables
Back port of Python 3.3's function signature tools from the inspect module,
modified to be compatible with Python 2.6, 2.7 and 3.3+.
� )�absolute_import�division�print_functionN)� OrderedDict)� __version__�BoundArguments� Parameter� Signature� signaturec C s8 t | t�r0| jdd|fkr | jS | jd | j S t| �S )N�builtinsZ __builtin__�.)�
isinstance�type�
__module__�__name__�repr)�
annotationZ base_module� r � /__init__.py�formatannotation s
r c G s\ z2| t krW d S t| |�}|D ]}t|||�}q W n tk
rH Y d S X t|t�sX|S d S �N)r �getattr�AttributeErrorr
�_NonUserDefinedCallables)�clsZ method_name�nestedZmeth�namer r r �_get_user_defined_method' s
r c
C s� t | �std�| ���t| tj�r�t| j�}| jdkr:|S t |j
� � �}|r^|d j t
tfkrftd��|d j }|ttfkr�|dd� }n|tk r�td��|j|d�S z
| j}W n tk
r� Y nX |dk r�|S z
| j}W n tk
r� Y n
X t|�S t| tj��rt�| �S t| tj��rt| j �}t |j
� � �}| j �pBd}| j �pNi }z|j!||�}W n6 tk
�r� } zd �| �}
t|
��W 5 d} ~ X Y nX |j"� � D ]R\} } || }
| |k�r�|
j| d
d �|| < n"|
j t
tfk�r�|
j#�s�|�$| � �q�|j|� � d�S d}t| t%��rtt&t%| �d �}|dk �r8t|�}n:t&| d
�}|dk �rVt|�}n t&| d�}|dk �r�t|�}n.t| t'��s�t&t%| �d d�}|dk �r�t|�}|dk �r�|jt |j
� � �dd� d�S t| tj(��r�d�| �}
t|
��td�| ���dS )z/Get a signature object for the passed callable.z {0!r} is not a callable objectNr zinvalid method signature� zinvalid argument type)�
parametersr z,partial object {0!r} has incorrect argumentsT)�default�_partial_kwarg�__call__�__new__�__init__Zim_funcz-no signature found for builtin function {0!r}z,callable {0!r} is not supported by signature))�callable� TypeError�formatr
�types�
MethodTyper
�__func__�__self__�tupler �values�kind� _VAR_KEYWORD�
_KEYWORD_ONLY�
ValueError�_POSITIONAL_OR_KEYWORD�_POSITIONAL_ONLY�_VAR_POSITIONAL�replaceZ
__signature__r � __wrapped__� FunctionTyper �
from_function� functools�partial�funcr �items�args�keywords� bind_partial� argumentsr! �popr r r �BuiltinFunctionType)�objZsig�paramsr. �wrappedZ
new_paramsZ partial_argsZpartial_keywordsZbaZex�msgZarg_nameZ arg_value�paramZcall�newZinitr r r r
7 s�
� �
c @ s e Zd ZdZdS )�_voidz0A private marker - used in Parameter & SignatureN)r r � __qualname__�__doc__r r r r rI � s rI c @ s e Zd ZdS )�_emptyN)r r rJ r r r r rL � s rL c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_ParameterKindc O s t j| f|�� }|d |_|S )Nr )�intr# �_name)�selfr= �kwargsrC r r r r# � s
z_ParameterKind.__new__c C s | j S r �rO �rP r r r �__str__� s z_ParameterKind.__str__c C s d� | j�S )Nz<_ParameterKind: {0!r}>)r' rO rS r r r �__repr__� s z_ParameterKind.__repr__N)r r rJ r# rT rU r r r r rM � s rM �POSITIONAL_ONLY�r r �POSITIONAL_OR_KEYWORD� �VAR_POSITIONAL� � KEYWORD_ONLY� � VAR_KEYWORDc @ s� e Zd ZdZdZeZeZe Z
e Z e
ZeZeedfdd�Zedd� �Zedd � �Zed
d � �Zed d
� �Zeeeeefdd�Zdd� Zdd� Zdd� Zdd� Z dd� Z dS )r a� Represents a parameter in a function signature.
Has the following public attributes:
* name : str
The name of the parameter as a string.
* default : object
The default value for the parameter if specified. If the
parameter has no default value, this attribute is not set.
* annotation
The annotation for the parameter if specified. If the
parameter has no annotation, this attribute is not set.
* kind : str
Describes how argument values are bound to the parameter.
Possible values: `Parameter.POSITIONAL_ONLY`,
`Parameter.POSITIONAL_OR_KEYWORD`, `Parameter.VAR_POSITIONAL`,
`Parameter.KEYWORD_ONLY`, `Parameter.VAR_KEYWORD`.
)rO �_kind�_default� _annotationr! Fc C s� |t ttttfkrtd��|| _|tk rF|ttfkrFd�|�}t|��|| _ || _
|d krr|t krjtd��|| _ n8t |�}|t kr�t
�d|t
j�s�d�|�}t|��|| _ || _d S )Nz,invalid value for 'Parameter.kind' attributez){0} parameters cannot have default valuesz<None is not a valid name for a non-positional-only parameterz
[a-z_]\w*$z#{0!r} is not a valid parameter name)r3 r2 r4 r0 r/ r1 r_ rL r' r` ra rO �str�re�match�Ir! )rP r r. r r r! rF r r r r$ � s. �
zParameter.__init__c C s | j S r rR rS r r r r s zParameter.namec C s | j S r )r` rS r r r r s zParameter.defaultc C s | j S r )ra rS r r r r s zParameter.annotationc C s | j S r )r_ rS r r r r. ! s zParameter.kindc C s\ |t kr| j}|t kr | j}|t kr*| j}|t kr8| j}|t krF| j}t| �|||||d�S )z+Creates a customized copy of the Parameter.)r r r! )rI rO r_ ra r` r! r )rP r r. r r r! r r r r5 % s �zParameter.replacec C s� | j }| j}|tkr*|d kr d}d�|�}| jtk rFd�|t| j��}| jtk rbd�|t| j��}|t krtd| }n|t
kr�d| }|S )N� z<{0}>z{0}:{1}z{0}={1}�*z**) r. rO r3 r' ra rL r r` r r4 r/ )rP r. � formattedr r r rT ; s"
�
zParameter.__str__c C s d� | jjt| �| j�S )Nz<{0} at {1:#x} {2!r}>)r' � __class__r �idr rS r r r rU S s
�zParameter.__repr__c C s d� | jj�}t|��d S �Nzunhashable type: '{0}'�r' ri r r& �rP rF r r r �__hash__W s zParameter.__hash__c C s< t |jt�o:| j|jko:| j|jko:| j|jko:| j|jkS r )�
issubclassri r rO r_ r` ra �rP �otherr r r �__eq__[ s
�
�
�
�zParameter.__eq__c C s | � |� S r �rr rp r r r �__ne__b s zParameter.__ne__N) r r rJ rK � __slots__r3 rV r2 rX r4 rZ r0 r\ r/ r^ rL �emptyr$ �propertyr r r r. rI r5 rT rU rn rr rt r r r r r � s8 �
�
c @ sT e Zd ZdZdd� Zedd� �Zedd� �Zedd � �Zd
d � Z d d
� Z
dd� Z dS )r a Result of `Signature.bind` call. Holds the mapping of arguments
to the function's parameters.
Has the following public attributes:
* arguments : OrderedDict
An ordered mutable mapping of parameters' names to arguments' values.
Does not contain arguments' default values.
* signature : Signature
The Signature object that created this instance.
* args : tuple
Tuple of positional arguments values.
* kwargs : dict
Dict of keyword arguments values.
c C s || _ || _d S r )r@ �
_signature)rP r
r@ r r r r$ w s zBoundArguments.__init__c C s | j S r )rx rS r r r r
{ s zBoundArguments.signaturec C s� g }| j j�� D ]j\}}|jttfks,|jr0 q|z| j| }W n tk
rX Y q|Y qX |jt krp|�
|� q|� |� qt |�S r )
rx r r< r. r/ r0 r! r@ �KeyErrorr4 �extend�appendr, )rP r= �
param_namerG �argr r r r= s �
zBoundArguments.argsc C s� i }d}| j j�� D ]~\}}|sJ|jttfks4|jr:d}n|| jkrJd}q|sPqz| j| }W n tk
rr Y qX |jtkr�|� |� q|||< q|S )NFT)
rx r r< r. r/ r0 r! r@ ry �update)rP rQ Zkwargs_startedr| rG r} r r r rQ � s* �
zBoundArguments.kwargsc C s d� | jj�}t|��d S rk rl rm r r r rn � s zBoundArguments.__hash__c C s$ t |jt�o"| j|jko"| j|jkS r )ro ri r r
r@ rp r r r rr � s
�
�zBoundArguments.__eq__c C s | � |� S r rs rp r r r rt � s zBoundArguments.__ne__N) r r rJ rK r$ rw r
r= rQ rn rr rt r r r r r f s
c @ s� e Zd ZdZdZeZeZe Z
de dfdd�Z e dd� �Z
ed d
� �Zed d � �Zeefd
d�Zdd� Zdd� Zdd� Zd dd�Zdd� Zdd� Zd d � ZdS ) r a A Signature object represents the overall signature of a function.
It stores a Parameter object for each parameter accepted by the
function, as well as information specific to the function itself.
A Signature object has the following public attributes and methods:
* parameters : OrderedDict
An ordered mapping of parameters' names to the corresponding
Parameter objects (keyword-only arguments are in the same order
as listed in `code.co_varnames`).
* return_annotation : object
The annotation for the return type of the function if specified.
If the function has no annotation for its return type, this
attribute is not set.
* bind(*args, **kwargs) -> BoundArguments
Creates a mapping from positional and keyword arguments to
parameters.
* bind_partial(*args, **kwargs) -> BoundArguments
Creates a partial mapping from positional and keyword arguments
to parameters (simulating 'functools.partial' behavior.)
)�_return_annotation� _parametersNTc C s� |dkrt � }n�|r�t � }t}t|�D ]z\}}|j}||k rXd} | �||j�} t| ��n|}|j}
|
dkr~t|�}
|j|
d�}|
|kr�d�|
�} t| ��|||
< q&nt dd� |D ��}|| _ || _
dS )z�Constructs Signature from the given list of Parameter
objects and 'return_annotation'. All arguments are optional.
Nz%wrong parameter order: {0} before {1}rW z duplicate parameter name: {0!r}c s s | ]}|j |fV qd S r rW )�.0rG r r r � <genexpr> s �z%Signature.__init__.<locals>.<genexpr>) r r3 � enumerater. r' r1 r rb r5 r� r ) rP r �return_annotation�__validate_parameters__rD Ztop_kind�idxrG r. rF r r r r r$ � s2
�zSignature.__init__c C s� t |tj�std�|���| j}|j}|j}|j}t |d|� �}t
|dd�}|||| � }t
|di �} |j }
t
|dd�} |
r�t |
�} nd} g }
|| }|d|� D ]$}| �
|t�}|
�|||td�� q�t||d� �D ].\}}| �
|t�}|
�|||t|
| d�� q�|jd @ �rD||| }| �
|t�}|
�|||td�� |D ]B}t}| dk �rf| �
|t�}| �
|t�}|
�|||t|d�� �qH|jd
@ �r�|| }|jd @ �r�|d 7 }|| }| �
|t�}|
�|||td�� | |
| �
d t�d
d�S )z2Constructs Signature for the given python functionz {0!r} is not a Python functionN�co_kwonlyargcountr �__annotations__�__kwdefaults__)r r. )r r. r r] � r �returnF)r� r� )r
r( r7 r& r' �_parameter_cls�__code__� co_argcount� co_varnamesr, r � __defaults__�len�getrL r{ r2 r� �co_flagsr4 r0 r/ )r r; r Z func_codeZ pos_countZ arg_namesZ
positionalZkeyword_only_countZ keyword_onlyZ annotations�defaultsZ
kwdefaultsZpos_default_countr Znon_default_countr r �offsetr �indexr r r r8 sp
�
�
�
�
�
�zSignature.from_functionc C s6 zt �| j�W S tk
r0 t| j�� � Y S X d S r )r( �MappingProxyTyper� r r r< rS r r r r T s zSignature.parametersc C s | j S r )r rS r r r r� [ s zSignature.return_annotationc C s0 |t kr| j�� }|t kr | j}t| �||d�S )z�Creates a customized copy of the Signature.
Pass 'parameters' and/or 'return_annotation' arguments
to override them in the new copy.
)r� )rI r r- r r )rP r r� r r r r5 _ s
�zSignature.replacec C s d� | jj�}t|��d S rk rl rm r r r rn n s zSignature.__hash__c C s� t t|�t�r.| j|jks.t| j�t|j�kr2dS tdd� t|j�� �D ��}t| j� � �D ]�\}\}}|j
t kr�z|j| }W n t k
r� Y dS X ||kr� dS q\z || }W n t k
r� Y dS X ||ks�||j| kr\ dS q\dS )NFc s s | ]\}}||fV qd S r r )r� r� rG r r r r� x s �z#Signature.__eq__.<locals>.<genexpr>T)
ro r r r� r� r �dictr� �keysr< r. r0 ry )rP rq Zother_positionsr� r| rG Z other_paramZ other_idxr r r rr r s4
�� �
�zSignature.__eq__c C s | � |� S r rs rp r r r rt � s zSignature.__ne__Fc C s� t � }t| j�� �}d}t|�}|rP| j�� D ] \}} | jr.||kr.| j||< q.z t|�}
W n� tk
�r< z t|�} W n tk
r� Y Y �q�Y n�X | j t
kr�Y �q�n�| j |kr�| j t kr�d} | j
| j d�} t| ��| f}Y �q�nR| j tks�| jtk �r
| f}Y �q�n.|�r | f}Y �q�nd} | j
| j d�} t| ��Y qPX z t|�} W n tk
�rh td��Y qPX | j ttfk�r�td��| j t
k�r�|
g} | �|� t| �|| j < �q�| j |k�r�tdj
| j d���|
|| j < qPd}
t�||�D ]�} | j t k�r tdj
| j d���| j tk�r | }
�q�| j }z|�|�}
W nD tk
�rx |�st| j t
k�rt| jtk�rttdj
|d���Y n
X |
||< �q�|�r�|
dk �r�|||
j < n td| ��| �| |�S ) z$Private method. Don't use directly.r zA{arg!r} parameter is positional only, but was passed as a keyword)r} z'{arg!r} parameter lacking default valuez too many positional argumentsz$multiple values for argument {arg!r}Nz too many keyword arguments %r)r �iterr r- r< r! r �next�
StopIterationr. r4 r r3 r' r& r/ rL r0 rz r, � itertools�chainrA ry �_bound_arguments_cls)rP r= rQ r: r@ r Z
parameters_exZarg_valsr| rG Zarg_valrF r- Z kwargs_paramr r r �_bind� s�
�
� � ��
zSignature._bindc O s | d � | dd� |�S )z�Get a BoundArguments object, that maps the passed `args`
and `kwargs` to the function's signature. Raises `TypeError`
if the passed arguments can not be bound.
r r N�r� )r= rQ r r r �bind s zSignature.bindc O s | j ||dd�S )z�Get a BoundArguments object, that partially maps the
passed `args` and `kwargs` to the function's signature.
Raises `TypeError` if the passed arguments can not be bound.
T)r: r� )rP r= rQ r r r r? s zSignature.bind_partialc C s� g }d}t | j�� �D ]H\}}t|�}|j}|tkr:d}n|tkrT|rT|�d� d}|�|� qd�d� |��}| j
t k r�t | j
�}|d�|�7 }|S )NTFrg z({0})z, z -> {0})
r� r r- rb r. r4 r0 r{ r' �joinr� rL r ) rP �resultZrender_kw_only_separatorr� rG rh r. ZrenderedZannor r r rT ! s
zSignature.__str__)F)r r rJ rK ru r r� r r� rL rv r$ � classmethodr8 rw r r� rI r5 rn rr rt r� r� r? rT r r r r r � s. �
&
H
)N)'rK Z
__future__r r r r� r9 rc r( � collectionsr � ImportErrorZ ordereddictZfuncsigs.versionr �__all__r r" Z_WrapperDescriptor�allZ_MethodWrapperrB r r r r
�objectrI rL rN rM r3 r2 r4 r0 r/ r r r r r r r �<module> sB
�
a