U $FZh{#@stUdZddlmZddlmZddlmZdZiZee e fe d<efddZ dd d Z d d ZGdddeZd S)z=Wrap long-running operations returned from Google Cloud APIs.)Dict)operations_pb2) json_formatztype.googleapis.com _TYPE_URL_MAPcCs|jj}d||fS)aCompute a type URL for a klass. :type klass: type :param klass: class to be used as a factory for the given type :type prefix: str :param prefix: URL prefix for the type :rtype: str :returns: the URL, prefixed as appropriate z%s/%s)Z DESCRIPTORZ full_name)klassprefixnamer f/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/google/cloud/operation/__init__.py_compute_type_urls r NcCsB|dkrt|}|tkr6t||k r6tdt|f|t|<dS)aRegister a klass as the factory for a given type URL. :type klass: :class:`type` :param klass: class to be used as a factory for the given type :type type_url: str :param type_url: (Optional) URL naming the type. If not provided, infers the URL from the type descriptor. :raises ValueError: if a registration already exists for the URL. Nz Conflict: %s)r r ValueError)rtype_urlr r r register_type,s  rcCst|j}||jS)aConvert an ``Any`` protobuf into the actual class. Uses the type URL to do the conversion. .. note:: This assumes that the type URL is already registered. :type any_pb: :class:`google.protobuf.any_pb2.Any` :param any_pb: An any object to be converted. :rtype: object :returns: The instance (of the correct type) stored in the any instance. )rr Z FromStringvalue)Zany_pbrr r r _from_anyAs rc@sxeZdZdZdZdZdZdZdZddZ e ddZ e dd Z e d d Zd d ZddZddZddZddZdS) OperationaRepresentation of a Google API Long-Running Operation. .. _protobuf: https://github.com/googleapis/googleapis/blob/ 050400df0fdb16f63b63e9dee53819044bffc857/ google/longrunning/operations.proto#L80 .. _service: https://github.com/googleapis/googleapis/blob/ 050400df0fdb16f63b63e9dee53819044bffc857/ google/longrunning/operations.proto#L38 .. _JSON: https://cloud.google.com/speech/reference/rest/ v1beta1/operations#Operation This wraps an operation `protobuf`_ object and attempts to interact with the long-running operations `service`_ (specific to a given API). (Some services also offer a `JSON`_ API that maps the same underlying data type.) :type name: str :param name: The fully-qualified path naming the operation. :type client: :class:`~google.cloud.client.Client` :param client: The client used to poll for the status of the operation. If the operation was created via JSON/HTTP, the client must own a :class:`~google.cloud._http.Connection` to send polling requests. If created via protobuf, the client must have a gRPC stub in the ``_operations_stub`` attribute. :type caller_metadata: dict :param caller_metadata: caller-assigned metadata about the operation NTcKs ||_||_||_d|_dS)NF)rclientcopycaller_metadata _complete)selfrrrr r r __init__s zOperation.__init__cKs$||j|f|}||d|_|S)aDFactory: construct an instance from a protobuf. :type operation_pb: :class:`~google.longrunning.operations_pb2.Operation` :param operation_pb: Protobuf to be parsed. :type client: object: must provide ``_operations_stub`` accessor. :param client: The client used to poll for the status of the operation. :type caller_metadata: dict :param caller_metadata: caller-assigned metadata about the operation :rtype: :class:`Operation` :returns: new instance, with attributes based on the protobuf. T)r _update_state _from_grpc)cls operation_pbrrresultr r r from_pbs zOperation.from_pbcKs4t|t}||j|f|}||d|_|S)aFactory: construct an instance from a dictionary. :type operation: dict :param operation: Operation as a JSON object. :type client: :class:`~google.cloud.client.Client` :param client: The client used to poll for the status of the operation. :type caller_metadata: dict :param caller_metadata: caller-assigned metadata about the operation :rtype: :class:`Operation` :returns: new instance, with attributes based on the protobuf. F)r ParseDictrrrrr)rZ operationrrrrr r r from_dicts  zOperation.from_dictcCs|jS)z|Has the operation already completed? :rtype: bool :returns: True if already completed, else false. )rrr r r completeszOperation.completecCstj|jd}|jj|S)zPolls the status of the current operation. Uses gRPC request to check. :rtype: :class:`~google.longrunning.operations_pb2.Operation` :returns: The latest status of the current operation. )r)rZGetOperationRequestrrZ_operations_stubZ GetOperation)rZ request_pbr r r _get_operation_rpcszOperation._get_operation_rpccCs.d|jf}|jjjd|d}t|tS)zChecks the status of the current operation. Uses HTTP request to check. :rtype: :class:`~google.longrunning.operations_pb2.Operation` :returns: The latest status of the current operation. z operations/%sGET)methodpath)rr _connectionZ api_requestrrrr)rr%Z api_responser r r _get_operation_https zOperation._get_operation_httpcCs|jr|S|SdS)zChecks the status of the current operation. :rtype: :class:`~google.longrunning.operations_pb2.Operation` :returns: The latest status of the current operation. N)rr"r'r r r r _get_operationszOperation._get_operationcCsV|jr d|_|dr"t|j|_|d}|dkr>|j|_n|dkrRt|j|_dS)zUpdate the state of the current object based on operation. :type operation_pb: :class:`~google.longrunning.operations_pb2.Operation` :param operation_pb: Protobuf to be parsed. TmetadatarerrorresponseN)donerZHasFieldrr)Z WhichOneofr*r+)rrZ result_typer r r rs    zOperation._update_statecCs&|jrtd|}|||jS)zCheck if the operation has finished. :rtype: bool :returns: A boolean indicating if the current operation has completed. :raises ValueError: if the operation has already completed. zThe operation has completed.)r!r r(r)rrr r r polls  zOperation.poll)__name__ __module__ __qualname____doc__targetr+r*r)rr classmethodrrpropertyr!r"r'r(rr-r r r r rUs$      r)N)r1typingrZgoogle.longrunningrZgoogle.protobufrZ_GOOGLE_APIS_PREFIXrstrtype__annotations__r rrobjectrr r r r s