File: //lib/python3.6/site-packages/awscli/customizations/codedeploy/__pycache__/systems.cpython-36.pyc
3
�T_� � @ sn d dl Z d dlZd dlZdZG dd� d�ZG dd� de�ZG dd� de�ZG d d
� d
e�ZG d d � d e�ZdS )
� Nzcodedeploy.onpremises.ymlc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �SystemzgOnly Ubuntu Server, Red Hat Enterprise Linux Server and Windows Server operating systems are supported.c C s |j | _ | j jd|jd�| _d S )N�s3)Z region_name)�sessionZ
create_client�regionr )�self�params� r � /usr/lib/python3.6/systems.py�__init__ s zSystem.__init__c C s t d��d S )N�validate_administrator)�NotImplementedError)r r r r r " s z System.validate_administratorc C s t d��d S )N�install)r )r r r r r r
% s zSystem.installc C s t d��d S )N� uninstall)r )r r r r r r ( s zSystem.uninstallN)�__name__�
__module__� __qualname__ZUNSUPPORTED_SYSTEM_MSGr
r r
r r r r r r s
r c @ sD e Zd ZdZdZdjee�ZdZdd� Zdd� Z d d
� Z
d d � Z d
S )�Windowsz C:\ProgramData\Amazon\CodeDeployzconf.onpremises.ymlz{0}\{1}zcodedeploy-agent.msic C s t jjj� std��d S )Nz.You must run this command as an Administrator.)�ctypesZwindllZshell32Z
IsUserAnAdmin� RuntimeError)r r r r r 2 s z Windows.validate_administratorc
C s d|kr|j | _tjdddddgtjtjd�}|j� \}}d}|jd kr^||kr^td
j|���| j j
|j |j d �}t
| jd ��}|j|d
j� � W d Q R X tjdj| j�dddgdd� tjdddddg� tjdddddgtjtjd�}|j� \}}d|k�r
td��d S )N� installerzpowershell.exez-Commandz Stop-Servicez-Name�codedeployagent)�stdout�stderrz;Cannot find any service with service name 'codedeployagent'r z,Failed to stop the AWS CodeDeploy Agent:
{0})�Bucket�Key�wb�Bodyz.\{0}z/quietz/lz".\codedeploy-agent-install-log.txtT)�shellzRestart-Servicez Get-ServiceZRunningz:The AWS CodeDeploy Agent did not start after installation.)r � INSTALLER�
subprocess�Popen�PIPE� communicate�
returncoder �formatr �
get_object�bucket�key�open�write�read�
check_call)r r �process�output�error� not_found�response�fr r r r
8 sF
zWindows.installc C s\ t jdddddgt jt jd�}|j� \}}d}|jdkrB| j� n||krXtd j|���d S )
Nzpowershell.exez-Commandz Stop-Servicez-Namer )r r z;Cannot find any service with service name 'codedeployagent'r z,Failed to stop the AWS CodeDeploy Agent:
{0})r r r! r" r# �
_remove_agentr r$ )r r r, r- r. r/ r r r r o s
zWindows.uninstallc C sJ t jdddddddgt jt jd�}|j� \}}|jd krFtd
j|���d S ) NZwmic�product�wherez name="CodeDeploy Host Agent"Zcallr z/nointeractive)r r r z1Failed to uninstall the AWS CodeDeploy Agent:
{0})r r r! r" r# r r$ )r r, r- r. r r r r2 � s
zWindows._remove_agentN) r r r �
CONFIG_DIR� CONFIG_FILEr$ � CONFIG_PATHr r r
r r2 r r r r r , s 7r c @ sT e Zd ZdZeZdjee�ZdZdd� Z dd� Z
dd � Z d
d � Z d d
� Z
dd� ZdS )�Linuxz/etc/codedeploy-agent/confz{0}/{1}r
c C s t j� dkrtd��d S )Nr z"You must run this command as sudo.)�os�geteuidr )r r r r r � s z Linux.validate_administratorc C s� d|kr|j | _| j|� | j|� | jj|j|jd�}t| jd��}|j |d j
� � W d Q R X t j dddj
| j�g� | jj� }tjj� }|j|d< |j|d < |j|d
< |jd k r�|j|d < t j dj
| j�d g|d
� d S )Nr )r r r r �chmodz+xz./{0}Z
AWS_REGIONZAWS_ACCESS_KEY_IDZAWS_SECRET_ACCESS_KEYZAWS_SESSION_TOKEN�auto)�env)r r �_update_system� _stop_agentr r% r& r' r( r) r* r r+ r$ r Zget_credentialsr9 �environ�copyr Z
access_keyZ
secret_key�token)r r r0 r1 Z credentialsZ environmentr r r r
� s&
z
Linux.installc C s" | j |�}|jdkr | j|� d S )Nr )r? r# r2 )r r r, r r r r � s
zLinux.uninstallc C s t d��d S )NZ
preinstall)r )r r r r r r> � s zLinux._update_systemc C s t d��d S )NZ remove_agent)r )r r r r r r2 � s zLinux._remove_agentc C sL t jdddgt jt jd�}|j� \}}|jdkrH|j|krHtdj|���|S )NZservicezcodedeploy-agent�stop)r r r z,Failed to stop the AWS CodeDeploy Agent:
{0})r r r! r" r# �
not_found_msgr r$ )r r r, r- r. r r r r? � s
zLinux._stop_agentN)r r r r5 �DEFAULT_CONFIG_FILEr6 r$ r7 r r r
r r> r2 r? r r r r r8 � s r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Ubuntuc C s& t jdddg� t jddddg� d S )Nzapt-getz-y�updater
zruby2.0)r r+ )r r r r r r> � s zUbuntu._update_systemc C s t jdddg� d S )NZdpkgz-rzcodedeploy-agent)r r+ )r r r r r r2 � s zUbuntu._remove_agentc C s d|_ tj| |�S )Nz&codedeploy-agent: unrecognized service)rD r8 r? )r r r r r r? � s zUbuntu._stop_agentN)r r r r> r2 r? r r r r rF � s rF c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�RHELc C s t jddddg� d S )N�yumz-yr
Zruby)r r+ )r r r r r r> � s zRHEL._update_systemc C s t jddddg� d S )NrI z-yZerasezcodedeploy-agent)r r+ )r r r r r r2 � s zRHEL._remove_agentc C s d|_ tj| |�S )Nz<Redirecting to /bin/systemctl stop codedeploy-agent.service)rD r8 r? )r r r r r r? � s zRHEL._stop_agentN)r r r r> r2 r? r r r r rH � s rH ) r r9 r rE r r r8 rF rH r r r r �<module> s k>