Edit File: manager.cpython-37.pyc
B p9�`�< � @ s� d Z ddlZddlZddlZddlZddlmZ ddlm Z ddl Z ddl m Z ddlmZ ddlmZ y ddlZW n ddlZY nX yddlmZ W n ddlmZ Y nX d d d d d gZe�e�ZG dd� de�ZG dd� de�ZG dd � d e�Z G dd� d�Z G dd � d e �Z G dd � d e �Z y ddl Z G dd � d e e �Z!W n& e"k �rn G dd � d e �Z!Y nX G dd � d e!�Z#dS )a Plugin Manager -------------- A plugin manager class is used to load plugins, manage the list of loaded plugins, and proxy calls to those plugins. The plugin managers provided with nose are: :class:`PluginManager` This manager doesn't implement loadPlugins, so it can only work with a static list of plugins. :class:`BuiltinPluginManager` This manager loads plugins referenced in ``nose.plugins.builtin``. :class:`EntryPointPluginManager` This manager uses setuptools entrypoints to load plugins. :class:`ExtraPluginsPluginManager` This manager loads extra plugins specified with the keyword `addplugins`. :class:`DefaultPluginMananger` This is the manager class that will be used by default. If setuptools is installed, it is a subclass of :class:`EntryPointPluginManager` and :class:`BuiltinPluginManager`; otherwise, an alias to :class:`BuiltinPluginManager`. :class:`RestrictedPluginManager` This manager is for use in test runs where some plugin calls are not available, such as runs started with ``python setup.py test``, where the test runner is the default unittest :class:`TextTestRunner`. It is a subclass of :class:`DefaultPluginManager`. Writing a plugin manager ======================== If you want to load plugins via some other means, you can write a plugin manager and pass an instance of your plugin manager class when instantiating the :class:`nose.config.Config` instance that you pass to :class:`TestProgram` (or :func:`main` or :func:`run`). To implement your plugin loading scheme, implement ``loadPlugins()``, and in that method, call ``addPlugin()`` with an instance of each plugin you wish to make available. Make sure to call ``super(self).loadPlugins()`` as well if have subclassed a manager other than ``PluginManager``. � N)�chain)�warn)�Failure)�IPluginInterface)� sort_list)�StringIO�DefaultPluginManager� PluginManager�EntryPointPluginManager�BuiltinPluginManager�RestrictedPluginManagerc @ sV e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z ddd�Z dS )� PluginProxya# Proxy for plugin calls. Essentially a closure bound to the given call and plugin list. The plugin proxy also must be bound to a particular plugin interface specification, so that it knows what calls are available and any special handling that is required for each call. c C sl yt | j|�| _W n( tk r: td|| jjf ��Y nX | �|�| _g | _x|D ]}| �||� qTW d S )Nz%s is not a valid %s method) �getattr� interface�method�AttributeError�__name__�makeCall�call�plugins� addPlugin)�selfr r �p� r �E/opt/alt/python37/lib/python3.7/site-packages/nose/plugins/manager.py�__init__W s zPluginProxy.__init__c O s | j ||�S )N)r )r �arg�kwr r r �__call__b s zPluginProxy.__call__c sV t ||d�}|dk rR|dkrBtt�|�d �dkrB|� � fdd�}| j�||f� dS )z`Add plugin to my list of plugins to call, if it has the attribute I'm bound to. N�loadTestsFromModuler � c s � | �S )Nr )�module�path�kwargs)� orig_methr r �<lambda>n � z'PluginProxy.addPlugin.<locals>.<lambda>)r �len�inspect� getargspecr �append)r �pluginr �methr )r$ r r e s zPluginProxy.addPluginc sH |dkr� j S � j}t|dd�r,� fdd�S t|dd�r>� jS � jS d S )N�loadTestsFromNames� generativeFc s t � j| |��S )N)�list�generate)r r )r r r r% { r& z&PluginProxy.makeCall.<locals>.<lambda>Z chainable)�_loadTestsFromNamesr r r �simple)r r r, r )r r r q s zPluginProxy.makeCallc O sZ d}dd� t t| jdg �|�D �}x2| jD ](\}}|||�}|dd� }|�|� q*W |S )z�Call plugins in a chain, where the result of each plugin call is sent to the next plugin as input. The final output result is returned. Nc S s g | ]\}}|r|�qS r r )�.0�static�ar r r � <listcomp>� s z%PluginProxy.chain.<locals>.<listcomp>Z static_args)�zipr r r r* )r r r �resultr4 r r, r r r r � s zPluginProxy.chainc o s� xz| j D ]p\}}d}y*|||�}|dk r<x|D ] }|V q.W W q ttfk rX � Y q t�� }t|� V wY qX qW dS )zFCall all plugins, yielding each item in each non-None result. N)r �KeyboardInterrupt� SystemExit�sys�exc_infor )r r r r r, r8 �r�excr r r r0 � s zPluginProxy.generatec O s. x(| j D ] \}}|||�}|dk r|S qW dS )z?Call all plugins, returning the first non-None result. N)r )r r r r r, r8 r r r r2 � s zPluginProxy.simpleNc C sJ g }x<| j D ]2\}}|||d�}|dk r |\}}|r |�|� q W ||fS )a Chainable but not quite normal. Plugins return a tuple of (tests, names) after processing the names. The tests are added to a suite that is accumulated throughout the full call, while names are input for the next plugin in the chain. )r! N)r �extend)r �namesr! �suiter r, r8 Z suite_partr r r r1 � s z PluginProxy._loadTestsFromNames)N)r � __module__� __qualname__�__doc__r r r r r r r r0 r2 r1 r r r r r N s r c @ sd e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� ZdS )� NoPluginsz(Null Plugin manager that has no plugins.c C s d | _ | _d S )Nr )�_pluginsr )r r r r r � s zNoPlugins.__init__c C s dS )Nr r )r r r r �__iter__� s zNoPlugins.__iter__c O s d S )Nr )r �args�kwdsr r r � _doNothing� s zNoPlugins._doNothingc O s dS )Nr r )r rH rI r r r �_emptyIterator� s zNoPlugins._emptyIteratorc C s( t | j|�}t |dd�r | jS | jS d S )Nr. F)r r rK rJ )r r r r r r � __getattr__� s zNoPlugins.__getattr__c C s t � �d S )N)�NotImplementedError)r �plugr r r r � s zNoPlugins.addPluginc C s t � �d S )N)rM )r r r r r � addPlugins� s zNoPlugins.addPluginsc C s d S )Nr )r �options�configr r r � configure� s zNoPlugins.configurec C s d S )Nr )r r r r � loadPlugins� s zNoPlugins.loadPluginsc C s d S )Nr )r r r r �sort� s zNoPlugins.sortN)r rB rC rD r r r rG rJ rK rL r rO rR rS rT r r r r rE � s rE c @ sv e Zd ZdZeZddd�Zdd� Zdd � Zd d � Z dd d �Z dd� Z dd� Z dd� Z dd� Zdd� Zeeedd�ZdS )r a� Base class for plugin managers. PluginManager is intended to be used only with a static list of plugins. The loadPlugins() implementation only reloads plugins from _extraplugins to prevent those from being overridden by a subclass. The basic functionality of a plugin manager is to proxy all unknown attributes through a ``PluginProxy`` to a list of plugins. Note that the list of plugins *may not* be changed after the first plugin call. r Nc C s2 g | _ d| _i | _|r | �|� |d k r.|| _d S )Nr )rF � _extraplugins�_proxiesrO � proxyClass)r r rW r r r r � s zPluginManager.__init__c C s< y | j | S tk r6 | �|| j�}|| j |< Y nX |S )N)rV �KeyErrorrW rF )r r �proxyr r r rL � s zPluginManager.__getattr__c C s t | j�S )N)�iterr )r r r r rG s zPluginManager.__iter__c s< t |dt� �� � fdd�| jD �| jd d �< | j�|� d S )N�namec s g | ]}t |d d�� kr|�qS )r[ N)r )r3 r )�new_namer r r6 s z+PluginManager.addPlugin.<locals>.<listcomp>)r �objectrF r* )r rN r )r\ r r s zPluginManager.addPluginc C s( || _ x t||�D ]}| �|� qW dS )z�extraplugins are maintained in a separate list and re-added by loadPlugins() to prevent their being overwritten by plugins added by a subclass of PluginManager N)rU � iterchainr )r r Z extrapluginsrN r r r rO s zPluginManager.addPluginsc C sT t �d� || _td| j�}|||� dd� | jD �}|| _| �� t �d|� dS )z�Configure the set of plugins with the given options and config instance. After configuration, disabled plugins are removed from the plugins list. zConfiguring pluginsrR c S s g | ]}|j r|�qS r )�enabled)r3 rN r r r r6 s z+PluginManager.configure.<locals>.<listcomp>zPlugins enabled: %sN)�log�debugrQ r rF r rT )r rP rQ Zcfgr_ r r r rR s zPluginManager.configurec C s x| j D ]}| �|� qW d S )N)rU r )r rN r r r rS " s zPluginManager.loadPluginsc C s t | jdd� dd�S )Nc S s t | dd�S )N�score� )r )�xr r r r% ' r& z$PluginManager.sort.<locals>.<lambda>T)�reverse)r rF )r r r r rT & s zPluginManager.sortc C s | j S )N)rF )r r r r � _get_plugins) s zPluginManager._get_pluginsc C s g | _ | �|� d S )N)rF rO )r r r r r � _set_plugins, s zPluginManager._set_pluginszPAccess the list of plugins managed by this plugin manager)r N)r r )r rB rC rD r rW r rL rG r rO rR rS rT rf rg �propertyr r r r r r � s c @ s^ e Zd ZdZdd� Zejfdd�Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� ZdS )�ZeroNinePluginz>Proxy for 0.9 plugins, adapts 0.10 calls to 0.9 standard. c C s || _ d S )N)r+ )r r+ r r r r 8 s zZeroNinePlugin.__init__c C s | j �||� d S )N)r+ Z add_options)r �parser�envr r r rP ; s zZeroNinePlugin.optionsc C s� t | jd�sd S ddlm}m} |\}}}t||�rRt | jd�sDd S | j�|j�S t||�rzt | jd�sld S | j�|j�S |j }| j� |j||�S )N�addErrorr )�SkipTest�DeprecatedTest�addSkip� addDeprecated) �hasattrr+ Znose.excrm rn � issubclassro �testrp �capturedOutputrl ) r rs �errrm rn ZecZev�tb�captr r r rl > s zZeroNinePlugin.addErrorc C s t | jd�r| j�|�S d S )N�loadTestsFromPath)rq r+ rx )r �filenamer r r �loadTestsFromFileP s z ZeroNinePlugin.loadTestsFromFilec C s0 t | jd�sd S |j}|j}| j�|j|||�S )N� addFailure)rq r+ rt �tbinfor{ rs )r rs ru rw r| r r r r{ T s zZeroNinePlugin.addFailurec C s* t | jd�sd S |j}| j�|j|� d S )N� addSuccess)rq r+ rt r} rs )r rs rw r r r r} \ s zZeroNinePlugin.addSuccessc C s t | jd�sd S | j�|j�S )N� startTest)rq r+ r~ rs )r rs r r r r~ b s zZeroNinePlugin.startTestc C s t | jd�sd S | j�|j�S )N�stopTest)rq r+ r rs )r rs r r r r g s zZeroNinePlugin.stopTestc C s t | j|�S )N)r r+ )r �valr r r rL l s zZeroNinePlugin.__getattr__N)r rB rC rD r �os�environrP rl rz r{ r} r~ r rL r r r r ri 5 s ri c s, e Zd ZdZddeffZ� fdd�Z� ZS )r zhPlugin manager that loads plugins from the `nose.plugins` and `nose.plugins.0.10` entry points. )znose.plugins.0.10Nz nose.pluginsc s� ddl m} i }x�| jD ]�\}}x�||�D ]�}|j|kr:q*d||j< t�d| jj|� y |�� }W nJ t k rx � Y n6 t k r� } zt d||f t � w*W dd}~X Y nX |r�||� �}n|� }| � |� q*W qW tt| ��� dS )zBLoad plugins by iterating the `nose.plugins` entry point. r )�iter_entry_pointsTz%s load plugin %sz Unable to load plugin %s: %sN)� pkg_resourcesr� � entry_pointsr[ r` ra � __class__r �loadr9 � Exceptionr �RuntimeWarningr �superr rS ) r r� ZloadedZ entry_pointZadapt�epZplugcls�erN )r� r r rS w s* z#EntryPointPluginManager.loadPlugins)r rB rC rD ri r� rS � __classcell__r r )r� r r p s c s e Zd ZdZ� fdd�Z� ZS )r zSPlugin manager that loads plugins from the list in `nose.plugins.builtin`. c s: ddl m} x|jD ]}| �|� � qW tt| ��� dS )z-Load plugins in nose.plugins.builtin r )�builtinN)Z nose.pluginsr� r r r� r rS )r r� rN )r� r r rS � s z BuiltinPluginManager.loadPlugins)r rB rC rD rS r� r r )r� r r � s c @ s e Zd ZdS )r N)r rB rC r r r r r � s c @ s e Zd ZdS )r N)r rB rC r r r r r � s c @ s* e Zd ZdZd dd�Zdd� Zdd � Zd S ) r z�Plugin manager that restricts the plugin list to those not excluded by a list of exclude methods. Any plugin that implements an excluded method will be removed from the manager's plugin list after plugins are loaded. r Tc C s( t �| |� || _|| _g | _d | _d S )N)r r r� �exclude�excluded� _excludedOpts)r r r� r� r r r r � s z RestrictedPluginManager.__init__c C sR | j d krBddlm} |dd�| _ x | jD ]}|j| j i d� q*W | j �d| �S )Nr )� OptionParserF)Zadd_help_option)rk z--)r� �optparser� r� rP Z get_option)r r[ r� r+ r r r �excludedOption� s z&RestrictedPluginManager.excludedOptionc C sl | j rt�| � g }xL| jD ]B}d}x*| jD ] }t||�r,d}| j�|� P q,W |r |�|� q W || _d S )NTF)r� r rS r r� rq r� r* )r Zallowr+ �okr r r r rS � s z#RestrictedPluginManager.loadPluginsN)r r T)r rB rC rD r r� rS r r r r r � s )$rD r( �loggingr� r; � itertoolsr r^ �warningsr Z nose.config�noseZ nose.failurer Znose.plugins.baser Znose.pyversionr �pickle�ior �__all__� getLoggerr r` r] r rE r ri r r r� r � ImportErrorr r r r r �<module>2 sB m&T;%