File: //home/arjun/projects/env/lib/python3.10/site-packages/boto/services/__pycache__/bs.cpython-310.pyc
o
we� � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl Z d dl
Z
d dl m Z G dd� de
�Zed kr?e� Ze�� dS dS )
� )� OptionParser)�
ServiceDef)� Submitter)�ResultProcessorN)�StringIOc @ sj e Zd ZdZddddddd�Zd d
� Zd d � Zd
d� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS ) �BSz*usage: %prog [options] config_file commandz#Clear input queue and output bucketz!Submit local files to the servicezStart the servicez6Report on the status of the service buckets and queuesz$Retrieve output generated by a batchz0List all batches stored in current output_domain)�reset�submit�start�status�retrieve�batchesc C s d | _ t| jd�| _| jjddddd� | jjddd d
d d � | jjd
dd d
dd � | jjddd d
ddd� | jjddd d
ddd� | jjddddd d� | jjd d � | jjd d d d
d!d"d� | jjd#d$� | jjd%d&d'd
d(d)d� | jjd*d+d d
d,d-d� d S ).N)�usagez--help-commands�
store_true�
help_commandsz'provides help on the available commands)�action�dest�helpz-az --access-key�store�stringzyour AWS Access Key)r �typer z-sz --secret-keyzyour AWS Secret Access Keyz-pz--path�pathz3the path to local directory for submit and retrieve)r r r r z-kz --keypair�keypairz.the SSH keypair used with launched instance(s)z-lz--leave�leavez>leave the files (don't retrieve) files during retrieve commandF)r z-nz--num-instances�
num_instancesz"the number of launched instance(s)� )r z-iz
--ignore-dirs�append�ignorez4directories that should be ignored by submit commandz-bz
--batch-id�batchz1batch identifier required by the retrieve command)� service_namer �Usage�parser�
add_option� set_defaults)�self� r% �I/home/arjun/projects/env/lib/python3.10/site-packages/boto/services/bs.py�__init__) s> ��������
�z BS.__init__c C s2 t d� | j�� D ]
}t d|| j| f � q d S )Nz
Commands:z %s %s)�print�Commands�keys)r$ �keyr% r% r& �print_command_helpA s �zBS.print_command_helpc C s� | j �d�}|r)td� d}|�� }|r#|d7 }|�|� |�� }|std| � | j �d�}| j �d�}|r\|rA|j|jkrAd S td� d}|D ]
}|d7 }|�� qItd | � d S d S )
N� input_queuezclearing out input queuer r zdeleted %d messages�
output_bucket� input_bucketz'delete generated files in output bucketzdeleted %d keys)�sd�get_objr( �read�delete_message�name�delete)r$ �iq�i�m�ob�ib�kr% r% r& �do_resetF s.
�
�z BS.do_resetc C s� | j js
| j�d� tj�| j j�s | j�d| j j � t| j�}|�| j jd | j j d d d| j j�}t
d|d � t
d|d � d S )N�No path provided�Invalid path (%s)Tz"A total of %d files were submittedr zBatch Identifier: %sr ) �optionsr r! �error�os�existsr r0 � submit_pathr r( )r$ �s�tr% r% r& � do_submit] s
�z BS.do_submitc
C s� | j �d�}| j �dd�}| j �dd�}|s | j�d� t�� }| j �d�s>| j �d� | j �dd|j � | j �dd |j
� t � }| j � |� |�
|g�}|d
}|j|�� | jj| jj||gd �}td | � td
|j � |jD ] } td| j � qrd S )N�ami_id�
instance_typezm1.small�security_group�defaultz3ami_id option is required when starting the service� Credentials�aws_access_key_id�aws_secret_access_keyr )� user_data�key_name� max_countrH �security_groupszStarting AMI: %sz0Reservation %s contains the following instances:z %s)r0 �getr! r@ �boto� connect_ec2� has_section� add_section�setrL rM r �write�get_all_images�run�getvaluer? r r r( �id� instances)
r$ rG rH rI �ec2rD �rs�img�rr7 r% r% r& �do_starth s0 �
�z BS.do_startc C s� | j �d�}|rtd|j|�� f � | j �d�}| j �d�}|rA|r+|j|jkr+d S d}|D ]}|d7 }q/td|j|f � d S d S )Nr- z7The input_queue (%s) contains approximately %s messagesr. r/ r r z'The output_bucket (%s) contains %d keys)r0 r1 r( r\ �countr4 )r$ r6 r9 r: �totalr; r% r% r& � do_status� s
�z BS.do_statusc C sx | j js
| j�d� tj�| j j�s | j�d| j j � | j js&| j�d� t| j j| j�}|j | j j| j j
d� d S )Nr= r> z1batch identifier is required for retrieve command)�get_file) r? r r! r@ rA rB r r r0 � get_resultsr )r$ rD r% r% r& � do_retrieve� s zBS.do_retrievec C sN | j �d�}|r td� |�d�}|D ] }td|j � qd S | j�d� d S )N�
output_domainzAvailable Batches:z['type'='Batch']z %sz&No output_domain specified for service)r0 r1 r( �queryr4 r! r@ )r$ �dr_ �itemr% r% r& �
do_batches� s
�z
BS.do_batchesc C s� | j �� \| _| _| jjr| �� t�d� t| j�dkr#| j � d� | jd | _
t | j
�| _ | jd | _
t| d| j
�rJt| d| j
�}|� d S | j � d| j
� d S )Nr � z$config_file and command are requiredr zdo_%szcommand (%s) not recognized)r! �
parse_argsr? �argsr r, �sys�exit�lenr@ � config_filer r0 �command�hasattr�getattr)r$ �methodr% r% r& �main� s
zBS.mainN)�__name__�
__module__� __qualname__r r) r' r, r< rF rb re rh rm ry r% r% r% r& r s$ �
r �__main__)�optparser �boto.services.servicedefr �boto.services.submitr �boto.services.resultr rS rq rA � boto.compatr �objectr rz �bsry r% r% r% r& �<module> s �