ruạṛ
a ���fn< � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dl m Z d dl m Z d dl mZmZmZ d dlmZ dej_G dd � d �ZdS ) � N)�datetime)� get_instance_principal_signer)�unicode_escape)�get_region_info�get_certificate_bundle�requests_no_retry_session)� HTTPError� c @ s� e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZd d d d d �Zdddd�ZdZdZdd� Zdd� Zd&dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Z d$d%� Z dS )'�OsmsIm�< i� r �2 NFz7/var/lib/oracle-cloud-agent/plugins/osms/.osms_im_ready� )�POSTzS/20210330/monitoredInstances/{monitoredInstanceId}/actions/activateMonitoringPlugin)�GETz2/20210330/monitoredInstances/{monitoredInstanceId})r zU/20210330/monitoredInstances/{monitoredInstanceId}/actions/deactivateMonitoringPlugin)r z&/20210330/workRequests/{workRequestId})�activate� getinstance� deactivate� workrequest)�monitoredInstanceId� workRequestId� compartmentId� c C s� t j�d�| _ddd�| _|r(|| jd< |d u rHt�d�}tjtjd� || _ |d u rbt j � |�}|| _ | j d | j d< | j d | j d < | j d | _t| j �}| jd d jf i |��| _t| j || jd �| _d S )Nz//etc/oracle-cloud-agent/plugins/osms/config.ymlzapplication/json)�acceptz content-type�oca� osmsx_ctl_im)�level�idr r �canonicalRegionName�osms_im�endpoint_format�agent)�oci_agent_utils�config� from_file�_config�_request_headers�logging� getLogger� basicConfig�WARN�_logger�instance_metadata�InstanceMetadata�_instance_metadata� _resources�_regionr �format�_endpoint_baser �_cert)�self� oca_versionr+ �logger� region_info� r7 �4/sparta/input/plugins/osms/agent/shared/im_common.py�__init__3 s& � zOsmsIm.__init__c C s� t �� }td| jd �| j }|dk r4| j| j }nt|| j| j �}|�� | j| k}|s�| j � d� | j| j |� d�d d� t �| j�� d�d d� |d �� |S )Nr � r zkactivate #{}, work-request #{}: not ready to run at {}, last attempt made at {}, enforcing delay of {} minsz%Y/%m/%d %H:%M:%S,%f���r )r �now�pow�_activate_calls_made�_delay_unit_in_sec� _max_delay�min� timestamp�_last_call_timestampr* �debugr0 �_work_request_calls_made�strftime� fromtimestamp)r3 r< �delay�readyr7 r7 r8 �is_ready_to_runN s" ��zOsmsIm.is_ready_to_runc C s� d }|d u r| j }| j| d }tj�| j| j| d jf i |���}|dvrf| j�d�||�� dS �zJ�z�t � � }d|kr�| j d7 _ d| _ d|kr�| j d7 _ |� � | _t� ��|}t| j| j| j| j| j�}| j�d�|||�� t||�� �|d | j| j|| jd �}|d u�r(|d v �rD|jd k�rD| �|j � d �� n�d|k�r�| � |�\} } | dk�r�| dfW d � W W |�r�|� � S nJd |k�r�| � |�\} } | dk�r�| dfW d � W W |�r�|� � S | �!||t � � | �"� | j||�\} } | | fW d � W W |�r&|� � S W d � n1 �s>0 Y W nV t#�y� } z<d�|| �} | j�| � | dfW Y d } ~ W |�r�|� � S d } ~ 0 0 W |�r�|� � n|�r�|� � 0 d S )Nr r: )r r z/Invalid http method '{}' passed for endpoint {})NzInvalid http methodr r z$About to execute '{}' action - {} {})�url�data�headers�verify�auth�timeout)r r �� zopc-work-request-idr z/Failed to call action '{}' due to exception: {})$r. �_actions�urllib�parse�urljoinr1 r0 r* �errorr r< r> rE rB rC r r r- r2 r/ r$ rD �getattr�lowerr% �request_timeout_seconds� status_code�_set_work_request_idrM �get�_check_work_request�close�_check_instance_status� _log_response� total_seconds� Exception) r3 �action� resources�response� httpmethod�endpoint�request_start_time�session�instance_principal_signer�cli_msg� call_status�msg�exr7 r7 r8 � call_actionf s� & � � � ���& � �zOsmsIm.call_actionc C sN |d u r4| j �d|� d�|||�}| j �|� |S z�| j �d�|jrT|j�dd�nd|�� d�||jrx|j�dd�nd|jr�|jnd|jt|�d d� �}| j �|� | j �d �|j�dd�t|j ��� |� � |d fW S t �yH d |k�r |jd k�r | � d �\}} || f Y S d�||||j�}| j �|� |df Y S 0 d S )NzRequest headers: {}z3Failed to call action '{}' {} {}. Response is none!z'opc-request-id: {}; request-headers: {}�opc-request-id� <missing>r zWCalled action '{}' with opc-request-id: {}; response-headers: {}; status: {}; took: {}sr; z opc-request-id: {}; text: {}r r i� r z,Failed to call action '{}' {} {}, status: {}r: ) r* rD r0 rM r\ rZ �str�infor �text�raise_for_statusr ro ) r3 rc re �request_duration�request_headersrf rg rm rk rl r7 r7 r8 r` � sH �� � � zOsmsIm._log_responsec C s | j d S �Nr �r. )r3 r7 r7 r8 �get_work_request_id� s zOsmsIm.get_work_request_idc C s || j d< d S rx ry )r3 �work_request_idr7 r7 r8 r[ � s zOsmsIm._set_work_request_idc C s0 |j | _d | _|j dk�rt�|j�}|�d�}|| _| j�d� |j �dd�| � � |�� d|kr�| j�d� |t |j��� | � � \}}| j�|� d| _| �d � ||fS |d v r�d� |t |j��}d| _| �d � | j�|� |d fS d� |t |j��}| j�|� |dfS nd � | � � |j �}|d fS d S ) N�� �statusz6opc-request-id: {} opc-work-request-id: {} - status {}rp rq � SUCCEEDEDzim activation {}: {}r )�FAILED�CANCELEDr: z3opc-work-request-id: {} - unexpected status code {})rZ � _work_request_last_httpstatus�_work_request_last_status�json�loadsrt r\ r* rD r0 rM rz r �_create_flag_filers rE r[ rV )r3 re � response_jsonr} rk rl r7 r7 r8 r] � s> � �zOsmsIm._check_work_requestc C s� |j dkr�t�|j�}|�d�}| j�d�|j�dd�|�� d|krr| � � \}}d�||�}| j� |� ||fS d�|t |j��}| j�|� |d fS n d �|j �}| j�|� |d fS d S ) Nr| �lifecycleStatez8opc-request-id: {} - monitoredInstance lifecycleState {}rp rq �ACTIVEz0{} - monitoredInstance lifecycleState already {}z'monitoredInstance lifecycleState {}: {}r: zLCould not check monitoredInstance lifecycleState - unexpected status code {}) rZ r� r� rt r\ r* rD r0 rM r� rs r )r3 re r� �staterk rl r7 r7 r8 r_ � s( � � z OsmsIm._check_instance_statusc C sL | j | jkrHd�| j| jd | j| j�}| j�|� d| _ | �d � |S d S )NzjReached max re-tries of {} for opc-work-request-id: {}; last http_status: {}; last work-request-status: {}r r ) rE �_work_request_max_callsr0 r. r� r� r* �warningr[ )r3 �warn_msgr7 r7 r8 �*reset_work_request_if_max_attempts_reached s � z1OsmsIm.reset_work_request_if_max_attempts_reachedc C s4 t j�| j�r0| js*d| _d}| j�|� | jS dS )NTzim already activatedF)�os�path�exists�_osms_im_ready_flag_file� _is_activatedr* rs )r3 rk r7 r7 r8 � has_flag_file s zOsmsIm.has_flag_filec C s� z`t t� | jtjtjB d�d��.}|�d� d| _d}|dfW d � W S 1 sT0 Y W nJ ty� } z2d�| jt |��}| j � |� |dfW Y d }~S d }~0 0 d S )Ni� zw+r Tz$im activation completed successfullyzEim activation completed successfully, but failed to create file {} {}r: ) �openr� r� �O_RDWR�O_CREAT�truncater� �IOErrorr0 �reprr* rV )r3 �frk rV r7 r7 r8 r� $ s ,� zOsmsIm._create_flag_file)N) �__name__� __module__� __qualname__r? r@ r> rE r� r� r� rC r� r� r/ r1 rR r. r2 rY r9 rJ ro r` rz r[ r] r_ r� r� r� r7 r7 r7 r8 r sD �� ='$ r )r& r� r� rS r �requests�oci_agent_utils.configr! �!oci_agent_utils.instance_metadata�"oci_agent_utils.instance_principalr �"oci_agent_utils.string_conversionsr �oci_agent_utils.utilsr r r �requests.exceptionsr �adapters�DEFAULT_RETRIESr r7 r7 r7 r8 �<module> s
cải xoăn