ruạṛ
a ���f� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZzd dlmZm Z W n" e yn d dl mZm Z Y n0 d dl m Z dZd"dd�Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z dS )#� N)�urlopen�Request)�configz4/var/lib/oracle-cloud-agent/plugins/osms/dbtimestamp�r c C sr |p| }t �d|� tj| tjtjdd�}|�� \}}|j|vrVtd||j||f ��t �d|j||� |�d�S )Nzrun_cmd args: %sT)�stdout�stderr� close_fdsz.Command: %s failed (%s): stderr: %s stdout: %sz.run_cmd returncode: %s, stdout: %s, stderr: %szutf-8) �logging�debug� subprocess�Popen�PIPE� communicate� returncode� Exception�decode)�args�success_return_codeZlogargs�proc� stdoutdata� stderrdata� r �8/usr/share/oracle-cloud-agent/plugins/osms/osms/utils.py�run_cmd s" � ��r c C s* zt g d�� W n ty$ Y dS 0 dS )N)�rpmz-qz alx-utilsFT)r r r r r r �is_alx* s r c C s\ z$t ddgdd�} d| v r W dS W dS tyV } zt�d|� W Y d }~dS d }~0 0 d S ) z� needs-restarting may return 1 on failure. Parsing output is safer than checking return code to determine whether restarting is needed. zneeds-restarting�-r)r � )r zReboot is requiredTFzneeds-restarting failed: %sN)r r r r )�output�errr r r �needs_restarting2 s r c C sJ zt g d���� W S tyD } zt�d| � W Y d } ~ dS d } ~ 0 0 d S )N)�ksplice�kernel�unamer z"failed to get effective kernel: %s� )r �stripr r r )r r r r �get_effective_kernelA s r& c C s� zpt dd��� �� } zdd� | D �}W n$ ttfyJ dd� | D �}Y n0 |d td�d d krld |d< |W S ty� } z t�d |� g W Y d }~S d }~0 0 d S ) Nz /proc/uptime�rc S s g | ]}t t|���qS r )�int�float��.0�ar r r � <listcomp>M � z get_uptime.<locals>.<listcomp>c S s g | ]}|d d� �qS )N���r r* r r r r- O r. r � � ���zget uptime failed: %s) �open�read�split� TypeError� ValueErrorr( r r r )Zuptime�resultr r r r � get_uptimeI s r9 c C sf z.t td�} | �� t�� }t�t||f� W n2 ty` } zt�dt|� W Y d }~n d }~0 0 d S )Nzw+z>Unable to touch the packages last update timestamp file %s: %s) r3 �#PACKAGES_LAST_UPDATE_TIMESTAMP_FILE�close�time�os�utimer r r )Zfile_d�tr r r r �$touch_packages_last_update_timestampY s �r@ c C sh z.t �t�d } tt�� �| dkr*W dS W dS tyb } z t�dt|� W Y d}~dS d}~0 0 dS )z>Return True if package list was updated within 60 seconds � �< TFz>Unable to check the packages last update timestamp file %s: %sN)r= �statr: r( r< r r r )�lastr r r r �$check_packages_last_update_timestampg s �rE c C sT t �� }d|_t jj| ddd�}t �dd�}|�|� |�|� |�t � |�� d S )Nr i P � )�filename�maxBytes� backupCountzK[%(asctime)s %(process)d] %(levelname)s (%(module)s:%(lineno)d) %(message)sz%Y-%m-%d %H:%M:%S) r � getLogger� propagate�handlers�RotatingFileHandler� Formatter� setFormatter� addHandler�setLevel� getLevelName)Zlog_file� log_level�logger�handler� formatterr r r �init_logx s �� rW c C sF zt �| � W n2 ty@ } z|jtjkr,� W Y d }~n d }~0 0 d S )N)r= �unlink�OSError�errno�ENOENT)�filepathr r r r � remove_file� s r] c C s� t j�| �r�t| d��d}|�� }tj||tjd�rltj|tjd�}|� ||�}|� d� |� d� |� |� W d � n1 s�0 Y d S )Nzr+)�flagsr ) r= �path�existsr3 r4 �re�search� MULTILINE�compile�sub�seek�truncate�write)r\ Z old_entryZ new_entry�fobjZfcontentZentry_rer r r � update_file� s rj c C sJ t | d��,}t�| tdd�� |�|� W d � n1 s<0 Y d S )N�w�0600rA )r3 r= �chmodr( rh )r\ �contentri r r r � create_file� s ro c C s� t �� } | d }|sntdddid�}t|dd�}|�� dkrJ|�� �� }n$td dd�}|�� dkrn|�� �� }t|t�r�|� � }|s�t d ��|S ) N�ocidz)http://169.254.169.254/opc/v2/instance/id� Authorizationz Bearer Oracle)�headers� )�timeout�� z)http://169.254.169.254/opc/v1/instance/idz Could not retrieve instance OCID) r �initUp2dateConfigr r �getcoder4 r% � isinstance�bytesr r )�cfgZ instance_ocid�request�responser r r �get_instance_ocid� s r} c C s t ddd� d S )Nz,/etc/oracle-cloud-agent/plugins/osms/up2datez serverURL=.*z3serverURL=https://enter.your.server.url.here/XMLRPC)rj r r r r �set_default_server_url� s �r~ c C s t d� d S )Nz-/etc/oracle-cloud-agent/plugins/osms/systemid)r] r r r r �remove_system_id_file� s r )r N) rZ r �logging.handlersr= ra r r< �urllib2r r � ImportErrorZurllib.request�osmsr r: r r r r& r9 r@ rE rW r] rj ro r} r~ r r r r r �<module> s4
cải xoăn