diff --git a/google/cloud/firestore_admin_v1/services/firestore_admin/async_client.py b/google/cloud/firestore_admin_v1/services/firestore_admin/async_client.py index 38e6406eb5..027c26590b 100644 --- a/google/cloud/firestore_admin_v1/services/firestore_admin/async_client.py +++ b/google/cloud/firestore_admin_v1/services/firestore_admin/async_client.py @@ -28,8 +28,8 @@ from google.auth import credentials # type: ignore from google.oauth2 import service_account # type: ignore -from google.api_core import operation as ga_operation # type: ignore -from google.api_core import operation_async # type: ignore +from google.api_core import operation as ga_operation +from google.api_core import operation_async from google.cloud.firestore_admin_v1.services.firestore_admin import pagers from google.cloud.firestore_admin_v1.types import field from google.cloud.firestore_admin_v1.types import field as gfa_field @@ -273,9 +273,9 @@ async def list_indexes( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -361,9 +361,9 @@ async def get_index( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -436,9 +436,9 @@ async def delete_index( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -519,9 +519,9 @@ async def get_field( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -712,9 +712,9 @@ async def list_fields( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, diff --git a/google/cloud/firestore_admin_v1/services/firestore_admin/client.py b/google/cloud/firestore_admin_v1/services/firestore_admin/client.py index f721cee47d..a2ee377754 100644 --- a/google/cloud/firestore_admin_v1/services/firestore_admin/client.py +++ b/google/cloud/firestore_admin_v1/services/firestore_admin/client.py @@ -19,10 +19,10 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Callable, Dict, Sequence, Tuple, Type, Union import pkg_resources -from google.api_core import client_options as client_options_lib # type: ignore +import google.api_core.client_options as ClientOptions # type: ignore from google.api_core import exceptions # type: ignore from google.api_core import gapic_v1 # type: ignore from google.api_core import retry as retries # type: ignore @@ -32,8 +32,8 @@ from google.auth.exceptions import MutualTLSChannelError # type: ignore from google.oauth2 import service_account # type: ignore -from google.api_core import operation as ga_operation # type: ignore -from google.api_core import operation_async # type: ignore +from google.api_core import operation as ga_operation +from google.api_core import operation_async from google.cloud.firestore_admin_v1.services.firestore_admin import pagers from google.cloud.firestore_admin_v1.types import field from google.cloud.firestore_admin_v1.types import field as gfa_field @@ -175,9 +175,9 @@ def parse_index_path(path: str) -> Dict[str, str]: def __init__( self, *, - credentials: Optional[credentials.Credentials] = None, - transport: Union[str, FirestoreAdminTransport, None] = None, - client_options: Optional[client_options_lib.ClientOptions] = None, + credentials: credentials.Credentials = None, + transport: Union[str, FirestoreAdminTransport] = None, + client_options: ClientOptions = None, client_info: gapic_v1.client_info.ClientInfo = DEFAULT_CLIENT_INFO, ) -> None: """Instantiate the firestore admin client. @@ -191,8 +191,8 @@ def __init__( transport (Union[str, ~.FirestoreAdminTransport]): The transport to use. If set to None, a transport is chosen automatically. - client_options (client_options_lib.ClientOptions): Custom options for the - client. It won't take effect if a ``transport`` instance is provided. + client_options (ClientOptions): Custom options for the client. It + won't take effect if a ``transport`` instance is provided. (1) The ``api_endpoint`` property can be used to override the default endpoint provided by the client. GOOGLE_API_USE_MTLS_ENDPOINT environment variable can also be used to override the endpoint: @@ -218,9 +218,9 @@ def __init__( creation failed for any reason. """ if isinstance(client_options, dict): - client_options = client_options_lib.from_dict(client_options) + client_options = ClientOptions.from_dict(client_options) if client_options is None: - client_options = client_options_lib.ClientOptions() + client_options = ClientOptions.ClientOptions() # Create SSL credentials for mutual TLS if needed. use_client_cert = bool( @@ -377,7 +377,7 @@ def create_index( response = rpc(request, retry=retry, timeout=timeout, metadata=metadata,) # Wrap the response in an operation future. - response = ga_operation.from_gapic( + response = operation.from_gapic( response, self._transport.operations_client, gfa_index.Index, @@ -779,7 +779,7 @@ def update_field( response = rpc(request, retry=retry, timeout=timeout, metadata=metadata,) # Wrap the response in an operation future. - response = ga_operation.from_gapic( + response = operation.from_gapic( response, self._transport.operations_client, gfa_field.Field, @@ -963,7 +963,7 @@ def export_documents( response = rpc(request, retry=retry, timeout=timeout, metadata=metadata,) # Wrap the response in an operation future. - response = ga_operation.from_gapic( + response = operation.from_gapic( response, self._transport.operations_client, gfa_operation.ExportDocumentsResponse, @@ -1065,7 +1065,7 @@ def import_documents( response = rpc(request, retry=retry, timeout=timeout, metadata=metadata,) # Wrap the response in an operation future. - response = ga_operation.from_gapic( + response = operation.from_gapic( response, self._transport.operations_client, empty.Empty, diff --git a/google/cloud/firestore_admin_v1/services/firestore_admin/transports/base.py b/google/cloud/firestore_admin_v1/services/firestore_admin/transports/base.py index ac4c4475f5..fe0dbaed78 100644 --- a/google/cloud/firestore_admin_v1/services/firestore_admin/transports/base.py +++ b/google/cloud/firestore_admin_v1/services/firestore_admin/transports/base.py @@ -124,9 +124,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -139,9 +139,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -154,9 +154,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -169,9 +169,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -187,9 +187,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, diff --git a/google/cloud/firestore_v1/services/firestore/async_client.py b/google/cloud/firestore_v1/services/firestore/async_client.py index d775a877cf..dd2efb1d01 100644 --- a/google/cloud/firestore_v1/services/firestore/async_client.py +++ b/google/cloud/firestore_v1/services/firestore/async_client.py @@ -154,9 +154,9 @@ async def get_document( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -218,9 +218,9 @@ async def list_documents( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -396,9 +396,9 @@ async def delete_document( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -458,9 +458,9 @@ def batch_get_documents( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=300.0, @@ -539,9 +539,9 @@ async def begin_transaction( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -710,9 +710,9 @@ async def rollback( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -770,9 +770,9 @@ def run_query( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=300.0, @@ -947,9 +947,9 @@ def listen( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=86400.0, @@ -1031,9 +1031,9 @@ async def list_collection_ids( maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, diff --git a/google/cloud/firestore_v1/services/firestore/client.py b/google/cloud/firestore_v1/services/firestore/client.py index 527ba3c6ad..e6fd7913d3 100644 --- a/google/cloud/firestore_v1/services/firestore/client.py +++ b/google/cloud/firestore_v1/services/firestore/client.py @@ -19,20 +19,10 @@ from distutils import util import os import re -from typing import ( - Callable, - Dict, - Optional, - Iterable, - Iterator, - Sequence, - Tuple, - Type, - Union, -) +from typing import Callable, Dict, Iterable, Iterator, Sequence, Tuple, Type, Union import pkg_resources -from google.api_core import client_options as client_options_lib # type: ignore +import google.api_core.client_options as ClientOptions # type: ignore from google.api_core import exceptions # type: ignore from google.api_core import gapic_v1 # type: ignore from google.api_core import retry as retries # type: ignore @@ -156,9 +146,9 @@ def from_service_account_file(cls, filename: str, *args, **kwargs): def __init__( self, *, - credentials: Optional[credentials.Credentials] = None, - transport: Union[str, FirestoreTransport, None] = None, - client_options: Optional[client_options_lib.ClientOptions] = None, + credentials: credentials.Credentials = None, + transport: Union[str, FirestoreTransport] = None, + client_options: ClientOptions = None, client_info: gapic_v1.client_info.ClientInfo = DEFAULT_CLIENT_INFO, ) -> None: """Instantiate the firestore client. @@ -172,8 +162,8 @@ def __init__( transport (Union[str, ~.FirestoreTransport]): The transport to use. If set to None, a transport is chosen automatically. - client_options (client_options_lib.ClientOptions): Custom options for the - client. It won't take effect if a ``transport`` instance is provided. + client_options (ClientOptions): Custom options for the client. It + won't take effect if a ``transport`` instance is provided. (1) The ``api_endpoint`` property can be used to override the default endpoint provided by the client. GOOGLE_API_USE_MTLS_ENDPOINT environment variable can also be used to override the endpoint: @@ -199,9 +189,9 @@ def __init__( creation failed for any reason. """ if isinstance(client_options, dict): - client_options = client_options_lib.from_dict(client_options) + client_options = ClientOptions.from_dict(client_options) if client_options is None: - client_options = client_options_lib.ClientOptions() + client_options = ClientOptions.ClientOptions() # Create SSL credentials for mutual TLS if needed. use_client_cert = bool( diff --git a/google/cloud/firestore_v1/services/firestore/transports/base.py b/google/cloud/firestore_v1/services/firestore/transports/base.py index 12c96dfb31..416bb3f723 100644 --- a/google/cloud/firestore_v1/services/firestore/transports/base.py +++ b/google/cloud/firestore_v1/services/firestore/transports/base.py @@ -117,9 +117,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -132,9 +132,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -158,9 +158,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -173,9 +173,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=300.0, @@ -188,9 +188,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -214,9 +214,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, @@ -229,9 +229,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=300.0, @@ -250,9 +250,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=86400.0, @@ -265,9 +265,9 @@ def _prep_wrapped_messages(self, client_info): maximum=60.0, multiplier=1.3, predicate=retries.if_exception_type( - exceptions.DeadlineExceeded, exceptions.InternalServerError, exceptions.ServiceUnavailable, + exceptions.DeadlineExceeded, ), ), default_timeout=60.0, diff --git a/synth.metadata b/synth.metadata index 61a3eb95b1..0e57949c0d 100644 --- a/synth.metadata +++ b/synth.metadata @@ -4,8 +4,7 @@ "git": { "name": ".", "remote": "https://github.com/googleapis/python-firestore.git", - "sha": "2021f38bb6f016c13bc43d59730c77b57ae5c352" - + "sha": "8e53a2c3aef160c4090241f5260686aee03323d2" } }, { @@ -45,6 +44,7 @@ } ], "generatedFiles": [ + ".coveragerc", ".flake8", ".github/CONTRIBUTING.md", ".github/ISSUE_TEMPLATE/bug_report.md", diff --git a/tests/unit/gapic/firestore_admin_v1/test_firestore_admin.py b/tests/unit/gapic/firestore_admin_v1/test_firestore_admin.py index 6773457e91..866badfa33 100644 --- a/tests/unit/gapic/firestore_admin_v1/test_firestore_admin.py +++ b/tests/unit/gapic/firestore_admin_v1/test_firestore_admin.py @@ -31,7 +31,7 @@ from google.api_core import gapic_v1 from google.api_core import grpc_helpers from google.api_core import grpc_helpers_async -from google.api_core import operation_async # type: ignore +from google.api_core import operation_async from google.api_core import operations_v1 from google.auth import credentials from google.auth.exceptions import MutualTLSChannelError