File manager - Edit - /opt/gsutil/gslib/__pycache__/command.cpython-39.pyc
Back
a +(WgN� � @ sj d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddlZdd lmZ dd lmZ dd lmZ dd l m Z dd l m Z ddl m Z ddl!m"Z" ddl#m$Z$ ddl%m&Z& ddl%m'Z' ddl(m)Z) ddl(m*Z* ddl(m+Z+ ddl(m,Z, ddl(m-Z- ddl.m/Z/ ddl0m1Z1 ddl2m3Z3 ddl2m4Z4 dd l2m5Z5 dd l2m6Z6 dd l2m7Z7 dd l8m9Z9 dd l8m:Z: dd!l8m;Z; dd"l8m<Z< dd#l8m=Z= dd$l>m?Z@ dd%lAmBZB dd&lAmCZC dd'lAmDZD dd(lAmEZE dd)lFmGZG dd*lFmHZH dd+lFmIZI dd,lJmKZK dd-lJmLZL dd.lMmNZN dd/lMmOZO ddlPZdd0lPmQZQ dd1lPmRZR dd2lPmSZS dd3lPmTZT dd4lPmUZU dd5lPmVZV dd6lPmWZW dd7lPmXZX dd8lPmYZY dd9lZm[Z[ dd:l\m]Z] dd;l^m_Z_ dd<l^m`Z` dd=lambZb dd>lamcZc dd?lamdZd dd@lemfZf ddAlgmhZi zddBljmkZl W n em�y� dZlY n0 dCZndDZodEdF� ZpdGdH� ZqdIdJ� ZrdKdL� Zsd�dMdN�ZtdOdP� Zug ZvdQdR� ZwdSdT� ZxdUdV� ZydWZzdXZ{dYdZ� Z|dCZ}d[d\gd[d]gd[d^gd_d]gd_d^gd`d\gd`d]gd`d^gg da�g db�dcd]gddd]gddd^gded]gded^gdf�Z~dgdh� Zdidj� Z�dkdl� Z�edmg dn��Z�G dodp� dpe$e]�Z�G dqdr� dre��Z�G dsdt� dtedtdu��Z�dvdw� Z�G dxdy� dyej��Z�G dzd{� d{e��Z�G d|d}� d}ej��Z�G d~d� de��Z�d�d�� Z�G d�d�� d�e��Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�dS )�ax Base class for gsutil commands. In addition to base class code, this file contains helpers that depend on base class state (such as GetAndPrintAcl) In general, functions that depend on class state and that are used by multiple commands belong in this file. Functions that don't depend on class state belong in util.py, and non-shared helpers belong in individual subclasses. � )�absolute_import)�print_function)�division)�unicode_literalsN)� namedtuple)� StorageUri)�AccessDeniedException)�ArgumentException)�ServiceException)�CloudApiDelegator)� ApiSelector)�GsutilApiMapFactory)�CommandException)� HelpProvider)�CaptureThreadStatException)�LogPerformanceSummaryParams)�CopyObjectInfo)�CopyObjectsIterator)�NameExpansionIterator)�NameExpansionResult)� SeekAheadNameExpansionIterator)�PluralityCheckableIterator)�SeekAheadThread)�ChildProcessSignalHandler)�GetCaughtSignals)� KillProcess)� MultithreadedMainSignalHandler)�RegisterSignalHandler)� HaveFileUrls)�HaveProviderUrls)�StorageUrlFromString)�UrlsAreForSingleProvider)� UrlsAreMixOfBucketsAndObjects)�storage_v1_messages)� FinalMessage)�MetadataMessage)�PerformanceSummaryMessage)�ProducerThreadMessage)�MainThreadUIQueue)� UIController)�UIThread)�GetFriendlyConfigFilePaths)�!GetMaxConcurrentCompressedUploads)�NO_MAX)�UTF8)� AtomicDict)�$CheckMultiprocessingAvailableAndInit)�multiprocessing_context)�ProcessAndThreadSafeInt)�PutToQueueWithTimeout)�SEEK_AHEAD_JOIN_TIMEOUT)� ShouldProhibitMultiprocessing)�UI_THREAD_JOIN_TIMEOUT)�ZERO_TASKS_TO_DO_ARGUMENT)�RsyncDiffToApply)�GcloudStorageCommandMixin)� GetTermLines)� IS_WINDOWS)�AclTranslation)�GetNonMetadataHeaders)�PRIVATE_DEFAULT_OBJ_ACL)�CreateWildcardIterator)�queue)�Random� i� c C sJ t �| �}|jsFd|_|�t jj� t �� }|�t � d�� |� |� |S )a� Fetches a logger with the given name that resembles 'print' output. Initial Logger Configuration: The logger abides by gsutil -d/-D/-DD/-q options. If none of those options were specified at invocation, the returned logger will display all messages logged with level INFO or above. Log propagation is disabled. If a logger with the specified name has already been created and configured, it is not reconfigured, e.g.: foo = CreateOrGetGsutilLogger('foo') # Creates and configures Logger "foo". foo.setLevel(logging.DEBUG) # Change level from INFO to DEBUG foo = CreateOrGetGsutilLogger('foo') # Does not reset level to INFO. Args: command_name: (str) Command name to create logger for. Returns: A logging.Logger object. Fz %(message)s) �logging� getLogger�handlers� propagate�setLevel�root�level� StreamHandler� setFormatter� Formatter� addHandler)� command_name�logZ log_handler� rP � /opt/gsutil/gslib/command.py�CreateOrGetGsutilLoggert s rR c C s | j �|� d S �N)�logger� exception��cls�erP rP rQ �_DefaultExceptionHandler� s rY c C s< | j s dS |j}|�� r8tj�|j�r8| j�d|� dS dS )NTz Skipping symbolic link %s...F) �exclude_symlinks�expanded_storage_url� IsFileUrl�os�path�islink� object_namerT �info)Zcommand_instance�urlZ exp_src_urlrP rP rQ �_UrlArgChecker� s rc c G s dS )NTrP )� unused_argsrP rP rQ �DummyArgChecker� s re c C s | j ||d�S )N�� thread_state)� SetAclFunc)rW �name_expansion_resultrg rP rP rQ �SetAclFuncWrapper� s rj c C s | j �t|�� d| _dS )zDException handler that maintains state about post-completion status.FN)rT �error�str�everything_set_okayrV rP rP rQ �SetAclExceptionHandler� s rn c C s t rtt dd �rt �� d S )N�atfork)� CryptoRandom�getattrro rP rP rP rQ �_CryptoRandomAtFork� s rr c C s t �t�} t�| � | S rS )r1 �Queue�MAX_QUEUE_SIZE�queues�append�Z new_queuerP rP rQ �_NewMultiprocessingQueue� s rx c C s t � t�} t�| � | S rS )rs rt ru rv rw rP rP rQ �_NewThreadsafeQueue� s ry i�~ i0u c C s t j�ddt�S )N�GSUtil�task_estimation_threshold)�boto�config�getint�!DEFAULT_TASK_ESTIMATION_THRESHOLDrP rP rP rQ �_GetTaskEstimationThreshold� s �r� �acl�ch�get�set�cors�defacl)rC r� �off)rC r� �onrC � versioning�web)ZchaclZgetaclZsetaclZgetcorsZsetcorsZchdefaclZ getdefaclZ setdefaclZdisableloggingZ enableloggingZ getloggingZ getversioningZ setversioningZ getwebcfgZ setwebcfgc C s� t �� ag ag at�� atd�at td�a t td�a t td�a t� � at�� at td�atd�atd�at td�at td�at td�at�� atd�at�t�at� t � �a dS )a� Initializes module-level variables that will be inherited by subprocesses. On Windows, a multiprocessing.Manager object should only be created within an "if __name__ == '__main__':" block. This function must be called, otherwise every command that calls Command.Apply will fail. While multiprocessing variables are initialized at the beginning of gsutil execution, new processes and threads are created only by calls to Command.Apply. When multiple processes and threads are used, the flow of startup/teardown looks like this: 1. __main__: initializes multiprocessing variables, including any necessary Manager processes (here and in gslib.utils.parallelism_framework_util). 2. __main__: Registers signal handlers for terminating signals responsible for cleaning up multiprocessing variables and manager processes upon exit. 3. Command.Apply registers signal handlers for the main process to kill itself after the cleanup handlers registered by __main__ have executed. 4. If worker processes have not been created for the current level of recursive calls, Command.Apply creates those processes. ---- Parallel operations start here, so steps are no longer numbered. ---- - Command.Apply in the main thread starts the ProducerThread. - The Producer thread adds task arguments to the global task queue. - It optionally starts the SeekAheadThread which estimates total work for the Apply call. - Command.Apply in the main thread starts the UIThread, which will consume messages from the global status queue, process them, and display them to the user. - Each worker process creates a thread pool to perform work. - The worker process registers signal handlers to kill itself in response to a terminating signal. - The main thread of the worker process moves items from the global task queue to the process-local task queue. - Worker threads retrieve items from the process-local task queue, perform the work, and post messages to the global status queue. - Worker threads may themselves call Command.Apply. - This creates a new pool of worker subprocesses with the same size as the main pool. This pool is shared amongst all Command.Apply calls at the given recursion depth. - This reuses the global UIThread, global status queue, and global task queue. - This starts a new ProducerThread. - A SeekAheadThread is not started at this level; only one such thread exists at the top level, and it provides estimates for top-level work only. - The ProducerThread runs out of tasks, or the user signals cancellation. - The ProducerThread cancels the SeekAheadThread (if it is running) via an event. - The ProducerThread enqueues special terminating messages on the global task queue and global status queue, signaling the UI Thread to shut down and the main thread to continue operation. - In the termination case, existing processes exit in response to terminating signals from the main process. ---- Parallel operations end here. ---- 5. Further top-level calls to Command.Apply can be made, which will repeat all of the steps made in #4, except that worker processes will be reused. T)�managerN) r1 �Managerr� �consumer_pools� task_queues�Lock�caller_id_lockr2 �caller_id_counterr/ � total_tasks�call_completed_map�global_return_values_map� Condition�need_pool_or_done_cond�worker_checking_level_lock�caller_id_finished_count�new_pool_needed�current_max_recursive_level�shared_vars_map�shared_vars_list_map� thread_stats�dict� class_map� failure_countrs rt �glob_status_queue�BoundedSemaphorer, �!concurrent_compressed_upload_lockrP rP rP rQ �"InitializeMultiprocessingVariables s, I �r� c C s t � t�� tjjj�� dS )z7Should be called by signal handlers prior to shut down.N)�ShutDownGsutilr� �shutdown�gslib�utils�parallelism_framework_util�top_level_managerrP rP rP rQ � TeardownMultiprocessingProcesses� s r� c C s~ t d�at� at�� at� at� at d�a t d�a t � t �a t� at�� at� at� at� ag at� at�� at�t� �adS )aD Initializes module-level variables used when running multi-threaded. When multiprocessing is not available (or on Windows where only 1 process is used), thread-safe analogs to the multiprocessing global variables must be initialized. This function is the thread-safe analog to InitializeMultiprocessingVariables. FN)r2 r� r/ r� � threadingr� r� r� r� r� r� rs rt r� r� r� r� r� r� r� r� r� r� r� r, r� rP rP rP rQ � InitializeThreadingVariables� s& �r� � CommandSpec� rN �usage_synopsis�command_name_aliases�min_args�max_args�supported_sub_args� file_url_ok�provider_url_ok�urls_start_arg�gs_api_support�gs_default_api�supported_private_args�argparse_argumentsc s� e Zd ZdZdZdZdZdZeddde ddddddddf dd ��Z d d � Z e e �Z d d � Zdd� ZdM� fdd� Zdd� Zdd� ZdNdd�Zdd� ZdOdd �Zd d � Zd d � ZdPd!d"�ZdQd#d$�Zd%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z dRd3d4�Z!dSd5d6�Z"d7d8� Z#dTd9d:�Z$d;d<� Z%d=d>� Z&G d?d@� d@e'�Z(de)ddddddfdAdB�Z*dCdD� Z+dEdF� Z,dUdGdH�Z-dIdJ� Z.dKdL� Z/� Z0S )V�Commandz#Base class for all gsutil commands.N) r� r� �iamZkms�labelrC � notificationZ retentionr� r ���� Fc C s8 t | ||p g ||||||| p"tjg| p*tj| | p2g d� S )z2Creates an instance of CommandSpec, with defaults.r� )r� r �XMLr� rP rP rQ �CreateCommandSpec s �zCommand.CreateCommandSpecc C s | j jS rS )� command_specrN ��selfrP rP rQ �_GetDefaultCommandName s z Command._GetDefaultCommandNamec C s | j jS )z�Calculate the index in args of the first URL arg. Returns: Index of the first URL arg (according to the command spec). )r� r� r� rP rP rQ �_CalculateUrlsStartArg$ s z Command._CalculateUrlsStartArgc C sL t �| jd�}|rH|dd� | }| j�d�t�d| j| jd� ��� |S )zBMap deprecated aliases to the corresponding new command, and warn.N� � z�You are using a deprecated alias, "%(used_alias)s", for the "%(command_name)s" command. This will stop working on 9/9/2014. Please use "%(command_name)s" with the appropriate sub-command in the future. See "gsutil help %(command_name)s" for details.)Z used_aliasrN ) � OLD_ALIAS_MAPr� �command_alias_usedrT �warn�join�textwrap�wraprN )r� �argsZnew_command_argsrP rP rQ �_TranslateDeprecatedAliases, s ���� z#Command._TranslateDeprecatedAliasesc sB t � �� || _|| _|| _|| _|| _| | _|| _| | _ || _ || _ d| _ d| _ d| _| | _d| _t| j�| _| r�| D ]} | j�| � qx| jdur�t| j�| _nd| _| jdu r�td| j ��| j�tj� | _ t | j t j � ddd�d�a!| �"|�| _#| �$� | jj%| �&� d�| _t'| j#�| jj(k �s<t'| j#�| jj)k�rD| �*� | j| j+v�rZ| �,� | jj-t.j/gd�}| jj0t.j/d�}t1�2| j ||�| _3d| _4t5| j | j3| jt6t7j8t!�| j| j| j| j| j d � | _9d | _:t;�r�| j:d 7 _:t<j=| _>| j:| j>7 _:| j?�r4| j?D ]*\}}|d k�s$|d k�rd| _ �q4�qt@� jA| _BdS )ag Instantiates a Command. Args: command_runner: CommandRunner (for commands built atop other commands). args: Command-line args (arg0 = actual arg, not command name ala bash). headers: Dictionary containing optional HTTP headers to pass to boto. debug: Debug level to pass in to boto connection (range 0..3). trace_token: Trace token to pass to the API implementation. parallel_operations: Should command operations be executed in parallel? bucket_storage_uri_class: Class to instantiate for cloud StorageUris. Settable for testing/mocking. gsutil_api_class_map_factory: Creates map of cloud storage interfaces. Settable for testing/mocking. logging_filters: Optional list of logging. Filters to apply to this command's logger. command_alias_used: The alias that was actually used when running this command (as opposed to the "official" command name, which will always correspond to the file name). perf_trace_token: Performance measurement trace token to use when making API calls. user_project: Project to be billed for this request. Implementation note: subclasses shouldn't need to define an __init__ method, and instead depend on the shared initialization that happens here. If you do define an __init__ method in a subclass you'll need to explicitly call super().__init__(). But you're encouraged not to do this, because it will make changing the __init__ interface more painful. FNzA"%s" command implementation is missing a command_spec definition.rz �dump_status_messages_file)� quiet_moder� )r� )�gs�s3��debug� http_headers� trace_token�perf_trace_token� user_projectr� zpython z-rz-RT)C�super�__init__�command_runner� unparsed_args�headersr� r� r� �parallel_operationsr� �bucket_storage_uri_class� gsutil_api_class_map_factoryrZ �recursion_requested� all_versionsr� �seek_ahead_gsutil_apirR rN rT � addFilterr= �non_metadata_headersr� r � isEnabledForrC �INFOr� r) r| r} r� � ui_controllerr� r� � ParseSubOpts�_replacer� �lenr� r� �$RaiseWrongNumberOfArgumentsException�&_commands_with_subcommands_and_subopts�CheckArgumentsr� r r� r� r � GetApiMap�gsutil_api_map� project_idr r( �sys�stderr� gsutil_apiZ gsutil_cmdr; r� � GSUTIL_PATHZ gsutil_path�sub_optsr0 � is_available� multiprocessing_is_available)r� r� r� r� r� r� r� r� r� �logging_filtersr� r� r� Z log_filter� support_map� default_map�oZunused_a�� __class__rP rQ r� ? s� * ��� ������� �zCommand.__init__c C sp t | j�| jjk r<| jjdkr"dnd}d| j| jj|f }nd| j| jjf }|d| jj| jf 7 }t|��dS )zCRaises exception for wrong number of arguments supplied to command.r� �sr� z/The %s command requires at least %d argument%s.z,The %s command accepts at most %d arguments.z4 Usage: %s For additional help run: gsutil help %sN)r� r� r� r� rN r� r� r )r� Ztail_str�messagerP rP rQ r� � s � � �z,Command.RaiseWrongNumberOfArgumentsExceptionc C s d| j j| jf }t|��dS )z?Raises exception for specifying an invalid argument to command.zRIncorrect option(s) specified. Usage: %s For additional help run: gsutil help %sN)r� r� rN r )r� r rP rP rQ � RaiseInvalidArgumentException� s �z%Command.RaiseInvalidArgumentExceptionTc C s� |du r| j }n|}z"t�|| jj| jjp,g �\}}W n tjyR | �� Y n0 |rt|| | _| _ |r�| �� n|r�t d��||fS dS )ae Parses sub-opt args. Args: check_args: True to have CheckArguments() called after parsing. args: List of args. If None, self.args will be used. should_update_sub_opts_and_args: True if self.sub_opts and self.args should be updated with the values returned after parsing. Else return a tuple of sub_opts, args returned by getopt.getopt. This is done to allow this method to be called from get_gcloud_storage_args in which case we do not want to update self.sub_opts and self.args. Raises: RaiseInvalidArgumentException: Invalid args specified. NzIRequested to check arguments but sub_opts and args have not been updated.) r� �getoptr� r� r� � GetoptErrorr r� r� � TypeError)r� Z check_argsr� Z should_update_sub_opts_and_argsr� Zparsed_sub_optsZ parsed_argsrP rP rQ r� � s" � zCommand.ParseSubOptsc C s\ | j js,t| j| j jd� �r,td| j ��| j jsXt| j| j jd� �rXtd| j ��dS )a Checks that command line arguments match the command_spec. Any commands in self._commands_with_subcommands_and_subopts are responsible for calling this method after handling initial parsing of their arguments. This prevents commands with sub-commands as well as options from breaking the parsing of getopt. TODO: Provide a function to parse commands and sub-commands more intelligently once we stop allowing the deprecated command versions. Raises: CommandException if the arguments don't match. NzN"%s" command does not support "file://" URLs. Did you mean to use a gs:// URL?z1"%s" command does not support provider-only URLs.) r� r� r r� r� r rN r� r r� rP rP rQ r� s ����zCommand.CheckArgumentsc C s t || j|| j| jd�S )a� Helper to instantiate gslib.WildcardIterator. Args are same as gslib.WildcardIterator interface, but this method fills in most of the values from instance state. Args: url_string: URL string naming wildcard objects to iterate. all_versions: If true, the iterator yields all versions of objects matching the wildcard. If false, yields just the live object version. Returns: WildcardIterator for use by caller. )r� r� rT )r? r� r� rT )r� � url_stringr� rP rP rQ �WildcardIterator$ s �zCommand.WildcardIteratorc C s< | j s6t| j| jt�d�t| j| j| j | j | j d� | _ | j S )aW Helper to instantiate a Cloud API instance for a seek-ahead iterator. This must be separate from the core command.gsutil_api instance for thread-safety, since other iterators typically use that instance and the SeekAheadIterator operates in parallel. Returns: Cloud API instance for use by the seek-ahead iterator. �dummyr� ) r� r r� r� rC rD r� r� r� r� r� r� r� rP rP rQ �GetSeekAheadGsutilApi9 s � z Command.GetSeekAheadGsutilApic C s t d| j ��dS )a Abstract function in base class. Subclasses must implement this. The return value of this function will be used as the exit status of the process, so subclass commands should return an integer exit code (0 for success, a value in [1,255] for failure). z5Command %s is missing its RunCommand() implementationN)r rN r� rP rP rQ � RunCommandV s �zCommand.RunCommandc C s6 g }t tt|��}t|�r(| js(td��|D ]x}|�� r�|�� r�| jrZd|_|� |j � q�| � |j �j dgd�D ]$}t |ddd|jdd�} || | � qpq,|� |j � q,t|�dk�rt| j| j| j| j|| j| j| jp�| j|d � } t| j| j| �� || j| jd �} | j|| || j | d � | j�s2| j�s2td ��dS ) a� Sets the standard or default object ACL depending on self.command_name. Args: acl_func: ACL function to be passed to Apply. acl_excep_handler: ACL exception handler to be passed to Apply. url_strs: URL strings on which to set ACL. object_fields: If present, list of object metadata fields to retrieve; if None, default name expansion iterator fields will be used. Raises: CommandException if an ACL could not be set. z/Cannot operate on a mix of buckets and objects.�*�id�� bucket_fieldsFN)Zsource_storage_urlZis_multi_source_requestZ!is_multi_top_level_source_request�names_containerr[ �expanded_resultr� )r� �continue_on_error�bucket_listing_fields)r� )� fail_on_error�seek_ahead_iterator�'ACLs for some objects could not be set.) �list�mapr r" r� r � IsCloudUrl�IsBucketr` rv r r � IterBucketsr � storage_urlr� r rN r� rT r� r� r r� r r �Applyrm ) r� �acl_func�acl_excep_handler�url_strsZ object_fieldsZmulti_threaded_url_args�urlsrb �blrZname_expansion_for_urlZname_expansion_iteratorr rP rP rQ � ApplyAclFuncg sh �� � � � �zCommand.ApplyAclFuncc C s� |r| j rJ �|}n| j}| j r$dnd}|j}| j�d||� |�|j�tjkrh|jdkrh| � ||� n | � ||� t |j t t�� d�� dS )z�Sets the object ACL for the name_expansion_result provided. Args: name_expansion_result: NameExpansionResult describing the target object. thread_state: If present, use this gsutil Cloud API instance for the set. zdefault object ACL�ACLzSetting %s on %s...r� )� message_timeN)�def_aclr� r[ rT ra �GetApiSelector�schemer r� �_SetAclXmlPassthrough�_SetAclGsutilApir3 � status_queuer% �time)r� ri rg r� Z op_stringrb rP rP rQ rh � s � �zCommand.SetAclFuncc C s� |j }ztz(tj|_ |j| j|| j| j|jd� W n@ typ } z(| j rZd| _ | j � |� n� W Y d}~n d}~0 0 W ||_ n||_ 0 dS )a� Sets the ACL for the URL provided using the XML passthrough functions. This function assumes that self.def_acl, self.canned, and self.continue_on_error are initialized, and that self.acl_arg is either an XML string or a canned ACL string. Args: url: CloudURL to set the ACL on. gsutil_api: gsutil Cloud API to use for the ACL set. Must support XML passthrough functions. )�canned� def_obj_acl�providerFN) � prefer_apir r� �XmlPassThroughSetAcl�acl_argr/ r( r* r r rm rT rk )r� rb r� Zorig_prefer_apirX rP rP rQ r+ � s � z Command._SetAclXmlPassthroughc C s� �zB|� � r�| jr~| jr:|j|jt�� | j|jdgd� q�t � | jtj �}|sX|� t � tj|d�}|j|j||jdgd� n\| jr�|j|jt�� | j|jdgd� n4t � | jtj�}tj|d�}|j|j||jdgd� nf| j�r |j|j|jt�� |j|j| jd� n8t � | jtj �}tj|d�}|j|j|j||j|jd� W nj t�yl } z� W Y d }~nLd }~0 t�y� } z*| j�r�d | _| j�|� n� W Y d }~n d }~0 0 d S ) ad Sets the ACL for the URL provided using the gsutil Cloud API. This function assumes that self.def_acl, self.canned, and self.continue_on_error are initialized, and that self.acl_arg is either a JSON string or a canned ACL string. Args: url: CloudURL to set the ACL on. gsutil_api: gsutil Cloud API to use for the ACL set. r )�canned_def_aclr1 �fields)�defaultObjectAcl)r1 r6 )� canned_aclr1 r6 )r� )r1 � generationr8 )r1 r9 NF)r r( r/ � PatchBucket� bucket_name�apitools_messages�Bucketr4 r* r< � JsonToMessage�ObjectAccessControlrv r> �BucketAccessControl�PatchObjectMetadatar` �Objectr9 r r r rm rT rk ) r� rb r� r0 �bucket_metadataZ bucket_aclZ object_acl�object_metadatarX rP rP rQ r, � s� �� ���� ��� � zCommand._SetAclGsutilApic C s� | j d }| j dd� }t|�s.td| j ��tj�|�rxt�|dt ��}|� � }W d � n1 sf0 Y d| _ n<t j |d | jd| jd�}|�� }||vr�td| ��d | _ d | _|| _| �|||� | js�| js�td ��dS ) z�Sets ACLs on the self.args using the passed-in acl function. Args: acl_func: ACL function to be passed to Apply. acl_excep_handler: ACL exception handler to be passed to Apply. r r� Nz,"%s" command spanning providers not allowed.�rF)r� �validater� zInvalid canned ACL "%s".Tr )r� r! r rN r] r^ �isfile�codecs�openr. �readr/ r| � storage_urir� r� � canned_aclsrm r4 r% r )r� r r! r4 �url_args�frK rL rP rP rQ �SetAclCommandHelper< s2 � &� zCommand.SetAclCommandHelperc C s, ddl m} |r(| j�d�t�d��� dS )z�Warns service account users who have received an AccessDenied error. When one of the metadata-related commands fails due to AccessDenied, user must ensure that they are listed as an Owner in the API console. r )�IS_SERVICE_ACCOUNTr� a� It appears that your service account has been denied access while attempting to perform a metadata operation. If you believe that you should have access to this metadata (i.e., if it is associated with your account), please make sure that your service accounts email address is listed as an Owner in the Permissions tab of the API console. See "gsutil help creds" for further information. N)Z$gcs_oauth2_boto_plugin.oauth2_pluginrP rT �warningr� r� r� )r� rP rP rP rQ �_WarnServiceAccountsg s ��z Command._WarnServiceAccountsc C s� | � |�}t|�}| j�|j�tjkr�|jdkr�z&| jj|| j|jd�}t |� � � W q� t y� } z| � � � W Y d}~q�d}~0 0 nN| j dkr�|jj}|s�| j�d|||� n|jj}|s�| �|� t t�|�� dS )z�Prints the standard or default object ACL depending on self.command_name. Args: url_str: URL string to get ACL for. r� )r0 r1 Nr� a� No default object ACL present for %s. This could occur if the default object ACL is private, in which case objects created in this bucket will be readable only by their creators. It could also mean you do not have OWNER permission on %s and therefore do not have permission to read the default object ACL. It could also mean that %s has Bucket Policy Only enabled and therefore object ACLs and default object ACLs are disabled (see https://cloud.google.com/storage/docs/bucket-policy-only).)�#GetAclCommandBucketListingReferencer r� r) r* r r� �XmlPassThroughGetAclr( �print�to_xmlr rR rN � root_objectr7 rT r� r� �_ListAccessControlsAclr< �JsonFromMessage)r� �url_strr$ rb r� �_rP rP rQ �GetAndPrintAcl s2 �� � zCommand.GetAndPrintAclc C sD |� � r | jj|j|jd�S |�� r<| jj|j|j|jd�S dS dS )at Returns either bucket or object access controls for a storage url. Args: storage_url: StorageUrl object representing the bucket or object. Returns: BucketAccessControls, ObjectAccessControls, or None if storage_url does not represent a cloud bucket or cloud object. Raises: ServiceException if there was an error in the request. )r1 N)r r� �ListBucketAccessControlsr; r* �IsObject�ListObjectAccessControlsr` )r� r rP rP rQ rX � s ��z Command._ListAccessControlsAclc C s� t |�}|�� r*t| �|�jdgd��}n.| jdkr<dg}ndg}t| �|�j|d��}|�� rhtd��|� � r�td|| jf ��t |�d S ) a' Gets a single bucket listing reference for an acl get command. Args: url_str: URL string to get the bucket listing reference for. Returns: BucketListingReference for the URL string. Raises: CommandException if string did not result in exactly one reference. r� )r r� r7 r �No URLs matchedzD%s matched more than one URL, which is not allowed by the %s commandr ) r r^ r r � IterObjectsrN r �IsEmptyr � HasPluralityr )r� rZ � wildcard_urlZplurality_iterr rP rP rQ rS � s0 �� ����z+Command.GetAclCommandBucketListingReferencec C sD | j ||d�}|�� r(td|| jf ��t|�d }t|j�|jfS )aP Gets a single bucket URL based on the command arguments. Args: arg: String argument to get bucket URL for. bucket_fields: Fields to populate for the bucket. Returns: (StorageUrl referring to a single bucket, Bucket metadata). Raises: CommandException if args did not match exactly one bucket. r zD%s matched more than one URL, which is not allowed by the %s commandr )�GetBucketUrlIterFromArgrc r rN r r r rW )r� �argr � plurality_checkable_iteratorr$ rP rP rQ �GetSingleBucketUrlFromArg� s �� z!Command.GetSingleBucketUrlFromArgc C sP t |�}|�� r|�� r&td| j ��t| �|�j|d��}|�� rLtd��|S )a- Gets a single bucket URL based on the command arguments. Args: arg: String argument to iterate over. bucket_fields: Fields to populate for the bucket. Returns: PluralityCheckableIterator over buckets. Raises: CommandException if iterator matched no buckets. z""%s" command must specify a bucketr r` ) r r r^ r rN r r r rb )r� rf r Zarg_urlrg rP rP rQ re � s ��z Command.GetBucketUrlIterFromArgc C s2 t j}|r"|D ]}|| j�� qi t _d t _d S rS )r � provider_pool� connection�close)r� Zconnection_pool�irP rP rQ �_ResetConnectionPool s z Command._ResetConnectionPoolc C s� | j s |rh|s$tj�ddtjjj�}|dk r8td| ��|sRtj�ddtjjj�}|dk rptd| ��nd}d}t � \}}|r�|dkr�td� t � d|d � t � �f ����| jd k}|r�|d kr�|dkr�|r�| j�d � | j�d |� | j�d|� ||fS )a� Determines the values of process_count and thread_count. These values are used for parallel operations. If we're not performing operations in parallel, then ignore existing values and use process_count = thread_count = 1. Args: process_count: A positive integer or None. In the latter case, we read the value from the .boto config file. thread_count: A positive integer or None. In the latter case, we read the value from the .boto config file. parallel_operations_override: Used to override self.parallel_operations. This allows the caller to safely override the top-level flag for a single call. print_macos_warning: Print a warning about parallel processing on MacOS if true. Returns: (process_count, thread_count): The number of processes and threads to use, respectively. rz �parallel_process_countr� z$Invalid parallel_process_count "%d".�parallel_thread_countz#Invalid parallel_thread_count "%d".r� z�It is not possible to set process_count > 1 on %s. Please update your config file(s) (located at %s) and set "parallel_process_count = 1".z, r �macOSa^ If you experience problems with multiprocessing on MacOS, they might be related to https://bugs.python.org/issue33725. You can disable multiprocessing by editing your .boto config or by adding the following flag to your command: `-o "GSUtil:parallel_process_count=1"`. Note that multithreading is still available even if you disable multiprocessing. zprocess count: %dzthread count: %d)r� r| r} r~ r� �commandsZ DEFAULT_PARALLEL_PROCESS_COUNTr Z DEFAULT_PARALLEL_THREAD_COUNTr5 r� r� r� r+ �recursive_apply_levelrT ra r� )r� � process_count� thread_count� parallel_operations_overrideZprint_macos_warning� should_prohibit_multiprocessing�os_name�is_main_threadrP rP rQ �_GetProcessAndThreadCount( sL ���� ��� �z!Command._GetProcessAndThreadCountc C s� t � t�� t�� }W d � n1 s*0 Y t�| �}| jd7 _d|_d|_d|_|t |< dt |< dt |< dt |< g t |< |S )zBSet up the state for a caller id, corresponding to one Apply call.Nr� r� Fr )r� r� � Increment�GetValue�copyrr rT r� r� r� r� r� r� r� )r� � caller_idrW rP rP rQ �_SetUpPerCallerStaten s & z Command._SetUpPerCallerStatec C s� g }t � }t�|� t�� t�� tkr0td��t|�D ]B}t t �}t j | j ||||fd�}d|_|�|� t� |�� q8t||�} t �| � dS )z7Create a new pool of processes that call _ApplyThreads.z,Recursion depth of Apply calls is too great.)�targetr� TN)rx r� rv r� rz r{ �MAX_RECURSIVE_DEPTHr �ranger� r� r1 �Process� _ApplyThreads�daemonrr �start� _ConsumerPool) r� � num_processes� num_threadsr- � processes� task_queuer[ rr �p� consumer_poolrP rP rQ �_CreateNewConsumerPool� s( �� z Command._CreateNewConsumerPoolc @ s e Zd ZdZdZdZdZdS )z Command.ParallelOverrideReasonzAEnum class to describe purpose of overriding parallel operations.�sliceZspeedZperfdiagN)�__name__� __module__� __qualname__�__doc__�SLICEZSPEEDZPERFDIAGrP rP rP rQ �ParallelOverrideReason� s r� c C sb |r,i } |D ] } t | | �| | < t| | d� q | �|||�\}}| jdkoP| jdk}|rbt||d� | op|| dk} |r�| js�|dkr�t| jd� | � � }|r�|t |< |D ]} dt || f< q�| jr�|nd}|| dk�r| j |||||||| | | |d� |�rt � n| �|||||| | � |�rN|D ](} | | t �|| f� }t| | |� �q$| �r^t�|�S dS )a_ Calls _Parallel/SequentialApply based on multiprocessing availability. Args: func: Function to call to process each argument. args_iterator: Iterable collection of arguments to be put into the work queue. exception_handler: Exception handler for WorkerThread class. shared_attrs: List of attributes to manage across sub-processes. arg_checker: Used to determine whether we should process the current argument or simply skip it. Also handles any logging that is specific to a particular type of argument. parallel_operations_override: A string (see ParallelOverrideReason) describing the reason to override self.parallel_operations. This allows the caller to safely override the top-level flag for a single call. process_count: The number of processes to use. If not specified, then the configured default will be used. thread_count: The number of threads per process. If not specified, then the configured default will be used.. should_return_results: If true, then return the results of all successful calls to func in a list. fail_on_error: If true, then raise any exceptions encountered when executing func. This is only applicable in the case of process_count == thread_count == 1. seek_ahead_iterator: If present, a seek-ahead iterator that will provide an approximation of the total number of tasks and bytes that will be iterated by the ProducerThread. Used only if multiple processes and/or threads are used. Returns: Results from spawned threads. r r� )r� r� r� )rT )r ru N)rq �setattrry rr �sequential_caller_idr r� r0 rT r~ r� r� �_ParallelApply�_AggregateThreadStats�_SequentialApplyr� r� )r� �func� args_iterator�exception_handlerZ shared_attrs� arg_checkerru rs rt �should_return_resultsr r Zoriginal_shared_vars_values�namerx r} Zusable_processes_countZ final_valuerP rP rQ r � sp 3� �� � ��� � �z Command.Applyc C sN t j�ddd�dkr(t j�ddd�dksJ| j�dt�d| jj � d � dS ) z2Outputs a suggestion to the user to use gsutil -m.rz rn r r� ro r� z�==> NOTE: You are performing a sequence of gsutil operations that may run significantly faster if you instead use gsutil -m %s ... Please see the -m section under "gsutil help options" for further information about when gsutil -m can be advantageous.N) r| r} r~ rT ra r� �fillr� rN r� rP rP rQ �_MaybeSuggestGsutilDashM6 s � ���z Command._MaybeSuggestGsutilDashMc C s� t dd| j| j| j| jd�}t|�}d} z t|�} W n� tyf } zW Y d} ~ �q@W Y d} ~ n�d} ~ 0 ty� } znt � |r�� nTz|| | � W n8 ty� } z | j � d|t � � � W Y d} ~ n d} ~ 0 0 W Y d} ~ q(W Y d} ~ n d} ~ 0 0 | d7 } | tk�s | t dk�r| �� || | �r(t|| |||||�} |�| | � q(| t }|t� k�r\| �� t| jjtt�� �� |j�|| � | �|� dS )a Performs all function calls sequentially in the current thread. No other threads or processes will be spawned. This degraded functionality is used when the multiprocessing module is not available or the user requests only one thread and one process. NF)r� r� r� r� r �4Caught exception while handling exception for %s: %sr� ) � WorkerThreadr� r� r� r� �iter�next� StopIteration� Exception�_IncrementFailureCountrT r� � traceback� format_exc�#OFFER_GSUTIL_M_SUGGESTION_THRESHOLD�#OFFER_GSUTIL_M_SUGGESTION_FREQUENCYr� �Task� PerformTaskr: r3 r� r- r$ r. �shared_vars_updater�Update�_ProcessSourceUrlTypes)r� r� r� r� r} r� r� r � worker_threadZsequential_call_countr� rX r[ �taskZ#lines_since_suggestion_last_printedrP rP rQ r� B sZ � "�" � �� �zCommand._SequentialApplyc C s | j dk} | | jjkr4| j dkr4t�tt�� d�� ts\| r\tj tj fD ]} t | t dd� qHt s�|dkrvt �t� � n>t� }t �|� t|| j|| j| j| jt| j| j| j| jd� |dk�rVz�| s�t�� | j t � � k�r2t �@ | r�| � ||t� n t j!dd� t �"� t �#� W d � n1 �s(0 Y W | �sTt�$� n| �sRt�$� 0 nx| �s�zft�� | j t%� k�r�t&� t� }t �|� t|| j|| j| j| jt| j| j| j| jd� W t�$� n t�$� 0 t | j }| �r�| �s�d} t't(�(| �|||||||| | | �rtndd� }d}| �r.t)tt*j+t,�}t �l t-| �rPW d � �q�n8|dk�r�| �r�t � � �r�t �!� | � ||t� t �"� t �#� W d � n1 �s�0 Y �q.| �r�t.tt/� |j0t1d � | �2|j3� |j4�r�|j4�|j5�r�| �r�|j5�| �r | �s t.tt6t�� �� dS ) a� Dispatches input arguments across a thread/process pool. Pools are composed of parallel OS processes and/or Python threads, based on options (-m or not) and settings in the user's config file. If only one OS process is requested/available, dispatch requests across threads in the current OS process. In the multi-process case, we will create one pool of worker processes for each level of the tree of recursive calls to Apply. E.g., if A calls Apply(B), and B ultimately calls Apply(C) followed by Apply(D), then we will only create two sets of worker processes - B will execute in the first, and C and D will execute in the second. If C is then changed to call Apply(E) and D is changed to call Apply(F), then we will automatically create a third set of processes (lazily, when needed) that will be used to execute calls to E and F. This might look something like: Pool1 Executes: B / \ Pool2 Executes: C D / \ Pool3 Executes: E F Apply's parallelism is generally broken up into 4 cases: - If process_count == thread_count == 1, then all tasks will be executed by _SequentialApply. - If process_count > 1 and thread_count == 1, then the main thread will create a new pool of processes (if they don't already exist) and each of those processes will execute the tasks in a single thread. - If process_count == 1 and thread_count > 1, then this process will create a new pool of threads to execute the tasks. - If process_count > 1 and thread_count > 1, then the main thread will create a new pool of processes (if they don't already exist) and each of those processes will, upon creation, create a pool of threads to execute the tasks. Args: caller_id: The caller ID unique to this call to command.Apply. See command.Apply for description of other arguments. r r� T)�is_final_handler) r� r� r� r� r- r� r� r� r� )� reset_valueN)r r- ��timeout)7rr r� r� r� �putr& r. r; �signal�SIGINT�SIGTERMr r r� rv rx ry � WorkerPoolrT r� r� r� r� r� r� r� r� �acquirer� r{ r� r� r� �Reset� notify_all�wait�release� _GetCurrentMaxRecursiveLevel�"_IncrementCurrentMaxRecursiveLevel�ProducerThreadr| r* r� r� r� r� r3 r7 r� r6 r� r� �unknown_exception�iterator_exceptionr$ )r� r� r� r� r} r� rs rt r� r r ru rx � signal_numr� Zproducer_threadZ ui_threadrP rP rQ r� � s� : � � � � * � � � ��, zCommand._ParallelApplyc C s* t |t�sdS t|j|j|j|jd� dS )z6Logs the URL type information to analytics collection.N)�is_daisy_chain� has_file_src� has_cloud_src�provider_types)� isinstancer r r� r� r� r� )r� r� rP rP rQ r� t s �z Command._ProcessSourceUrlTypesc C s� |dksJ d��t � t� D ]}t�|t� q | �� || _t| }t�|�}t || j || j | j | j || j| j| j| jd� }d} |jdd�s�t�d� q�|�� } | jtkr�|�| � | d7 } q�|�� q�dS ) a� Assigns the work from the multi-process global task queue. Work is assigned to an individual process for later consumption either by the WorkerThreads or (if thread_count == 1) this thread. Args: thread_count: The number of threads used to perform the work. If 1, then perform all work in this thread. process_count: The number of processes used to perform the work. recursive_apply_level: The depth in the tree of recursive calls to Apply of this thread. status_queue: Multiprocessing/threading queue for progress reporting and performance aggregation. r� zCInvalid state, calling command._ApplyThreads with only one process.� �worker_semaphorer� r� r� r- r� r� r� r� r F)�blockingg{�G�z�?N)rr r r� r rm rr r� r� r� r� rT r� r� r� r� r� r� r� r� r. �sleepr� r� r7 �AddTaskr� ) r� rt rs rr r- Z catch_signalr� r� Z worker_poolZ num_enqueuedr� rP rP rQ r� } s> � � zCommand._ApplyThreads)NNNN)FNT)F)N)N)N)N)T)NN)1r� r� r� r� r� r� rr r� � staticmethodr- r� r� �propertyrN r� r� r� r� r r� r� r r r r% rh r+ r, rO rR r\ rX rS rh re rm ry r~ r� �objectr� rc r r� r� r� r� r� � __classcell__rP rP r rQ r� � s� � � � ) � T J++& � F%� R � k r� c @ s e Zd Zdd� Zdd� ZdS )r� c C s || _ || _d S rS )r� r� )r� r� r� rP rP rQ r� � s z_ConsumerPool.__init__c C s | j D ]}t|j� qd S rS )r� r �pid)r� �processrP rP rQ �ShutDown� s z_ConsumerPool.ShutDownN)r� r� r� r� r� rP rP rP rQ r� � s r� c @ s e Zd ZdZdS )r� a= Task class representing work to be completed. Args: func: The function to be executed. args: The arguments to func. caller_id: The globally-unique caller ID corresponding to the Apply call. exception_handler: The exception handler to use if the call to func fails. should_return_results: True iff the results of this function should be returned from the Apply call. arg_checker: Used to determine whether we should process the current argument or simply skip it. Also handles any logging that is specific to a particular type of argument. fail_on_error: If true, then raise any exceptions encountered when executing func. This is only applicable in the case of process_count == thread_count == 1. N)r� r� r� r� rP rP rP rQ r� � s r� zUfunc args caller_id exception_handler should_return_results arg_checker fail_on_errorc C s t | |t�S )a� Initializes and runs the seek-ahead thread. We defer starting this thread until it is needed, since it is only useful when the ProducerThread iterates more results than it can store on the global task queue. Args: seek_ahead_iterator: Iterator that yields SeekAheadResults. seek_ahead_thread_cancel_event: threading.Event for signaling the seek-ahead thread to terminate. Returns: The thread object for the initialized thread. )r r� )r � seek_ahead_thread_cancel_eventrP rP rQ �_StartSeekAheadThread� s �r� c s* e Zd ZdZd� fdd� Zdd� Z� ZS )r� z<Thread used to enqueue work for other processes and threads.Nc sv t t| ��� || _|| _|| _|| _|| _|| _|| _ || _ | | _ t � | _ d| _d| _d| _| | _| | _| �� dS )a Initializes the producer thread. Args: cls: Instance of Command for which this ProducerThread was created. args_iterator: Iterable collection of arguments to be put into the work queue. caller_id: Globally-unique caller ID corresponding to this call to Apply. func: The function to be called on each element of args_iterator. task_queue: The queue into which tasks will be put, to later be consumed by Command._ApplyThreads. should_return_results: True iff the results for this call to command.Apply were requested. exception_handler: The exception handler to use when errors are encountered during calls to func. arg_checker: Used to determine whether we should process the current argument or simply skip it. Also handles any logging that is specific to a particular type of argument. fail_on_error: If true, then raise any exceptions encountered when executing func. This is only applicable in the case of process_count == thread_count == 1. seek_ahead_iterator: If present, a seek-ahead iterator that will provide an approximation of the total number of tasks and bytes that will be iterated by the ProducerThread. status_queue: status_queue to inform task_queue estimation. Only valid when calling from the main thread, else None. Even if this is the main thread, the status_queue will only properly work if args is a collection of NameExpansionResults, which is the type that gives us initial information about files to be processed. Otherwise, nothing will be added to the queue. TN)r� r� r� r� rW r� r} r� r� r� r� r �_SharedVariablesUpdater�shared_variables_updaterr� r� r� r r- r� ) r� rW r� r} r� r� r� r� r� r r r- r rP rQ r� s" *zProducerThread.__init__c C s. d}d }d }d }d }d }d}d }�zX�zpd} t | j�| _zt| j�}W n� tyx } zW Y d } ~ �q�W Y d } ~ n�d } ~ 0 t�y } z�t� | jr�| | _� nnz| �| j | � W n< ty� } z$| j j � d| j t �� � W Y d } ~ n d } ~ 0 0 | j�| j| j � W Y d } ~ q8W Y d } ~ n d } ~ 0 0 | �| j |�r8|d7 }| j�r�|d �s�t|t��snt|t��snt|t��r�t| jt|| t�� �� t|t��s�t|t��r�|j�r�t � |j�} d| v �r�| t | d �7 } n"t|t��r�|j �r�| t |j �7 } |�s^|d u �rt � }|dk�rd}nH||k�r^| j!�rZt"�#� }t$| j!|�}t%j&�'dd d ��rZ|j(t)d � d}|}t*| j || j| j| j+| j| j�}|r8| j,�-|� q8W n4 t�y� } z| j�s�| | _.W Y d } ~ n d } ~ 0 0 W |t/| j< |�s�t*d t0| jd d d d �}| j,�-|� |d u�r |�1� |j(t)d � | j�rft|t��sJt|t��sJt|t��rft| jt|| t�� dd �� t2| jt3�'| j�� n�|t/| j< |�s�t*d t0| jd d d d �}| j,�-|� |d u�r�|�1� |j(t)d � | j�rt|t��s�t|t��s�t|t��rt| jt|| t�� dd �� t2| jt3�'| j�� 0 d S ) Nr Fr� r� �d �sizeTrz Ztask_estimation_forcer� )�finished)4r� r� r� r� r� r� r r� r� rW rT r� r� r� r� r� r� r} r� r- r� r r r8 r3 r' r. r �json�loads�intZ copy_sizer� r r� �Eventr� r| r} r� r� r4 r� r� r� r� r� r� r7 r� � _NotifyIfDoner� ) r� Z num_tasksZcur_taskZ last_taskr{ Zseek_ahead_threadr� Z seek_ahead_thread_consideredr� � total_sizerX r[ Zjson_expanded_resultrP rP rQ �runI s " �" ��� � � � � ������ �� � ������ �zProducerThread.run)NN)r� r� r� r� r� r� r� rP rP r rQ r� s �<r� c @ s" e Zd ZdZddd�Zdd� ZdS ) r� z3Pool of worker threads to which tasks can be added.Nr c C s� |d u |d u ksJ �| | _ | | _| | _| | _|p4t� | _g | _t|�D ]@} t| j||||||| j | j| j| jd� }| j� |� |� � qFd S )Nr� ) r� r� r� r� ry r� �threadsr� r� rv r� )r� rt rT r� r� r� r� r� r- r� r� r� r� r[ r� rP rP rQ r� � s. � zWorkerPool.__init__c C s | j �|� dS )zCAdds a task to the task queue; used only in the multi-process case.N)r� r� )r� r� rP rP rQ r� � s zWorkerPool.AddTask) NNNNr NNNNN)r� r� r� r� r� r� rP rP rP rQ r� � s � .r� c sJ e Zd ZdZd� fdd� Zedd� �Zedd � �Zd d � Zd d � Z � Z S )r� a8 Thread where all the work will be performed. This makes the function calls for Apply and takes care of all error handling, return value propagation, and shared_vars. Note that this thread is NOT started upon instantiation because the function- calling logic is also used in the single-threaded case. Nr c s� t t| ��� t�� | _t�� | _|| _|| _ d| _ i | _ t � | _ || _| | _| | _| | _d| _|r�|r�t|||||| j| j| j| jd� | _dS )ar Initializes the worker thread. Args: task_queue: The thread-safe queue from which this thread should obtain its work. logger: Logger to use for this thread. worker_semaphore: threading.BoundedSemaphore to be released each time a task is completed, or None for single-threaded execution. bucket_storage_uri_class: Class to instantiate for cloud StorageUris. Settable for testing/mocking. gsutil_api_map: Map of providers and API selector tuples to api classes which can be used to communicate with those providers. Used for the instantiating CloudApiDelegator class. debug: debug level for the CloudApiDelegator class. status_queue: Queue for reporting status updates. user_project: Project to be billed for this request. TN)r� r� r� r� r� )r� r� r� r] �getpidr� r. � init_timer� r� r� �cached_classesr� r� r� r� r� r� �thread_gsutil_apir ) r� r� rT r� r� r� r� r- r� r� r� r� r rP rQ r� s2 �zWorkerThread.__init__c C sX t �| j| jf�du r,t| j�t | j| jf< t | j| jf }|�� |t | j| jf< dS )zEUpdate the thread_stats AtomicDict before task_queue.get() is called.N)r� r� r� �ident� _ThreadStatr� �StartBlockedTime�r� � thread_statrP rP rQ �_StartBlockedTimeG s z WorkerThread._StartBlockedTimec C s, t | j| jf }|�� |t | j| jf< dS )zDUpdate the thread_stats AtomicDict after task_queue.get() is called.N)r� r� r� �EndBlockedTimer� rP rP rQ �_EndBlockedTimeS s z WorkerThread._EndBlockedTimec C s, |j }z�z0|j||j| jd�}|jr6tj||gg d� W n~ ty� } zft� |j rX� nJz|� ||� W n8 ty� } z |j � d|t �� � W Y d}~n d}~0 0 W Y d}~n d}~0 0 W | jr�| j�� | j�||� t�|d�}t||� n8| j�r| j�� | j�||� t�|d�}t||� 0 dS )z�Makes the function call for a task. Args: task: The Task to perform. cls: The instance of a class which gives context to the functions called by the Task's function. E.g., see SetAclFuncWrapper. rf )� default_valuer� Nr� )r} r� r� r� r� r� rz r� r� r r� rT r� r� r� r� r� r� r� r� r� )r� r� rW r} �resultsrX r[ �num_donerP rP rQ r� Z s8 � �2 � zWorkerThread.PerformTaskc C st | � � | j�� }| �� |jtkr&q |j}| j�|d �}|sbt�t | �}t |j �|_ || j|< | � ||� q d S rS )r� r� r� r� r� r7 r} r� r| r� rR rN rT r� )r� r� r} rW rP rP rQ r� � s zWorkerThread.run) NNNr NNNNN) r� r� r� r� r� r r� r� r� r� r� rP rP r rQ r� � s" �; &r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) r� z1Stores thread idle and execution time statistics.c C s d| _ || _t�� | _d| _d S )Nr )�total_idle_time�end_block_timer. �start_block_time�total_execution_time)r� r� rP rP rQ r� � s z_ThreadStat.__init__c C s( t � � | _| j| j }| j|7 _d S rS )r. r� r� r� )r� Z exec_timerP rP rQ r� � s z _ThreadStat.StartBlockedTimec C s( t � � | _| j| j }| j|7 _d S rS )r. r� r� r� )r� Z idle_timerP rP rQ r� � s z_ThreadStat.EndBlockedTimec C s: | j | jkr"| j|| j 7 _n| j|| j 7 _dS )z)Decide final stats upon Apply completion.N)r� r� r� r� )r� �end_timerP rP rQ � AggregateStat� s z_ThreadStat.AggregateStatN)r� r� r� r� r� r� r� r� rP rP rP rQ r� � s r� c C sL t � � } d }}t�� D ]"}|�| � ||j7 }||j7 }qt||d� dS )z�At the end of the top-level Apply call, aggregate the thread stats dict. This should only be called in the main process and thread because it logs to the MetricsCollector. r )�thread_idle_time�thread_execution_timeN)r. r� �valuesr� r� r� r )�cur_timer� r� r� rP rP rQ r� � s �r� c @ s e Zd ZdZdd� Zdd� ZdS )r� a� Used to update shared variable for a class in the global map. Note that each thread will have its own instance of the calling class for context, and it will also have its own instance of a _SharedVariablesUpdater. This is used in the following way: 1. Before any tasks are performed, each thread will get a copy of the calling class, and the globally-consistent value of this shared variable will be initialized to whatever it was before the call to Apply began. 2. After each time a thread performs a task, it will look at the current values of the shared variables in its instance of the calling class. 2.A. For each such variable, it computes the delta of this variable between the last known value for this class (which is stored in a dict local to this class) and the current value of the variable in the class. 2.B. Using this delta, we update the last known value locally as well as the globally-consistent value shared across all classes (the globally consistent value is simply increased by the computed delta). c C s i | _ d S rS )�last_shared_var_valuesr� rP rP rQ r� � s z _SharedVariablesUpdater.__init__c C s\ t �|d�}|rX|D ]B}||f}| j�|d�}t||�| }|| | j|< t�||� qdS )z.Update any shared variables with their deltas.Nr )r� r� r rq r� rz )r� r} rW Z shared_varsr� �key� last_value�deltarP rP rQ r� � s z _SharedVariablesUpdater.UpdateN)r� r� r� r� r� r� rP rP rP rQ r� � s r� c C sP t | }||krL|dkrLt� dt| < t�� W d � n1 sB0 Y dS )a� Notify any threads waiting for results that something has finished. Each waiting thread will then need to check the call_completed_map to see if its work is done. Note that num_done could be calculated here, but it is passed in as an optimization so that we have one less call to a globally-locked data structure. Args: caller_id: The caller_id of the function whose progress we're checking. num_done: The number of tasks currently completed for that caller_id. r TN)r� r� r� r� )r} r� Z num_to_dorP rP rQ r� � s r� c C sT t D ] } z | �� W q Y q0 qtD ] }|�� q(z t�� W n Y n0 dS )zEShut down all processes in consumer pools in preparation for exiting.N)ru Zcancel_join_threadr� r� r� )�qr� rP rP rQ r� s r� c C s t �� S rS )r� r{ rP rP rP rQ r� ! s r� c C s t �� d S rS )r� rz rP rP rP rQ r� & s r� c C s t �� d S rS )r� rz rP rP rP rQ r� + s r� c C s t �� d S rS )r� � DecrementrP rP rP rQ �DecrementFailureCount0 s r c C s t �� S )zAReturns the number of failures processed during calls to Apply().)r� r{ rP rP rP rQ �GetFailureCount5 s r c C s t �� dS )zEResets the failure_count variable to 0 - useful if error is expected.N)r� r� rP rP rP rQ �ResetFailureCount; s r )N)�r� � __future__r r r r rH � collectionsr r| r r� rC r] r� r� r� r� r. r� r| �boto.storage_urir r� �gslib.cloud_apir r r �gslib.cloud_api_delegatorr �gslib.cs_api_mapr r �gslib.exceptionr Zgslib.help_providerr Z gslib.metricsr r Zgslib.name_expansionr r r r r Z"gslib.plurality_checkable_iteratorr Zgslib.seek_ahead_threadr �gslib.sig_handlingr r r r r �gslib.storage_urlr r r r! r" �"gslib.third_party.storage_apitoolsr# r< �gslib.thread_messager$ r% r& r' Zgslib.ui_controllerr( r) r* Zgslib.utils.boto_utilr+ r, �gslib.utils.constantsr- r. �&gslib.utils.parallelism_framework_utilr/ r0 r1 r2 r3 r4 r5 r6 r7 Zgslib.utils.rsync_utilr8 Zgslib.utils.shim_utilr9 Zgslib.utils.system_utilr: r; � gslib.utils.translation_helperr< r= r> Zgslib.wildcard_iteratorr? � six.movesr@ rs ZCryptorA rp � ImportErrorr� r� rR rY rc re rj rn ru rr rx ry rt r r� r� r� r� r� r� r� r� r� r� r� r� �Threadr� r� r� r� r� r� r� r� r� r� r� r r r rP rP rP rQ �<module> s: % � %� g �� <6 '.
| ver. 1.4 |
Github
|
.
| PHP 8.2.28 | Generation time: 0.01 |
proxy
|
phpinfo
|
Settings