File: //lib/python3.6/site-packages/agent/controller/__pycache__/__init__.cpython-36.pyc
3
�*�g; � @ sn d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZ e jd�Z G dd� de
�Z e G dd� de
��Z dS ) � N)�
import_module)�domain�featureZkeentunec @ s e Zd Zdd� Zdd� ZdS )� Singletonc C s || _ i | _d S )N)�_cls� _instance)�self�cls� r
� /usr/lib/python3.6/__init__.py�__init__ s zSingleton.__init__c C s( | j | jkr | j � | j| j < | j| j S )N)r r )r r
r
r �__call__ s zSingleton.__call__N)�__name__�
__module__� __qualname__r r
r
r
r
r r s r c @ sV e Zd Zejd�Zdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )�ContextzAclass (.*)\((Feature|BaseDomain|BaseBenchmark|FileConfig|Nginx)\)c C s2 d| _ i | _i | _i | _i | _| j� | j� d S )NF)�debug� domain_class�
feature_class�
domain_object�feature_object�
loadObject�loadActiveParams)r r
r
r r s zContext.__init__c C s
|| _ d S )N)r )r r r
r
r �_debug' s zContext._debugc C s< t jdtjjtj�� �� x\tjtjjtj��D ]D}tjjtjjtj�|�}| j |�}|dk r0|| j
dj |j |j
�< q0W t jdt| j
�� dt| j
j� �� �� t jdtjjtj�� �� xXtjtjjtj��D ]@}tjjtjjtj�|�}| j |�}|dk r�|| jdj |j �< q�W t jdt| j�� dt| jj� �� �� dS ) z-Try to parse domain & feature file to classeszLoad feature classes from Nz{}:{}zFind z parameter feature: zLoad domain classes from z{}z parameter domain: )�loggerr �os�path�dirnamer �__file__�listdir�join�_Context__loadObjectImplr �formatZ
__domain__Z __feature__�len�list�keysr r )r � file_path�objr
r
r r * s
&
zContext.loadObjectc C s& t jj|�rdS t|��}|j� }W dQ R X y(tj| j|�jd�}tj d|�sRt
�W n t k
rh dS X t jj |�d j d�d }t jj |�d j d�d }y4tj d|�s�t
�tj d|�s�t
�t
dj||�� W n8 t k
�r
} ztjd|� d |� �� dS d}~X nX td
j|||��}|S dS ) z5Parse file content to checkout domain & feature classN� z^\w+$r �/�.z agent.{}.{}z failed to import feature file z: zagent.{}.{}.{}���)r r �isdir�open�read�re�search� _pattern_�group�match�AssertionError� Exception�splitr r" r �warning�eval) r r& �fZcontent�
class_nameZ folder_name� file_name�eZ_objectr
r
r Z__loadObjectImpl< s(
zContext.__loadObjectImplc C s� xx| j j� D ]j\}}|j� r tjd|� �� y|� | j|< W q tk
rt } z tjd|� d|� �� W Y dd}~X q X q W xx| jj� D ]j\}}|j� r�tjd|� �� y|� | j |< W q� tk
r� } z tjd|� d|� �� W Y dd}~X q�X q�W dS )z1Check backup data and init actived domain/featurezload actived feature: z failed to load actived feature z: Nzload actived domain: z failed to load actived domain )
r �itemsZ isActivedr �infor r5 r7 r r )r � feature_nameZ feature_objr< � domain_nameZ
domain_objr
r
r r [ s ,zContext.loadActiveParamsc C sD dj ||�}| jj|�s dS | jj|�s:| j| � | j|< | j| S )z~A feature handler is wanted, init and return feature object
Raise:
Exception: object init failed
z{}:{}N)r" r � __contains__r )r r@ r? r
r
r �searchFeatureHandlerm s z Context.searchFeatureHandlerc C sN ddl m} | jj|�s&|dj|���| jj|�sD| j| � | j|< | j| S )Nr )�
InitExceptionzcan not find domain:{})Zagent.common.exceptionrC r rA r" r )r r@ rC r
r
r �searchDomainHandler} s zContext.searchDomainHandlerc C sj i }x�t | jj� �D ]�}| j| }y$tjd|� �� |j� |j� W nB tk
r� } z&tjd|� d|� �� |� ||< W Y d d }~X qX tjd|� d�� | jj |� qW t
dd� | j j � D �dd � d
d �}x�|D ]�\}}}y tjd |� �� |j� W nH tk
�rB } z*tjd|� d|� �� d
j
|�||< W Y d d }~X q�X tjd|� d�� | j j |� q�W |S )Nzstart rollback domain z rollback z failed: zrollback domain z done.c S s g | ]\}}|j ||f�qS r
)Z __priority__)�.0r? �feature_handlerr
r
r �
<listcomp>� s z$Context.rollback.<locals>.<listcomp>c S s | d S )Nr r
)�xr
r
r �<lambda>� s z"Context.rollback.<locals>.<lambda>T)�key�reversezstart rollback feature z{}zrollback feature )r$ r r% r r> �rollbackZ
post_rollbackr5 r7 �pop�sortedr r= r" ) r Zrollback_resultr@ Zdomain_handlerr< Zsorted_feature_list�_r? rF r
r
r rL � s6
zContext.rollbackN)r r r r/ �compiler1 r r r r! r rB rD rL r
r
r
r r s
r )
Zloggingr r/ � importlibr Zagentr r Z getLoggerr �objectr r r
r
r
r �<module> s