From 1ddd9f5e80283a17ada9bac3942c0ee98cfb32d7 Mon Sep 17 00:00:00 2001 From: Tres Seaver Date: Thu, 18 Apr 2019 15:19:29 -0400 Subject: [PATCH] Add routing header to method metadata (via synth). --- .../firestore_v1/gapic/firestore_client.py | 118 ++++++++++++++++++ .../gapic/firestore_client.py | 118 ++++++++++++++++++ firestore/synth.metadata | 12 +- 3 files changed, 242 insertions(+), 6 deletions(-) diff --git a/firestore/google/cloud/firestore_v1/gapic/firestore_client.py b/firestore/google/cloud/firestore_v1/gapic/firestore_client.py index deee20ea3960..773c4a98c6f2 100644 --- a/firestore/google/cloud/firestore_v1/gapic/firestore_client.py +++ b/firestore/google/cloud/firestore_v1/gapic/firestore_client.py @@ -23,6 +23,7 @@ import google.api_core.gapic_v1.client_info import google.api_core.gapic_v1.config import google.api_core.gapic_v1.method +import google.api_core.gapic_v1.routing_header import google.api_core.grpc_helpers import google.api_core.page_iterator import google.api_core.path_template @@ -309,6 +310,19 @@ def get_document( request = firestore_pb2.GetDocumentRequest( name=name, mask=mask, transaction=transaction, read_time=read_time ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("name", name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["get_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -635,6 +649,19 @@ def update_document( mask=mask, current_document=current_document, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("document.name", document.name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["update_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -697,6 +724,19 @@ def delete_document( request = firestore_pb2.DeleteDocumentRequest( name=name, current_document=current_document ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("name", name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + self._inner_api_calls["delete_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -806,6 +846,19 @@ def batch_get_documents( new_transaction=new_transaction, read_time=read_time, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["batch_get_documents"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -871,6 +924,19 @@ def begin_transaction( request = firestore_pb2.BeginTransactionRequest( database=database, options=options_ ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["begin_transaction"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -942,6 +1008,19 @@ def commit( request = firestore_pb2.CommitRequest( database=database, writes=writes, transaction=transaction ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["commit"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1003,6 +1082,19 @@ def rollback( request = firestore_pb2.RollbackRequest( database=database, transaction=transaction ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + self._inner_api_calls["rollback"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1104,6 +1196,19 @@ def run_query( new_transaction=new_transaction, read_time=read_time, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("parent", parent)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["run_query"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1310,6 +1415,19 @@ def list_collection_ids( request = firestore_pb2.ListCollectionIdsRequest( parent=parent, page_size=page_size ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("parent", parent)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( diff --git a/firestore/google/cloud/firestore_v1beta1/gapic/firestore_client.py b/firestore/google/cloud/firestore_v1beta1/gapic/firestore_client.py index 367d6463de1f..4b080e08bccc 100644 --- a/firestore/google/cloud/firestore_v1beta1/gapic/firestore_client.py +++ b/firestore/google/cloud/firestore_v1beta1/gapic/firestore_client.py @@ -23,6 +23,7 @@ import google.api_core.gapic_v1.client_info import google.api_core.gapic_v1.config import google.api_core.gapic_v1.method +import google.api_core.gapic_v1.routing_header import google.api_core.grpc_helpers import google.api_core.page_iterator import google.api_core.path_template @@ -309,6 +310,19 @@ def get_document( request = firestore_pb2.GetDocumentRequest( name=name, mask=mask, transaction=transaction, read_time=read_time ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("name", name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["get_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -635,6 +649,19 @@ def update_document( mask=mask, current_document=current_document, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("document.name", document.name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["update_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -697,6 +724,19 @@ def delete_document( request = firestore_pb2.DeleteDocumentRequest( name=name, current_document=current_document ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("name", name)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + self._inner_api_calls["delete_document"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -806,6 +846,19 @@ def batch_get_documents( new_transaction=new_transaction, read_time=read_time, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["batch_get_documents"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -871,6 +924,19 @@ def begin_transaction( request = firestore_pb2.BeginTransactionRequest( database=database, options=options_ ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["begin_transaction"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -942,6 +1008,19 @@ def commit( request = firestore_pb2.CommitRequest( database=database, writes=writes, transaction=transaction ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["commit"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1003,6 +1082,19 @@ def rollback( request = firestore_pb2.RollbackRequest( database=database, transaction=transaction ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("database", database)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + self._inner_api_calls["rollback"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1104,6 +1196,19 @@ def run_query( new_transaction=new_transaction, read_time=read_time, ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("parent", parent)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + return self._inner_api_calls["run_query"]( request, retry=retry, timeout=timeout, metadata=metadata ) @@ -1310,6 +1415,19 @@ def list_collection_ids( request = firestore_pb2.ListCollectionIdsRequest( parent=parent, page_size=page_size ) + if metadata is None: + metadata = [] + metadata = list(metadata) + try: + routing_header = [("parent", parent)] + except AttributeError: + pass + else: + routing_metadata = google.api_core.gapic_v1.routing_header.to_grpc_metadata( + routing_header + ) + metadata.append(routing_metadata) + iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( diff --git a/firestore/synth.metadata b/firestore/synth.metadata index 942c38b56000..2f2daff09cd6 100644 --- a/firestore/synth.metadata +++ b/firestore/synth.metadata @@ -1,26 +1,26 @@ { - "updateTime": "2019-03-27T19:35:27.286829Z", + "updateTime": "2019-04-18T19:15:20.848201Z", "sources": [ { "generator": { "name": "artman", - "version": "0.16.19", - "dockerImage": "googleapis/artman@sha256:70ba28fda87e032ae44e6df41b7fc342c1b0cce1ed90658c4890eb4f613038c2" + "version": "0.16.25", + "dockerImage": "googleapis/artman@sha256:d9597f983d1d4e61272c63cb97b7d8f8234da9999526c35d357de3d781f0ec1b" } }, { "git": { "name": "googleapis", "remote": "https://github.com/googleapis/googleapis.git", - "sha": "1119e688a00927cb02a2361929f0ca3190f88466", - "internalRef": "240608914" + "sha": "9a89b9443aa9d43ccecbd5200cb866e551bbd5e7", + "internalRef": "244215550" } }, { "template": { "name": "python_library", "origin": "synthtool.gcp", - "version": "2019.2.26" + "version": "2019.4.10" } } ],