Skip to content

Commit

Permalink
🦉 Updates from OwlBot post-processor
Browse files Browse the repository at this point in the history 
See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
  • Loading branch information
@gcf-owl-bot
gcf-owl-bot[bot] committed Oct 11, 2024
1 parent e7249e0 commit d0340b6
Show file tree
Hide file tree
Showing 289 changed files with 20,196 additions and 136,210 deletions.
24 changes: 4 additions & 20 deletions google/cloud/spanner_admin_database_v1/services/database_admin/async_client.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3602,11 +3602,7 @@ async def list_operations(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.list_operations,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self.transport._wrapped_methods[self._client._transport.list_operations]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -3659,11 +3655,7 @@ async def get_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.get_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self.transport._wrapped_methods[self._client._transport.get_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -3720,11 +3712,7 @@ async def delete_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.delete_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self.transport._wrapped_methods[self._client._transport.delete_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -3777,11 +3765,7 @@ async def cancel_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.cancel_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self.transport._wrapped_methods[self._client._transport.cancel_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down
24 changes: 4 additions & 20 deletions google/cloud/spanner_admin_database_v1/services/database_admin/client.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4091,11 +4091,7 @@ def list_operations(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method.wrap_method(
self._transport.list_operations,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self._transport._wrapped_methods[self._transport.list_operations]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -4148,11 +4144,7 @@ def get_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method.wrap_method(
self._transport.get_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self._transport._wrapped_methods[self._transport.get_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -4209,11 +4201,7 @@ def delete_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method.wrap_method(
self._transport.delete_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self._transport._wrapped_methods[self._transport.delete_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down Expand Up @@ -4266,11 +4254,7 @@ def cancel_operation(

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = gapic_v1.method.wrap_method(
self._transport.cancel_operation,
default_timeout=None,
client_info=DEFAULT_CLIENT_INFO,
)
rpc = self._transport._wrapped_methods[self._transport.cancel_operation]

# Certain fields should be provided within the metadata header;
# add these here.
Expand Down
0 ...ices/database_admin/transports/README.rst → ...ices/database_admin/transports/README.rst
View file
File renamed without changes.
20 changes: 20 additions & 0 deletions google/cloud/spanner_admin_database_v1/services/database_admin/transports/base.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -458,6 +458,26 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.cancel_operation: gapic_v1.method.wrap_method(
self.cancel_operation,
default_timeout=None,
client_info=client_info,
),
self.delete_operation: gapic_v1.method.wrap_method(
self.delete_operation,
default_timeout=None,
client_info=client_info,
),
self.get_operation: gapic_v1.method.wrap_method(
self.get_operation,
default_timeout=None,
client_info=client_info,
),
self.list_operations: gapic_v1.method.wrap_method(
self.list_operations,
default_timeout=None,
client_info=client_info,
),
}

def close(self):
Expand Down
83 changes: 58 additions & 25 deletions google/cloud/spanner_admin_database_v1/services/database_admin/transports/grpc_asyncio.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
import inspect
import warnings
from typing import Awaitable, Callable, Dict, Optional, Sequence, Tuple, Union

Expand Down Expand Up @@ -246,6 +247,9 @@ def __init__(
)

# Wrap messages. This must be done after self._grpc_channel exists
self._wrap_with_kind = (
"kind" in inspect.signature(gapic_v1.method_async.wrap_method).parameters
)
self._prep_wrapped_messages(client_info)

@property
Expand Down Expand Up @@ -1131,7 +1135,7 @@ def list_backup_schedules(
def _prep_wrapped_messages(self, client_info):
"""Precompute the wrapped methods, overriding the base class method to use async wrappers."""
self._wrapped_methods = {
self.list_databases: gapic_v1.method_async.wrap_method(
self.list_databases: self._wrap_method(
self.list_databases,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1146,12 +1150,12 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.create_database: gapic_v1.method_async.wrap_method(
self.create_database: self._wrap_method(
self.create_database,
default_timeout=3600.0,
client_info=client_info,
),
self.get_database: gapic_v1.method_async.wrap_method(
self.get_database: self._wrap_method(
self.get_database,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1166,7 +1170,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.update_database: gapic_v1.method_async.wrap_method(
self.update_database: self._wrap_method(
self.update_database,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1181,7 +1185,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.update_database_ddl: gapic_v1.method_async.wrap_method(
self.update_database_ddl: self._wrap_method(
self.update_database_ddl,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1196,7 +1200,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.drop_database: gapic_v1.method_async.wrap_method(
self.drop_database: self._wrap_method(
self.drop_database,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1211,7 +1215,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.get_database_ddl: gapic_v1.method_async.wrap_method(
self.get_database_ddl: self._wrap_method(
self.get_database_ddl,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1226,12 +1230,12 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.set_iam_policy: gapic_v1.method_async.wrap_method(
self.set_iam_policy: self._wrap_method(
self.set_iam_policy,
default_timeout=30.0,
client_info=client_info,
),
self.get_iam_policy: gapic_v1.method_async.wrap_method(
self.get_iam_policy: self._wrap_method(
self.get_iam_policy,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1246,22 +1250,22 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=30.0,
client_info=client_info,
),
self.test_iam_permissions: gapic_v1.method_async.wrap_method(
self.test_iam_permissions: self._wrap_method(
self.test_iam_permissions,
default_timeout=30.0,
client_info=client_info,
),
self.create_backup: gapic_v1.method_async.wrap_method(
self.create_backup: self._wrap_method(
self.create_backup,
default_timeout=3600.0,
client_info=client_info,
),
self.copy_backup: gapic_v1.method_async.wrap_method(
self.copy_backup: self._wrap_method(
self.copy_backup,
default_timeout=3600.0,
client_info=client_info,
),
self.get_backup: gapic_v1.method_async.wrap_method(
self.get_backup: self._wrap_method(
self.get_backup,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1276,7 +1280,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.update_backup: gapic_v1.method_async.wrap_method(
self.update_backup: self._wrap_method(
self.update_backup,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1291,7 +1295,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.delete_backup: gapic_v1.method_async.wrap_method(
self.delete_backup: self._wrap_method(
self.delete_backup,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1306,7 +1310,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.list_backups: gapic_v1.method_async.wrap_method(
self.list_backups: self._wrap_method(
self.list_backups,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1321,12 +1325,12 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.restore_database: gapic_v1.method_async.wrap_method(
self.restore_database: self._wrap_method(
self.restore_database,
default_timeout=3600.0,
client_info=client_info,
),
self.list_database_operations: gapic_v1.method_async.wrap_method(
self.list_database_operations: self._wrap_method(
self.list_database_operations,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1341,7 +1345,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.list_backup_operations: gapic_v1.method_async.wrap_method(
self.list_backup_operations: self._wrap_method(
self.list_backup_operations,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1356,7 +1360,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.list_database_roles: gapic_v1.method_async.wrap_method(
self.list_database_roles: self._wrap_method(
self.list_database_roles,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1371,7 +1375,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.create_backup_schedule: gapic_v1.method_async.wrap_method(
self.create_backup_schedule: self._wrap_method(
self.create_backup_schedule,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1386,7 +1390,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.get_backup_schedule: gapic_v1.method_async.wrap_method(
self.get_backup_schedule: self._wrap_method(
self.get_backup_schedule,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1401,7 +1405,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.update_backup_schedule: gapic_v1.method_async.wrap_method(
self.update_backup_schedule: self._wrap_method(
self.update_backup_schedule,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1416,7 +1420,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.delete_backup_schedule: gapic_v1.method_async.wrap_method(
self.delete_backup_schedule: self._wrap_method(
self.delete_backup_schedule,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1431,7 +1435,7 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.list_backup_schedules: gapic_v1.method_async.wrap_method(
self.list_backup_schedules: self._wrap_method(
self.list_backup_schedules,
default_retry=retries.AsyncRetry(
initial=1.0,
Expand All @@ -1446,11 +1450,40 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=3600.0,
client_info=client_info,
),
self.cancel_operation: self._wrap_method(
self.cancel_operation,
default_timeout=None,
client_info=client_info,
),
self.delete_operation: self._wrap_method(
self.delete_operation,
default_timeout=None,
client_info=client_info,
),
self.get_operation: self._wrap_method(
self.get_operation,
default_timeout=None,
client_info=client_info,
),
self.list_operations: self._wrap_method(
self.list_operations,
default_timeout=None,
client_info=client_info,
),
}

def _wrap_method(self, func, *args, **kwargs):
if self._wrap_with_kind: # pragma: NO COVER
kwargs["kind"] = self.kind
return gapic_v1.method_async.wrap_method(func, *args, **kwargs)

def close(self):
return self.grpc_channel.close()

@property
def kind(self) -> str:
return "grpc_asyncio"

@property
def delete_operation(
self,
Expand Down
Loading

0 comments on commit d0340b6

Please sign in to comment.