File: //lib/python3.6/site-packages/awscli/customizations/emr/__pycache__/hbase.cpython-36.pyc
3
�T_�$ � @ s� d dl mZ d dl mZ d dl mZ d dl mZ d dlmZ G dd� de�ZG dd � d e�ZG d
d � d e�Z G d d
� d
e�Z
dS )� )� constants)�emrutils)�
hbaseutils)�helptext)�Commandc @ sH e Zd ZdZdej Zddejd�ddejd�dej d�gZ
d d
� Z d S ) �RestoreFromHBaseBackupzrestore-from-hbase-backupzRestores HBase from S3. z
cluster-idT)�name�required� help_text�dirzbackup-version)r r
c C sX g }t j|j|j�}tjtjtjtj |d�}|j
|� |j |d�}tj | j
d||� dS )N)�jarr �action_on_failure�args)� JobFlowId�Steps�AddJobFlowStepsr )r Z$build_hbase_restore_from_backup_argsr Zbackup_versionr �
build_stepr �HBASE_JAR_PATHZHBASE_RESTORE_STEP_NAME�CANCEL_AND_WAIT�append�
cluster_id�call_and_display_response�_session)�self� parsed_args�parsed_globals�stepsr � step_config�
parameters� r �/usr/lib/python3.6/hbase.py�_run_main_command" s
z(RestoreFromHBaseBackup._run_main_commandN) �__name__�
__module__� __qualname__�NAMEr � AVAILABLE_ONLY_FOR_AMI_VERSIONS� DESCRIPTION�
CLUSTER_ID�HBASE_BACKUP_DIRZHBASE_BACKUP_VERSION� ARG_TABLEr! r r r r r s r c
@ s� e Zd ZdZdej Zddejd�dddd�ddejd�d dd
d�d dd d�d
dd�dddd�gZ dd� Z
dd� Z dd� Z dd� Z
d S ) �ScheduleHBaseBackupzschedule-hbase-backupz0Adds a step to schedule automated HBase backup. z
cluster-idT)r r r
�typez<<p>Backup type. You can specify 'incremental' or 'full'.</p>r �intervalz <p>The time between backups.</p>�unitz|<p>The time unit for backup's time-interval. You can specify one of the following values: 'minutes', 'hours', or 'days'.</p>z
start-timez]<p>The time of the first backup in ISO format.</p> e.g. 2014-04-21T05:26:10Z. Default is now.)r r
�
consistent�
store_truezp<p>Performs a consistent backup. Pauses all write operations to the HBase cluster during the backup process.</p>)r �actionr
c C sj g }| j |j� | j|j� | j|�}tjtjtj tj
|d�}|j |� |j |d�}tj
| jd||� dS )N)r r r
r )r r r r )� _check_typer, � _check_unitr. �!_build_hbase_schedule_backup_argsr r r r � HBASE_SCHEDULE_BACKUP_STEP_NAMEr r r r r )r r r r r r r r r r r! P s
z%ScheduleHBaseBackup._run_main_commandc C s< |j � }|tjkr8|tjkr8tdtj d tj d ��d S )Nz0aws: error: invalid type. type should be either z or �.)�lowerr �FULL� INCREMENTAL�
ValueError)r r, r r r r2 c s
z ScheduleHBaseBackup._check_typec C sP |j � }|tjkrL|tjkrL|tjkrLtdtj d tj d tj d ��d S )NzFaws: error: invalid unit. unit should be one of the following values: z, z or r6 )r7 r ZMINUTESZHOURSZDAYSr: )r r. r r r r3 k s
z ScheduleHBaseBackup._check_unitc C s� t jt jt jt j|jg}|jj� }|jj� }|j dkrB|j
t j � |t j krZ|j
t j
� n |j
t j� |j
|j� |t j kr�|j
t j� n |j
t j� |j
|� |j
t j� |jd k r�|j
|j� n |j
t j� |S )NT)r �
HBASE_MAIN�HBASE_SCHEDULED_BACKUPZTRUEr) r r, r7 r. r/ r �HBASE_BACKUP_CONSISTENTr8 ZHBASE_FULL_BACKUP_INTERVALZ!HBASE_INCREMENTAL_BACKUP_INTERVALr- Z HBASE_FULL_BACKUP_INTERVAL_UNITZ&HBASE_INCREMENTAL_BACKUP_INTERVAL_UNITZHBASE_BACKUP_STARTTIMEZ
start_timeZNOW)r r r r, r. r r r r4 u s&
z5ScheduleHBaseBackup._build_hbase_schedule_backup_argsN)r" r# r$ r% r r& r' r( r) r* r! r2 r3 r4 r r r r r+ 5 s*
r+ c @ sP e Zd ZdZdej Zddejd�ddejd�ddd d
�gZ d d � Z
d
d� Z dS )�CreateHBaseBackupzcreate-hbase-backupz Creates a HBase backup in S3. z
cluster-idT)r r r
r r/ r0 zp<p>Performs a consistent backup. Pauses all write operations to the HBase cluster during the backup process.</p>)r r1 r
c C sR g }| j |�}tjtjtjtj|d�}|j|� |j|d�}tj | j
d||� dS )N)r r r
r )r r r r ) �_build_hbase_backup_argsr r r r ZHBASE_BACKUP_STEP_NAMEr r r r r )r r r r r r r r r r r! � s
z#CreateHBaseBackup._run_main_commandc C s. t jt jt j|jg}|jdkr*|jt j� |S )NT)r r; Z HBASE_BACKUPr) r r/ r r= )r r r r r r r? � s
z*CreateHBaseBackup._build_hbase_backup_argsN) r" r# r$ r% r r& r' r( r) r* r! r? r r r r r> � s
r> c @ sN e Zd ZdZdej Zddejd�dddd �d
dd d �gZd d
� Z dd� Z
dS )�DisableHBaseBackupszdisable-hbase-backupsz/Add a step to disable automated HBase backups. z
cluster-idT)r r r
�fullr0 zDisables full backup.)r r1 r
� incrementalz Disables incremental backup.c C sP g }| j |�}tjtjtjtj|�}|j|� |j|d�}tj | j
d||� dS )N)r r r r ) �!_build_hbase_disable_backups_argsr r r r r5 r r r r r )r r r r r r r r r r r! � s
z%DisableHBaseBackups._run_main_commandc C st t jt jt jg}|jdkrD|jdkrDdt j d t j d }t|��|jdkrZ|j t j
� |jdkrp|j t j � |S )NFz!Should specify at least one of --z and --r6 T) r r; r< ZFALSErA rB r8 r9 r: r ZHBASE_DISABLE_FULL_BACKUPZ HBASE_DISABLE_INCREMENTAL_BACKUP)r r r Z
error_messager r r rC � s
z5DisableHBaseBackups._build_hbase_disable_backups_argsN) r" r# r$ r% r r& r' r( r* r! rC r r r r r@ � s
r@ N) Zawscli.customizations.emrr r r r Z!awscli.customizations.emr.commandr r r+ r> r@ r r r r �<module> s a+