|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/boto/manage/__pycache__/ |
Upload File : |
o
ckF[gU � @ s� d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZ ddlmZ ddlmZ dd lmZ ddlZddlZdd
lmZ dd lmZ dd lm Z m Z g d
�Z G dd� de �Z G dd� de �Z!G dd� de �Z"dS )z)
High-level abstraction of an EC2 server
� N)�IObject)�BotoConfigPath�Config)�Model)�StringProperty�IntegerProperty�BooleanProperty�CalculatedProperty)�propget)�Zone)�KeyPair)�closing)�EC2ResponseError)�six�StringIO)�m1.smallzm1.largez m1.xlarge� c1.mediumz c1.xlargez
m2.2xlargez
m2.4xlargec @ s< e Zd Zddd�Zdd� Zdd� Zdd � Z
dd d
�Zd
S )�Bundler�rootc C s* ddl m} || _|| _|||d�| _d S )Nr �� SSHClient)�uname)�boto.manage.cmdshellr �serverr �
ssh_client)�selfr r r � r �4/usr/lib/python3/dist-packages/boto/manage/server.py�__init__- s zBundler.__init__c C sr t d� | j�� tj�|�\}}d| | _| j�|| j� tj�|�\}}d| | _| j�|| j� t d� d S )Nz5 copying cert and pk over to /mnt directory on serverz/mnt/%s� ...complete!) �printr � open_sftp�os�path�split�remote_key_file�put_file�remote_cert_file)r �key_file� cert_filer# �namer r r � copy_x5093 s
zBundler.copy_x509c C s� d}| j dkr d}|d7 }|d| j| jf 7 }|d| jjj 7 }|d| 7 }|d| 7 }|d 7 }| jjd
ks<| jjd krB|d 7 }|S |d
7 }|S )N� r �sudo zec2-bundle-vol z -c %s -k %s z-u %s z-p %s z-s %d z-d /mnt r r z-r i386z -r x86_64)r r' r% r � _reservation�owner_id�
instance_type)r �prefix�size�ssh_key�commandr r r � bundle_image> s
�zBundler.bundle_imagec C sZ d}| j dkr d}|d7 }|d| 7 }|d| 7 }|d| jjj 7 }|d| jjj 7 }|S ) Nr, r r- zec2-upload-bundle z-m /mnt/%s.manifest.xml z-b %s z-a %s z-s %s )r r �ec2�aws_access_key_id�aws_secret_access_key)r �bucketr1 r3 r4 r r r �
upload_bundleN s
zBundler.upload_bundleNTc C s` t � } |s
| �d�}|s| �d�}|s| �d�}|s | �d�}|s&| �d�}|s-| j�� }| �||� |s8t� }|�dt � |�d� |rK|�d� |�| �
|||�� |�d � |�| � |||�� |�d � |�d
t � |�d � |� � }
t
d � t
|
� | j�|
�} t
d
| d � t
d
| d � t
d� t
d� | jjj|d||f d�| _| jS )NzName of S3 bucketzPrefix for AMI filez Path to RSA private key filez Path to RSA public cert filez Size (in MB) of bundled imagezsudo mv %s /mnt/boto.cfg; z0mv ~/.ssh/authorized_keys /mnt/authorized_keys; z history -c; z; zsudo mv /mnt/boto.cfg %s; z.mv /mnt/authorized_keys ~/.ssh/authorized_keysz3running the following command on the remote server:z %sr � r zregistering image...z%s/%s.manifest.xml)r* �image_location)r �
get_string� get_filename�get_intr �get_ssh_key_filer+ r �writer r5 r: �getvaluer r �runr6 �register_image�image_id) r r9 r1 r( r) r2 r3 �fp�
clear_history�iobjectr4 �tr r r �bundleY sH
zBundler.bundle�r )NNNNNNNT)�__name__�
__module__� __qualname__r r+ r5 r: rJ r r r r r + s
�r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�CommandLineGetterc C s8 g }| j �� D ]}|j�d�dkr|�|j|f� q|S )N�pyamir )r6 �get_all_images�location�find�append)r �my_amis�amir r r � get_ami_list� s �z CommandLineGetter.get_ami_listc C sb |� dd �}t|t�rtj�|�}||d< |s(| j�d�}tj |tjj d�|d< |d �
� | _d S )N�region� region_name)�choices) �get�
isinstance�
basestring�botor6 �
get_region�cls�
find_propertyr
�regions�connect)r �paramsrX �propr r r r_ � s
z CommandLineGetter.get_regionc C �. |� dd �s| j�d�}t� |�|d< d S d S )Nr* �r[ r` ra r
�r rd re r r r �get_name� � �zCommandLineGetter.get_namec C rf )N� descriptionrg rh r r r �get_description� rj z!CommandLineGetter.get_descriptionc C s0 |� dd �stddtd�}t� |�|d< d S d S )Nr0 �
Instance Type�r* � verbose_namerZ )r[ r �
InstanceTypesr
rh r r r �get_instance_type� s ��z#CommandLineGetter.get_instance_typec C s. |� dd �stddd�}t� |�|d< d S d S )N�quantityzNumber of Instances�r* ro )r[ r r
rh r r r � get_quantity� rj z CommandLineGetter.get_quantityc C s4 |� dd �stdd| jjd�}t� |�|d< d S d S )N�zonezEC2 Availability Zonern )r[ r r6 �
get_all_zonesr
rh r r r �get_zone� s ��zCommandLineGetter.get_zonec C s| d}|s<|� dd �}|stddd�}t� |�}z| j�|g�}t|�dkr-d}|d |d< W n ty7 Y nw |rd S d S )NFrV �AMIrs r; Tr )r[ r r
r6 rQ �lenr )r rd �validrV re �rsr r r �
get_ami_id� s
� � �z CommandLineGetter.get_ami_idc C sl |� dd �}t|t�r | j�� }|D ]
}|j|kr |}||d< q|s4tdd| jjd�}t� |�|d< d S d S )N�groupzEC2 Security Grouprn )r[ r\ r] r6 �get_all_security_groupsr* r r
)r rd r} �
group_list�gre r r r � get_group� s
���zCommandLineGetter.get_groupc C sr |� dd �}t|t�r"| j�� }|D ]}|j|kr!|j}|j|d< q|s7tdd| jjd�}t� |�j|d< d S d S )N�keypairz EC2 KeyPairrn )r[ r\ r] r6 �get_all_key_pairsr* r r
)r rd r� �key_list�kre r r r �get_key� s
���zCommandLineGetter.get_keyc C sr || _ | �|� |d �� | _| �|� | �|� | �|� | �|� | �|� | � |� | �
|� | � |� d S )NrX ) r` r_ rc r6 ri rl rq rw rt r| r� r� )r r` rd r r r r[ � s
zCommandLineGetter.getN)rL rM rN rW r_ ri rl rq rt rw r| r� r� r[ r r r r rO � s
rO c s e Zd Zeddd�Zedd�Zedd�Zedd�Zedd�Ze d d
d �Z
e d e dd
�Z
e de dd
�Ze de dd
�Ze de dd
�Ze dedd
�Ze de dd
�Ze de dd
�Ze de dd
�Ze de dd
�Ze de dd
�Ze dedd
�Zg Zg Ze dd� �Z e dUdd ��Z e dVd d ��Z e d d!� �Z dW� fd"d#� Z!d$d%� Z"d&d'� Z#d(d)� Z$d*d+� Z%d,d-� Z&d.d/� Z'd0d1� Z(d2d3� Z)d4d5� Z*d6d7� Z+d8d9� Z,� fd:d;�Z-� fd<d=�Z.d>d?� Z/d@dA� Z0dBdC� Z1dDdE� Z2dFdG� Z3dHdI� Z4dJdK� Z5dLdM� Z6dXdOdP�Z7dYdQdR�Z8dSdT� Z9� Z:S )Z�ServerT�Name)�uniquero � Description)ro zEC2 Region NamezEC2 Instance IDzEC2 Elastic IP AddresszIs This Server ProductionF)ro �defaultzAMI ID)ro �calculated_type�
use_methodzAvailability Zone NamezPublic DNS NamezPrivate DNS NamezSecurity GroupszPrimary Security Group NamezKey Namerm zCurrent StatuszServer Launch TimezConsole Outputc C sj |� d�s
|�d� |�dd|� |�dd|� |� d�s"|�d� |�ddd� |�dd| jjj� d S )N� Credentialsr7 r8 � DB_Server�db_type�SimpleDB�db_name)� has_section� add_section�set�_manager�domainr* )r` �cfgr7 r8 r r r �add_credentials s
zServer.add_credentialsNc K s |rt |d�}|�d�r |�d�D ]}||vr |�d|�||< qt� }|�| |� |�d�}|�� }| �||j|j� |�d�} |�d�}
|�d�} |�d�} |dkr[|� d d
|j
� t � }
|� |
� t
| t�rk| j
} t
|
t�rs|
j
}
| jd |�d d �|
| g|�d
�| |
�� d�}g }d}|�d�}|j}|dur�|�� dkr�|d }td� t�d� |�� dkr�t�d � |�� dks�|�|� td| � |D ]<}| � }||_|dkr�|�d�d nt|�|_
|�d�|_|j
|_ |j |_ |r�|dkr�||_ |� � |�!|� |d 7 }q�|S )aq
Create a new instance based on the specified configuration file or the specified
configuration and the passed in parameters.
If the config_file argument is not None, the configuration is read from there.
Otherwise, the cfg argument is used.
The config file may include other config files with a #import reference. The included
config files must reside in the same directory as the specified file.
The logical_volume argument, if supplied, will be used to get the current physical
volume ID and use that as an override of the value specified in the config file. This
may be useful for debugging purposes when you want to debug with a production config
file but a test Volume.
The dictionary argument may be used to override any EC2 configuration values in the
config file.
)r# �EC2rX rV r� r} ru N�EBS�logical_volume_namer; rr r0 )� min_count� max_count�key_name�security_groupsr0 � placement� user_datar �
elastic_ipzEWaiting for instance to start so we can set its elastic IP address...� �runningz.set the elastic IP of the first instance to %sr* r, rk )"r r� �optionsr[ rO rc r� r7 r8 r� r* r rA r\ r r rC rB � instances�__len__r �time�sleep�update�use_ipr6 �strrk rY �id� instance_idr� �putrT )r` � config_file�logical_volumer� rd �option�getterrX r6 rV �kpr} ru �cfg_fp� reservation�l�ir� r� �instance�sr r r �create
st
�
�
�
z
Server.creater, c
C s� t j�� }|D ]J}|�� }z|�|g�}W n g }Y t|�dkrQ| � }||_||_||_|j|_||_ |d |_
|j
j D ]
} | j |krH| |_
q>|�� | S qd S �Nr; r )r^ r6 rb rc �get_all_reservationsry r* rk rY r� r. r� r� � _instancer� )
r` r� r* rk rb rX r6 r{ r� r� r r r �create_from_instance_id^ s.
��
z Server.create_from_instance_idc C s� g }t j�� }|D ]P}|�� }|�� }|D ]C}|jD ]=}zttj|j d�� t j
� d|j � W q t yW | � }||_|j |_
|j
|_|j |_||_|�� |�|� Y qw qq |S )N)r� z Server for %s already exists)r^ r6 rb rc r� r� �nextr� rS r� �log�info�
StopIterationr* rY r� r. r� rT ) r` �serversrb rX r6 r{ r� r� r� r r r � create_from_current_instancesv s.
���z$Server.create_from_current_instancesc sB t t| �j|fi |�� d | _d | _d | _d | _d | _| �� d S �N) �superr� r � ssh_key_filer6 � _cmdshellr. r� �
_setup_ec2)r r� �kw�� __class__r r r � s zServer.__init__c C s� | j r | jr | jr d S | jrZ| jr\tj �� D ]G}|j| jkrY|�� | _ | j rY| jsYz&| j �
| j g�}t |�dkrN|d j D ]}|j| j krM|d | _|| _q=W q t
yX Y qw qd S d S d S r� )r6 r� r. r� rY r^ rb r* rc r� r� ry r� r )r rX r{ r� r r r r� � s.
� ���zServer._setup_ec2c C s d}| j r| j �� | j j}|S �Nr, )r� r� �state)r �statusr r r �_status� s
zServer._statusc C � d}| j r | j j}|S r� )r� �public_dns_name�r �hostnamer r r � _hostname� � zServer._hostnamec C r� r� )r� �private_dns_namer� r r r �_private_hostname� r� zServer._private_hostnamec C r� r� )r� r0 )r �itr r r �_instance_type� r� zServer._instance_typec C r� r� )r� � launch_time)r �ltr r r � _launch_time� r� zServer._launch_timec C s d}| j r
| j �� }|S r� )r� �get_console_output)r �cor r r �_console_output� s
zServer._console_outputc C s g }| j r | j j}|S r� )r. �groups)r �gnr r r �_groups� r� zServer._groupsc C s" | � � }t|�dkr|d jS dS )Nr; r r, )r� ry r� )r r� r r r �_security_group� s
zServer._security_groupc C � d }| j r | j j}|S r� )r� r� )r ru r r r �_zone� r� z Server._zonec C r� r� )r� r� )r �knr r r � _key_name� r� zServer._key_namec s t t| ��� | �� d S r� )r� r� r� r� �r r� r r r� � s z
Server.putc s | j rtd��tt| ��� d S �Nz Can't delete a production server)�
production�
ValueErrorr� r� �deleter� r� r r r� � s z
Server.deletec C �& | j rtd��| jr| j�� d S d S r� )r� r� r� �stopr� r r r r� � �
�z Server.stopc C r� r� )r� r� r� � terminater� r r r r� � r� zServer.terminatec C s | j r
| j �� d S d S r� )r� �rebootr� r r r r� � s �z
Server.rebootc C s& | j dkrt�d� | j dksd S d S )Nr� r� )r� r� r� r� r r r �wait s
�z Server.waitc C sb | j s.tj�d�}tj�|�r"tj�|d| j �}tj�|�r"|| _ | j s.t� }|� d�| _ | j S )Nz~/.sshz%s.pemzPath to OpenSSH Key file)
r� r" r# �
expanduser�isdir�joinr� �isfiler r> )r �ssh_dir�ssh_filerH r r r r@ s zServer.get_ssh_key_filec C s, | j sddlm} | �� |�| �| _ | j S )Nr )�cmdshell)r� � boto.manager� r@ �start)r r� r r r � get_cmdshell s
zServer.get_cmdshellc C s
d | _ d S r� )r� r� r r r �reset_cmdshell s
zServer.reset_cmdshellc C s< t | �� ��}|�|�}W d � |S 1 sw Y |S r� )r
r� rC )r r4 �cmdr� r r r rC s
��z
Server.runr c C s | � � t| |�S r� )r@ r )r r r r r � get_bundler s
zServer.get_bundlerc C s" ddl m} | �� || ||d�S )Nr r )r �ssh_pwd)r r r@ )r r r r r r r �get_ssh_client# s zServer.get_ssh_clientc C s | � d| �S )Nzapt-get -y install %s)rC )r �pkgr r r �install( s zServer.install)NNN)r, r� rK )r N);rL rM rN r r* rk rY r� r� r r� r r� �ami_idru r� �private_hostname�listr� �security_groupr� r0 r� r� �open�console_output�packages�plugins� classmethodr� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r@ r� r rC r r r �
__classcell__r r r� r r� � sj
P
r� )#�__doc__�boto.ec2r^ �boto.mashups.iobjectr �boto.pyami.configr r �boto.sdb.db.modelr �boto.sdb.db.propertyr r r r r� r
�
boto.ec2.zoner �boto.ec2.keypairr r" r� �
contextlibr
�boto.exceptionr � boto.compatr r rp �objectr rO r� r r r r �<module> s" Ue