CURRENT PATH:
/
opt
/
gsutil
/
gslib
/
__pycache__
/
[ ⬅ KEMBALI ]
|
[ HOME ]
Upload File:
Upload Ke Sini
Dir Baru
File Baru
Editing:
tab_complete.cpython-39.pyc
a +(Wg+ � @ sf d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl Z ddl Z ddl m Z ddl mZ dd l mZ dd l mZ dd lmZ dd lmZ dd lmZ dZdZd�� ZG dd� de�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� dej �Z G dd � d e!�Z"G d d � d e�Z#G d d � d e�Z$G d!d"� d"e�Z%d#d$� Z&d%d&� Z'dS )'zShell tab completion.� )�absolute_import)�print_function)�division)�unicode_literalsN)�CannedACLStrings)�IsFileUrlString)�StorageUrlFromString)� StripOneSlash)� GetTabCompletionCacheFilename)�GetTabCompletionLogFilename)�CreateWildcardIterator� i� z� Tab completion aborted (took >%ss), you may complete the command manually. The timeout can be adjusted in the gsutil configuration file. c @ s$ e Zd ZdZdZdZdZdZdZdS )� CompleterTypeZ cloud_bucketZ cloud_objectZcloud_or_local_objectZ local_objectZlocal_object_or_canned_aclZno_opN) �__name__� __module__� __qualname__� CLOUD_BUCKET� CLOUD_OBJECT�CLOUD_OR_LOCAL_OBJECT� LOCAL_OBJECT�LOCAL_OBJECT_OR_CANNED_ACL�NO_OP� r r �!/opt/gsutil/gslib/tab_complete.pyr / s r c @ s e Zd ZdZdd� Zdd� ZdS )�LocalObjectCompleterz!Completer object for local files.c C s ddl m} |� | _d S )Nr )�FilesCompleter)Zargcomplete.completersr �files_completer)�selfr r r r �__init__; s z LocalObjectCompleter.__init__c K s | j |fi |��S �N)r )r �prefix�kwargsr r r �__call__B s z LocalObjectCompleter.__call__N�r r r �__doc__r r" r r r r r 8 s r c @ s e Zd ZdZdd� Zdd� ZdS )� LocalObjectOrCannedACLCompleterz|Completer object for local files and canned ACLs. Currently, only Google Cloud Storage canned ACL names are supported. c C s t � | _d S r )r �local_object_completer)r r r r r L s z(LocalObjectOrCannedACLCompleter.__init__c s, | j � fi |��}� fdd�tD �}|| S )Nc s g | ]}|� � �r|�qS r �� startswith)�.0�acl�r r r � <listcomp>Q � z<LocalObjectOrCannedACLCompleter.__call__.<locals>.<listcomp>)r& r )r r r! Z local_objects� canned_aclsr r+ r r" O s z(LocalObjectOrCannedACLCompleter.__call__Nr# r r r r r% F s r% c @ s< e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z d S ) �TabCompletionCachez!Cache for tab completion results.c C s || _ || _|| _|| _d S r )r �results� timestamp�partial_results)r r r0 r1 r2 r r r r X s zTabCompletionCache.__init__c C s� z\t | d��>}t�|�� �}|d }|d }|d }|d }W d � n1 sP0 Y W n" ty~ d}g }d}d}Y n0 t||||�S ) z�Instantiates the cache from a file. Args: filename: The file to load. Returns: TabCompletionCache instance with loaded data or an empty cache if the file cannot be loaded �rr r0 r1 �partial-resultsNr F)�open�json�loads�read� Exceptionr/ )�filename�fpZ cache_dictr r0 r1 r2 r r r � LoadFromFile^ s * z TabCompletionCache.LoadFromFilec s� t � � }|| j tkrdS d}� | jkr0| j}n<| jsl� �| j�rl� �d�| j�d�krl� fdd�| jD �}|dur�t � � | _|S dS )z>Returns the cached results for prefix or None if not in cache.N�/c s g | ]}|� � �r|�qS r r' )r) �xr+ r r r, � r- z7TabCompletionCache.GetCachedResults.<locals>.<listcomp>)�timer1 �TAB_COMPLETE_CACHE_TTLr r0 r2 r( �count)r r � current_timer0 r r+ r �GetCachedResultsz s � z#TabCompletionCache.GetCachedResultsc C s || _ || _|| _t�� | _dS )zBUpdates the in-memory cache with the results for the given prefix.N)r r0 r2 r? r1 )r r r0 r2 r r r � UpdateCache� s z TabCompletionCache.UpdateCachec C sl t �| j| j| j| jd��}z8t|d��}|�|� W d � n1 sH0 Y W n tyf Y n0 dS )z'Writes out the cache to the given file.)r r0 r4 r1 �wN) r6 �dumpsr r0 r2 r1 r5 �write�IOError)r r: �json_strr; r r r � WriteToFile� s � , z TabCompletionCache.WriteToFileN) r r r r$ r � staticmethodr< rC rD rJ r r r r r/ U s r/ c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�CloudListingRequestThreadz@Thread that performs a listing request for the given URL string.c s* t t| ��� d| _|| _|| _d| _dS )z�Instantiates Cloud listing request thread. Args: wildcard_url_str: The URL to list. gsutil_api: gsutil Cloud API instance to use. TN)�superrL r �daemon�_wildcard_url_str� _gsutil_apir0 )r Zwildcard_url_str� gsutil_api�� __class__r r r � s z"CloudListingRequestThread.__init__c C s4 t | j| j�jdgd�}dd� t�|t�D �| _d S )N�name)�bucket_listing_fieldsc S s g | ] }t |��qS r )�str)r) �cr r r r, � s z1CloudListingRequestThread.run.<locals>.<listcomp>)r rO rP �IterAll� itertools�islice�_TAB_COMPLETE_MAX_RESULTSr0 )r �itr r r �run� s �� �z CloudListingRequestThread.run)r r r r$ r r] � __classcell__r r rR r rL � s rL c @ s e Zd ZdS )� TimeoutErrorN)r r r r r r r r_ � s r_ c @ s* e Zd ZdZd dd�Zdd� Zdd� Zd S ) �CloudObjectCompleterz Completer object for Cloud URLs.Fc C s || _ || _dS )z�Instantiates completer for Cloud URLs. Args: gsutil_api: gsutil Cloud API instance to use. bucket_only: Whether the completer should only match buckets. N)rP � _bucket_only)r rQ � bucket_onlyr r r r � s z CloudObjectCompleter.__init__c C sL t || j�}|�� |�|� |�� rBddl}|�t| � t� �|j }|S )a< Perform a remote listing request for the given wildcard URL. Args: wildcard_url: The wildcard URL to list. timeout: Time limit for the request. Returns: Cloud resources matching the given wildcard URL. Raises: TimeoutError: If the listing does not finish within the timeout. r N) rL rP �start�join�is_alive� argcomplete�warn�_TIMEOUT_WARNINGr_ r0 )r � wildcard_url�timeoutZrequest_threadrf r0 r r r �_PerformCloudListing� s z)CloudObjectCompleter._PerformCloudListingc K s0 |s d}n t |�rg S |d }t|�}| jr8|�� s8g S tj�ddd�}|dkrTg S t�� }t� t � �}|� |�}d} |d ur�|} d} nfzJ| � ||�} | jr�t | �d kr�t| d �g} t | �tk} |�|| | � W n ty� d } g } Y n0 |�t � � t�� } t | �} | | }td | | || | |f � | S ) Nzgs://�*�GSUtil�tab_completion_timeout� r � z (from cache)� z (request timeout)z:%s results%s in %.2fs, %.2f results/second for prefix: %s )r r ra �IsBucket�boto�config�getintr? r/ r< r rC rk �lenr r[ rD r_ rJ �_WriteTimingLog)r r r! ri �urlrj � start_time�cacheZcached_resultsZtiming_log_entry_typer0 r2 �end_timeZ num_resultsZelapsed_secondsr r r r" � sN ���z CloudObjectCompleter.__call__N)F)r r r r$ r rk r" r r r r r` � s r` c @ s e Zd ZdZdd� Zdd� ZdS )�CloudOrLocalObjectCompleterz�Completer object for Cloud URLs or local files. Invokes the Cloud object completer if the input looks like a Cloud URL and falls back to local file completer otherwise. c C s t |�| _t� | _d S r )r` �cloud_object_completerr r& )r rQ r r r r # s z$CloudOrLocalObjectCompleter.__init__c K s& t |�r| j}n| j}||fi |��S r )r r& r} )r r r! � completerr r r r" ' s z$CloudOrLocalObjectCompleter.__call__Nr# r r r r r| s r| c @ s e Zd ZdZdd� ZdS )� NoOpCompleterz(Completer that always returns 0 results.c K s g S r r )r Z unused_prefix� unused_kwargsr r r r" 2 s zNoOpCompleter.__call__N)r r r r$ r" r r r r r / s r c C sz | t jkrt|�S | t jkr"t� S | t jkr2t� S | t jkrHt|dd�S | t j krZt|�S | t j krjt � S t d| ��dS )a Create a completer instance of the given type. Args: completer_type: The type of completer to create. gsutil_api: gsutil Cloud API instance to use. Returns: A completer instance. Raises: RuntimeError: if completer type is not supported. T)rb zUnknown completer "%s"N) r r r| r r r r% r r` r r r � RuntimeError)Zcompleter_typerQ r r r � MakeCompleter6 s r� c C sJ t j�ddd�rFtt� d��}|�| � W d � n1 s<0 Y dS )zAWrite an entry to the tab completion timing log, if it's enabled.rm �tab_completion_time_logsF�abN)rs rt �getboolr5 r rG )�messager; r r r rw Q s rw )(r$ � __future__r r r r rY r6 � threadingr? rs � boto.gs.aclr �gslib.storage_urlr r r Zgslib.utils.boto_utilr r Zgslib.wildcard_iteratorr r@ r[ �rstriprh �objectr r r% r/ �ThreadrL r9 r_ r` r| r r� rw r r r r �<module> s< RX
SIMPAN
[ BATAL ]