o Mf{#@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) DESCRIPTOR full_name)klassprefixnamer a/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/google/cloud/operation/__init__.py_compute_type_urls r NcCsB|durt|}|tvrt||urtdt|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 FromStringvalue)any_pbrr r r _from_anyAs  rc@seZdZdZdZ dZ dZ dZ dZddZ 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)r clientcopycaller_metadata _complete)selfr rrr r r __init__s  zOperation.__init__cKs(||j|fi|}||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_pbcKs8t|t}||j|fi|}||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 ParseDictrrr rr)r operationrrrr r 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 )rGetOperationRequestr r_operations_stub GetOperation)r 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)r r _connection api_requestrr"rr)rr. api_responser r r _get_operation_https zOperation._get_operation_httpcCs|jr|S|S)zChecks the status of the current operation. :rtype: :class:`~google.longrunning.operations_pb2.Operation` :returns: The latest status of the current operation. )rr+r2r%r r r _get_operationszOperation._get_operationcCs\|jrd|_|drt|j|_|d}|dkr |j|_dS|dkr,t|j|_dSdS)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. Tmetadatar errorresponseN)donerHasFieldrr4 WhichOneofr5r6)rr 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&rr3r)rrr r r polls  zOperation.poll)__name__ __module__ __qualname____doc__targetr6r5r4rr classmethodr!r$propertyr&r+r2r3rr;r r r r rUs.      r)N)r?typingrgoogle.longrunningrgoogle.protobufr_GOOGLE_APIS_PREFIXrstrtype__annotations__r rrobjectrr r r r s