File: //lib/fm-agent/library/__pycache__/plugin_manager.cpython-39.pyc
a
��i�. � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZzd dl m
Z
W n d dl m Z
Y n0 z d dl Z W n d dl
mZ Y n0 z d dlZW n d dlZY n0 d dd�ZG dd� de�Zd d
� ZdS ) � )�PluginN)�sha1)�shac C sJ |d u r i }|� | dd�� t| d�}tt|�� | d�||� |�� 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 �| jj�| _|| _|| _|| _| �|�| _ d | _
d | _ i | _ t
j�| j�r�t�� }|�| j� |�� D ]V}z|�dd�}W n d}Y n0 d|i| j |< |�|�D ]\}}|| j | |< q�qjd 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 �dd �}|r||d< d S )N�dem� wifi_results)r$ r+ )r- r3 Z
dem_configr r r �add_dem_wifi_resultsC s z"PluginManager.add_dem_wifi_resultsc C s, | j d }tjdd�}d}|�d�d }|�d�d }zHt�|�}d|�� v rf|�� �d��d�d }|�|� � � |�
� W n( t y� | j �
dt�� � Y d S 0 |d
v �r*zLt�|j�}|�� D ]2} | �d �r�| j �d | � |�| |� |d
7 }q�W n( t �y& | j �
dt�� � Y n0 n�|dk�r�zPt�|j�}
|
�� D ]6} | �d ��rJ| j �d | � |
�| |� |d
7 }�qJW n( t �y� | j �
dt�� � Y n0 nR|dk�r�| j �d | � t�d|jtj� ||�f � |d
7 }n| j �
d| � |�r| j �d| � t�d|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)�tarZtgz�.pyz Installing plugin %s� z"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 �tempfileZNamedTemporaryFile�split�urllib2�urlopen�info� getheader�writer
r � Exceptionr �error� traceback�
format_exc�tarfiler �nameZgetnames�endswith�extract�zipfile�ZipFileZnamelistr% �systemr& �join) r- �url�base_custom_plugin_dirr Z
num_installed�ext�fname�r�t�file�zr r r �install_remote_pluginsH sd
�
�
�
�z$PluginManager.install_remote_pluginsc
C s8 z�| j �di ��d�}d }t|dd �r�|j|kr�|j�d�d }z| jd | }W n@ |jdkrx|jdkrxdd i}n| j� d
|� Y W d S Y n0 | j
|j |�}| j �|ji �}|�d d �}|r�|� ||j|j |�}n|�
|j|j |�}|d ur�||9 }W n4 t�y2 | j�d
|j|jt�� f � Y d S 0 |S )Nr �
server_key�:r5 �docker_containersZdockerzstatus.running�StateZdeletedz#Could not find container with id %s�scaleg �?z%s/%s check failed: %s)r$ r+ �getattrrZ 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 rc � sR
����
� � ���zPluginManager.checkc s� i �d}t j��j�rft�� }|��j� z|�dd�}W n d}Y n0 |rftt t
j |� d���}� ��fdd�}|� D ]@}t
|t�r|t|t�r|t|d�r||j|vr||jdkr||�|j< q|�S ) N� r �plugin_blacklist�,c
3 s � D ]�} �j �d| � t�| �D ]�}z�|dd � dks>|dkrBW q tj�| |�}i }i }tjd dkrtt|||� n t |||� t
|� � �D ]
}|V q��j � d|� W q t
y� t�� d d� \}}�j �d |� �j � |� Y q Y q 0 q �j �d
t��� qd S ) NzLooking in directory %r���r: z __init__.pyr � z Loaded %r� z#Error importing plugin %s, skippingzLoaded %d plugins)r rB r% �listdirr& rP �sys� version_infor Zexecfile�list�valuesr rE �exc_info�len)Z
plugin_dirrJ Z plugin_file� f_globals�f_locals�objrV �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+ rs �map�str�stripr? �
isinstance�type�
issubclassr �hasattrr. )r- r rk r r| ry r r{ r r � s.
!
���
zPluginManager._load_pluginsc C s� | j d kr�i }t| j�� �D ]�}| j�|ji �}zt|�|��}W n | j �
d|j� Y q Y n0 | j � dt t|�
� ��|j� t|d|j�}||f||j< q || _ | j S )Nz error getting metadata from %rzgot %d textkeys for %r�label)r" rs r! rt r$ r+ r. �format_metadata� get_metadatar rd rB rv �keysr_ )r- �metadatarh r$ �metar� r r r r� � s
�zPluginManager.metadatac C s� | j d kr�g }t| j�� �D ]`\}\}}g }t|�� �D ]&\}}tt|��}|�t||f�� q8|�� |�t|||f�� q |�� tt|�� � ��
� | _ | j S )N) r# rs 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 �d| �� | jd � | jd | �� 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 rY rc r �propertyr� r� r� r r r r r ( s =5?
r c C s� | � � D ]�\}}|d }g }|s"q|D ]L}tjd dkrTt|�tkrr|�d|i� q&t|�ttfv r&|�d|i� q&|r�||d< |�d�sddi|d< q| 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 rn �resourceZoptions_schema�string)r, rq rr r� r~ r� �unicoder+ )r� r� �entryr� Z new_optionsrb r r r r� - s(
��
r� )NN)�
agent_utilr r r% rq rG r> rI rM �hashlibr r r@ �urllib.request�requestr) r( r �objectr r� r r r r �<module> s0