3
� �f(, � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZyd dl m
Z
W n d dl m Z
Y nX y d dl Z W n d dl
jZ Y nX y d dlZW n d dlZY nX d dd�ZG dd� de�Zd d
� ZdS ) � )�PluginN)�sha1)�shac C sJ |d kr i }|j | dd�� t| d�}tt|j� | d�||� |j� d S )N�__main__)�__file__�__name__�rb�exec)�update�openr �compile�read�close)�filepath�globals�locals�f� r �+/usr/lib/fm-agent/library/plugin_manager.py� execfile3 s
r c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zed d � �Z d
d� Z
dd� Z dS )�
PluginManagerc G s� t j| jj�| _|| _|| _|| _| j|�| _ d | _
d | _ i | _ t
jj| j�r�tj� }|j| j� xf|j� D ]Z}y|jdd�}W n d}Y nX d|i| j |< x$|j|�D ]\}}|| j | |< q�W qlW d S )N�agent�debugF)�logging� getLogger� __class__r �log�db� config_file� directories�
_load_plugins�plugins� _metadata�_hashed_metadata�config�os�path�exists� configparser� ConfigParserr
�sections�get�items)�selfr r r �textkeyr �k�vr r r �__init__% s&
zPluginManager.__init__c C s | j jdd �}|r||d< d S )N�dem� wifi_results)r$ r+ )r- r3 Z
dem_configr r r �add_dem_wifi_results= s z"PluginManager.add_dem_wifi_resultsc C s8 | j d }tjdd�}d}|jd�d }|jd�d }yHtj|�}d|j� krf|j� jd�jd�d }|j|j � � |j
� W n( t k
r� | j j
dtj� � d S X |dk�r0yPtj|j�}x>|j� D ]2} | jd �r�| j jd
| � |j| |� |d7 }q�W W n* t k
�r, | j j
dtj� � Y nX n�|dk�r�yTtj|j�}
xB|
j� D ]6} | jd ��rR| j jd
| � |
j| |� |d7 }�qRW W n* t k
�r� | j j
dtj� � Y nX nR|dk�r�| j jd
| � tjd|jtjj ||�f � |d7 }n| j j
d| � |�r$| j jd| � tjd|j � d S )zu
Grab the remote url and extract the contents, putting them in the
appropiate plugin folder.
� F)�deleter �.�/zcontent-dispositionzUnable to download URL: %sN�tar�tgzz.pyz Installing plugin %sz"Unable to extract tar contents: %s�zipz"Unable to extract zip contents: %s�pyzcp %s %sz5Unable to install Agent Plugin, unknown extension: %sz
Installed %s Agent pluginszrm -f %s���r= r= r= )r9 r: ) r �tempfileZNamedTemporaryFile�split�urllib2�urlopen�info� getheader�writer
r � Exceptionr �error� traceback�
format_exc�tarfiler �nameZgetnames�endswith�extract�zipfileZZipFileZnamelistr% �systemr& �join) r- �url�base_custom_plugin_dirr Z
num_installedZext�fname�r�t�file�zr r r �install_remote_pluginsB sT
z$PluginManager.install_remote_pluginsc C s4 y�| j jdi �jd�}d }t|dd �r�|j|kr�|jjd�d }y| jd | }W n< |jdkrx|jdkrxdd i}n| jj d
|� d S Y nX | j
|j |�}| j j|ji �}|jd d �}|r�|j ||j|j |�}n|j
|j|j |�}|d k r�||9 }W n4 tk
�r. | jjd
|j|jtj� f � d S X |S )Nr �
server_key�:r5 �docker_containersZdockerzstatus.runningZStateZdeletedz#Could not find container with id %s�scaleg �?z%s/%s check failed: %sr= )r$ r+ �getattrrX r? r �plugin_textkey�resource_textkeyr rF r! Z check_docker�option�checkrE � exceptionrG rH ) r- �scheduleZself_key� container� container_id�pluginr$ r[ �valuer r r r` w s6
zPluginManager.checkc s� i �d}t jj�j�rftj� }|j�j� y|jdd�}W n d}Y nX |rftt t
j |j d���}� ��fdd�}xJ|� D ]@}t
|t�r~t|t�r~t|d�r~|j|kr~|jdkr~|�|j< q~W �S ) N� r �plugin_blacklist�,c
3 s �x� D �]} �j jd| � x�tj| �D ]�}y�|d
d � dksF|dkrHw(tjj| |�}i }i }tjd dkrzt|||� n t |||� xt
|j � �D ]
}|V q�W �j j d|� W q( t
k
r� tj� d d� \}}�j jd|� �j j |� w(Y q(X q(W �j jd t��� qW d S ) NzLooking in directory %r� z.pyz __init__.pyr z Loaded %r� z#Error importing plugin %s, skippingzLoaded %d plugins���)r rB r% �listdirr& rO �sys� version_infor Zexecfile�list�valuesr rE �exc_info�len)Z
plugin_dirrJ Z plugin_file� f_globals�f_locals�objrT �e)r r! r- r r �plugin_file_checker� s,
z8PluginManager._load_plugins.<locals>.plugin_file_checkerr. ZTEMPLATE_PLUGIN)r% r& r' r r( r) r
r+ rp �map�str�stripr? �
isinstance�type�
issubclassr �hasattrr. )r- r rh r rx rv r )r r! r- r r � s$
"
zPluginManager._load_pluginsc C s� | j d kr�i }x�t| jj� �D ]�}| jj|ji �}yt|j|��}W n | j j
d|j� w Y nX | j j dt t|j
� ��|j� t|d|j�}||f||j< q W || _ | j S )Nz error getting metadata from %rzgot %d textkeys for %r�label)r" rp r! rq r$ r+ r. �format_metadata� get_metadatar ra rB rs �keysr\ )r- �metadatare r$ �metar� r r r r� � s
zPluginManager.metadatac C s� | j d kr�g }xvt| jj� �D ]d\}\}}g }x6t|j� �D ]&\}}tt|��}|jt||f�� q<W |j� |jt|||f�� q W |j� tt|�j � �j
� | _ | j S )N) r# rp r� r, �sorted�repr�append�sortr �encode� hexdigest) r- Z flat_datar. r� �plugin_metadataZplugin_optionsZ
resource_nameZresource_metadataZ
resource_metar r r �hashed_metadata� s
z PluginManager.hashed_metadatac C s, | j jd| j� | jd � | jd | j� kS )Nz$comparing metadata hashes %r with %r�
last_metadata)r rB r� r )r- r r r �is_metadata_stale s z PluginManager.is_metadata_staleN) r �
__module__� __qualname__r1 r4 rW r` r �propertyr� r� r� r r r r r $ s 5'< r c C s� x�| j � D ]�\}}|d }g }|s$q
xT|D ]L}tjd dkrXt|�tkrv|jd|i� q*t|�ttfkr*|jd|i� q*W |r�||d< |jd�s
ddi|d< q
W | S )z�
Go over the metadata entry and make sure the option resources
are matching to the new style of dictionaries if they
are a string.
�optionsr rj �resourceZoptions_schema�string)r, rn ro r} rz r� �unicoder+ )r� r� �entryr� Z new_optionsr_ r r r r� s
r� )NN)�
agent_utilr r r% rn rG r> rI rM �hashlibr r r@ �urllib.request�requestr) r( r �objectr r� r r r r �<module> s.
f