File: //usr/lib/python3/dist-packages/boto/mashups/__pycache__/server.cpython-310.pyc
o
ckF[�6 � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZm Z ddl
m Z ddl m
Z
ddlmZ ddlZG d d
� d
e�ZG d d � d e
�ZdS )
z)
High-level abstraction of an EC2 server
� N)�StringIO)�IObject)�Config�BotoConfigPath)�interactive_shell)�Model)�StringPropertyc @ s e Zd Zdd� Zdd� ZdS )� ServerSetc C s^ g }d}| D ] }zt ||�}t|�rd}|�|� W q |�d � Y q|r-|| _| jS |S )NFT)�getattr�callable�append�map_list�map)�self�name�results� is_callable�server�val� r �5/usr/lib/python3/dist-packages/boto/mashups/server.py� __getattr__& s
zServerSet.__getattr__c G s" g }| j D ] }|�||� � q|S �N)r
r )r �argsr �fnr r r r 6 s
z
ServerSet.mapN)�__name__�
__module__� __qualname__r r r r r r r $ s r c s e Zd Zedd� �Zedd� �Zed[dd��Zd\� fd d
� Ze d d d
�Z
e dd�Z e � Z e dd�Z
e dd�Ze ddd�Ze dd�Ze dd�Ze dd�Ze dd�Ze � Zdd� Zdd � Zeeedd �Zd d � Zeeedd �Zd!d"� Zeeedd#�Zd$d%� Z ee edd&�Z d'd(� Z ee edd)�Z d*d+� Z ee edd,�Z!d-d.� Z"ee"edd/�Z#d0d1� Z$ee$edd2�Z%d3d4� Z&d5d6� Z'ee&e'dd7�Z(d8d9� Z)d:d;� Z*d<d=� Z+d>d?� Z,d@dA� Z- B Cd]dDdE�Z.dFdG� Z/dHdI� Z0dJdK� Z1d\dLdM�Z2dNdO� Z3dPdQ� Z4d^dRdS�Z5d_dUdV�Z6dWdX� Z7dYdZ� Z8� Z9S )`�Serverc C s | j d u r
t�� | _ | j S r )�_ec2�boto� connect_ec2�r r r r �ec2> s
z
Server.ec2c C s& t � }| �� }|D ]}|�|� q |S )zl
Returns a list of Server instances, one for each Server object
persisted in the db
)r �findr )�cls�l�rsr r r r � InventoryD s
zServer.Inventory� c C s$ | � }||_ ||_||_|�� |S r )r � instance_id� description�save)r% r r* r+ �sr r r �RegisterP s zServer.RegisterNc s@ t t| �j|fi |�� d | _d | _d | _d | _d | _d | _d S r ) �superr �__init__� _reservation� _instance� _ssh_client�_pkey�_configr )r �id�kw�� __class__r r r0 Y s
zServer.__init__T�Name)�unique� verbose_namez Instance ID)r< zAMI IDzAvailability ZonezSecurity Group�default)r< r= zKey Namez
Elastic IPz
Instance Type� Descriptionc C s t �r )�AttributeError)r �valuer r r � setReadOnlyn s zServer.setReadOnlyc C sX | j s)| jr)z
| j�| jg�}W n Y d S t|�dkr)|d | _| jjd | _ | j S )Nr )r2 r* r# �get_all_reservations�lenr1 � instances)r r' r r r � getInstanceq s
zServer.getInstancezThe Instance for the serverc C � | j r| j jS d S r )�instance�image_idr" r r r �getAMI � �z
Server.getAMIzThe AMI for the serverc C s | j r | j �� | j jS d S r )rG �update�stater" r r r � getStatus� s
�zServer.getStatuszThe status of the serverc C rF r )rG �public_dns_namer" r r r � getHostname� rJ zServer.getHostnamez!The public DNS name of the serverc C rF r )rG �private_dns_namer" r r r �getPrivateHostname� rJ zServer.getPrivateHostnamez"The private DNS name of the serverc C rF r )rG � launch_timer" r r r �
getLaunchTime� rJ zServer.getLaunchTimez The time the Server was startedc C s | j r| j �� S d S r )rG �get_console_outputr" r r r �getConsoleOutput� s
�zServer.getConsoleOutputz&Retrieve the console output for serverc C rF r )r1 �groupsr" r r r � getGroups� s zServer.getGroupsz5The Security Groups controlling access to this serverc C s2 | j st}d| jj }| �||� t|�| _ | j S �Nz%s.ini)r5 r rG r6 �get_filer )r � remote_file�
local_filer r r � getConfig� s
zServer.getConfigc C s< d| j j }t|�}|�|� |�� | �|t� || _d S rX )rG r6 �open�write�close�put_filer r5 )r �configr[ �fpr r r � setConfig� s
zServer.setConfigz!The instance data for this serverc C s || _ | j �d| j� dS )z&
Set SDB based config
� botoConfigsN)r5 � dump_to_sdbr6 )r ra r r r �
set_config� s zServer.set_configc C s t dd�| _| j�d| j� d S )NF)�do_loadrd )r r5 �
load_from_sdbr6 r" r r r � load_config� s zServer.load_configc C � | j r
| j �� d S d S r )rG �stopr" r r r rk � � �z Server.stopc C s> | � � t�� }|jt| j�gd�d }|jt| j�gd�}| js%| � � | j�
d�sC| j� d� | j� dd|j
� | j� dd|j� | j�
d�sO| j� d� | jjrg| j� dd| jjj� | j� dd | j� t� }| j�|� |�� }|jd
d
| j|| j| j|d �}|jd }|j| _| � � | j r�|� | j| j � d S d S ) N)� image_idsr )�
groupnames� Credentials�aws_access_key_id�aws_secret_access_key�Pyami�server_sdb_domain�server_sdb_name� )� min_count� max_count�key_name�security_groups�
instance_type� placement� user_data) rk r r! �get_all_images�str�ami_id�get_all_security_groups�security_groupr5 ri � has_section� add_section�setrp rq �_manager�domainr r r^ �getvalue�runrx rz �zonerD r6 r* �put�
elastic_ip�associate_address)r r# �amirV �cfg�r�ir r r �start� s@ �
�z Server.startc C rj r )rG �rebootr" r r r r� � rl z
Server.reboot�~/.ssh/known_hosts�rootc C s� dd l }| js
td� d S | jsI|st� }|�d�}|j�|�| _|� � | _| j�
� | j� t j
�|�� | j�|�� � | jj| jj|| jd� | jS )Nr zNo instance yet!zPath to OpenSSH Key file)�username�pkey)�paramikorG �printr3 r � get_filename�RSAKey�from_private_key_filer4 � SSHClient�load_system_host_keys�load_host_keys�os�path�
expanduser�set_missing_host_key_policy�
AutoAddPolicy�connectrN )r �key_file�
host_key_file�unamer� �iobjectr r r �get_ssh_client� s"
�zServer.get_ssh_clientc C � | � � }|�� }|�||� d S r )r� � open_sftp�get)r �
remotepath� localpath�
ssh_client� sftp_clientr r r rY � zServer.get_filec C r� r )r� r� r� )r r� r� r� r� r r r r` r� zServer.put_filec C s | � � }|�� }|�|�S r )r� r� �listdir)r r� r� r� r r r r� s
zServer.listdirc C s | � |�}|�� }t|� d S r )r� � invoke_shellr )r r� r� �channelr r r �shell s
z Server.shellc C sF t d� t d� | �� }|�� }tj�|�\}}d| } | �|| � tj�|�\}}d| }
| �||
� t dt � z|�t� W n Y d} | d|
| f 7 } | d| j j
7 } | d| 7 } | d | 7 } | d
7 } | j j d kss| j d krx| d
7 } n| d7 } t d| � |�
| �} | d �� }
t d|
� t d| d �� � t d� d S )Nzbundling image...z5 copying cert and pk over to /mnt directory on serverz/mnt/%sz deleting %szsudo ec2-bundle-vol z -c %s -k %s z-u %s z-p %s z-s %d z-d /mnt zm1.smallz c1.mediumz-r i386z -r x86_64� %sru � � ...complete!)r� r� r� r� r� �splitr` r �remover1 �owner_idrG rz � exec_command�read)r �prefixr� � cert_file�sizer� r� r� r �remote_key_file�remote_cert_file�command�t�responser r r � bundle_image" s<
zServer.bundle_imagec C s� t d� d}|d| 7 }|d| 7 }|d| jj 7 }|d| jj 7 }t d| � | �� }|�|�}|d �� }t d| � t d|d �� � t d
� d S ) Nzuploading bundle...zec2-upload-bundle z-m /mnt/%s.manifest.xml z-b %s z-a %s z-s %s r� ru r� r� )r� r# rp rq r� r� r� )r �bucketr� r� r� r� r� r r r �
upload_bundleD s
zServer.upload_bundlec C s� t � }|s
|�d�}|s|�d�}|s|�d�}|s |�d�}|s&|�d�}| �||||� | �||� td� | j�d||f �| _ | j S )NzName of S3 bucketzPrefix for AMI filez Path to RSA private key filez Path to RSA public cert filez Size (in MB) of bundled imagezregistering image...z%s/%s.manifest.xml)
r �
get_stringr� �get_intr� r� r� r# �register_imagerH )r r� r� r� r� r� r� r r r � create_imageS s
zServer.create_image�/dev/sdpc C s* t |d�r |j}n|}| jj|| j|d�S )z�
Attach an EBS volume to this server
:param volume: EBS Volume to attach
:type volume: boto.ec2.volume.Volume
:param device: Device to attach to (default to /dev/sdp)
:type device: string
r6 )� volume_idr* �device)�hasattrr6 r# �
attach_volumer* )r �volumer� r� r r r r� e s
zServer.attach_volumec C s( t |d�r |j}n|}| jj|| jd�S )z�
Detach an EBS volume from this server
:param volume: EBS Volume to detach
:type volume: boto.ec2.volume.Volume
r6 )r� r* )r� r6 r# �
detach_volumer* )r r� r� r r r r� u s
zServer.detach_volumec C sj t d| � d| }t d| � | �� }|�|�}|d �� }t d| � t d|d �� � t d� d S )Nzinstalling %s...zyum -y install %sr� ru r� r� )r� r� r� r� )r � package_namer� r� r� r� r r r �install_package� s
zServer.install_package)r) r )Nr� r� )NNNNN)r� ):r r r �propertyr# � classmethodr( r. r0 r r r* �
config_urir r� r� rx r� rz r+ �logrA rE rG rI r� rM �statusrO �hostnamerQ �private_hostnamerS rR rU �console_outputrW rV r\ rc ra rf ri rk r� r� r� rY r` r� r� r� r� r� r� r� r� �
__classcell__r r r8 r r < s�
�������#
�
"
r )�__doc__r �
boto.utils� boto.compatr �boto.mashups.iobjectr �boto.pyami.configr r �boto.mashups.interactiver �boto.sdb.db.modelr �boto.sdb.db.propertyr r� �listr r r r r r �<module> s