/
opt
/
gsutil
/
gslib
/
utils
/
__pycache__
/
up file
home
a +(Wg�) � @ s4 d Z ddlmZ ddlmZ ddlmZ ddlZddlm Z ddl m Z ddl m Z d Zd e Zed e �Zd Ze d d� �Ze dd� �Ze dd� �Ze dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Z d4d d �Z d5d d �Z d6d d!�Z d7d"d#�Z d$d%� Z!d&d'� Z"d(d)� Z#d*d+� Z$d,d-� Z%d.d/� Z&d0d1� Z'd2d3� Z(dS )8zEImplementation of Retention Policy configuration command for buckets.� )�absolute_import)�input)�DecimalN)�CommandException)� LazyWrapper)�storage_v1_messagesi�Q � g �v@znThis will PERMANENTLY set the Retention Policy on gs://{} to: {} This setting cannot be reverted! Continue?c C s t �d�S )Nz(?P<number>\d+)y$��re�compile� r r �)/opt/gsutil/gslib/utils/retention_util.py�<lambda>" � r c C s t �d�S )Nz(?P<number>\d+)m$r r r r r r $ r c C s t �d�S )Nz(?P<number>\d+)d$r r r r r r & r c C s t �d�S )Nz(?P<number>\d+)s$r r r r r r ( r c C sz d}|rd| ddf }nd| ddf }t |��� }|s:|S |dvrLtd� q&|d ks\|dkr`d S |d ksp|dkr&d S q&d S )a Prompts user to confirm an action with yes or no response. Args: question: Yes/No question to be used for the prompt. default_response: Default response to the question: True, False Returns: Returns the rough equivalent duration in seconds. � z %s [%s|%s]: �Y�n�y�N)r �yesr �noz+ Please respond with 'yes'/'y' or 'no'/'n'.r Tr FN)r �lower�print)ZquestionZdefault_response�prompt�responser r r �_ConfirmWithUserPrompt+ s r c C s� t | �}d}|t |t kr,d�|t �}nd|t |t krLd�|t �}nD|t |t krld�|t �}n$|tkr�d�| |t �}n d�| �}d�|�S )z�Converts Retention Period to Human readable format. Args: retention_period: Retention duration in seconds (integer value). Returns: Returns a string representing retention duration in human readable format. Nz {} Year(s)z {} Month(s)z {} Day(s)z{} Seconds (~{} Day(s))z {} Second(s)z Duration: {})r �SECONDS_IN_YEAR�format�SECONDS_IN_MONTH�SECONDS_IN_DAY)�retention_periodZperiodZ duration_strr r r �_RetentionPeriodToStringH s � r! c C sV d}| rH| j rH| jrdnd}t| j �}d�| j�d��}d�|||�}n d�|�}|S )z3Converts Retention Policy to Human readable format.r z(LOCKED)z (UNLOCKED)z Effective Time: {}z%a, %d %b %Y %H:%M:%S GMTz Retention Policy {}: {} {}z{} has no Retention Policy.)�retentionPeriod�isLockedr! r � effectiveTime�strftime)�retention_policy� bucket_urlZretention_policy_strZ locked_stringr Zretention_effective_timer r r �RetentionPolicyToStringe s � �� r( c C s t || �}t�| |�}t|d�S )NF)r( � _LOCK_PROMPTr r )r' r& Z lock_promptr r r �ConfirmLockRequestv s r* c C s | j �|� d| _dS )zDException handler that maintains state about post-completion status.FN)�logger�error�everything_set_okay)�cls�er r r �$UpdateObjectMetadataExceptionHandler| s r0 c C s& d}t jdd�}| j||||d� d S )Nz Setting Temporary Hold on %s...T�� temporaryHold�� thread_state��apitools_messages�ObjectZObjectUpdateMetadataFunc�r. �name_expansion_resultr4 Z log_templateZobject_metadata_updater r r �SetTempHoldFuncWrapper� s �r: c C s& d}t jdd�}| j||||d� d S )Nz!Releasing Temporary Hold on %s...Fr1 r3 r5 r8 r r r �ReleaseTempHoldFuncWrapper� s �r; c C s& d}t jdd�}| j||||d� d S )Nz!Setting Event-Based Hold on %s...T��eventBasedHoldr3 r5 r8 r r r �SetEventHoldFuncWrapper� s �r> c C s& d}t jdd�}| j||||d� d S )Nz#Releasing Event-Based Hold on %s...Fr< r3 r5 r8 r r r �ReleaseEventHoldFuncWrapper� s �r? c C s | t S )z�Converts duration specified in days to equivalent seconds. Args: days: Retention duration in number of days. Returns: Returns the equivalent duration in seconds. )r ��daysr r r � DaysToSeconds� s rB c C s | t S )a� Converts duration specified in months to equivalent seconds. GCS bucket lock API uses following duration equivalencies to convert durations specified in terms of months or years to seconds: - A month is considered to be 31 days or 2,678,400 seconds. - A year is considered to be 365.25 days or 31,557,600 seconds. Args: months: Retention duration in number of months. Returns: Returns the rough equivalent duration in seconds. )r ��monthsr r r �MonthsToSeconds� s rE c C s | t S )a� Converts duration specified in years to equivalent seconds. GCS bucket lock API uses following duration equivalencies to convert durations specified in terms of months or years to seconds: - A month is considered to be 31 days or 2,678,400 seconds. - A year is considered to be 365.25 days or 31,557,600 seconds. Args: years: Retention duration in number of years. Returns: Returns the rough equivalent duration in seconds. )r �Zyearsr r r �YearsToSeconds� s rG c C s t � �| �S )aa Test whether the string matches retention in years pattern. Args: years: string to match for retention specified in years format. Returns: Returns a match object if the string matches the retention in years pattern. The match object will contain a 'number' group for the duration in number of years. Otherwise, None is returned. )�_RETENTION_IN_YEARS�matchrF r r r �RetentionInYearsMatch� s rJ c C s t � �| �S )af Test whether the string matches retention in months pattern. Args: months: string to match for retention specified in months format. Returns: Returns a match object if the string matches the retention in months pattern. The match object will contain a 'number' group for the duration in number of months. Otherwise, None is returned. )�_RETENTION_IN_MONTHSrI rC r r r �RetentionInMonthsMatch� s rL c C s t � �| �S )a\ Test whether the string matches retention in days pattern. Args: days: string to match for retention specified in days format. Returns: Returns a match object if the string matches the retention in days pattern. The match object will contain a 'number' group for the duration in number of days. Otherwise, None is returned. )�_RETENTION_IN_DAYSrI r@ r r r �RetentionInDaysMatch� s rN c C s t � �| �S )ak Test whether the string matches retention in seconds pattern. Args: seconds: string to match for retention specified in seconds format. Returns: Returns a match object if the string matches the retention in seconds pattern. The match object will contain a 'number' group for the duration in number of seconds. Otherwise, None is returned. )�_RETENTION_IN_SECONDSrI )�secondsr r r �RetentionInSecondsMatch� s rQ c C s� d}t | �}t| �}t| �}t| �}|r<tt|�d���}nL|rTtt|�d���}n4|rltt|�d���}n |r�t|�d��}nt d��|S )z�Converts a retention period string pattern to equivalent seconds. Args: pattern: a string pattern that represents a retention period. Returns: Returns the retention period in seconds. If the pattern does not match N�numberz�Incorrect retention period specified. Please use one of the following formats to specify the retention period : <number>y, <number>m, <number>d, <number>s.) rJ rL rN rQ rG �int�grouprE rB r )�patternrP Z year_matchZ month_matchZ day_matchZ second_matchr r r �RetentionInSeconds s rV )N)N)N)N))�__doc__� __future__r � six.movesr �decimalr r �gslib.exceptionr �gslib.lazy_wrapperr �"gslib.third_party.storage_apitoolsr r6 r r rS r r) rH rK rM rO r r! r( r* r0 r: r; r> r? rB rE rG rJ rL rN rQ rV r r r r �<module> sB �