diff --git a/google/cloud/dataflow_v1beta3/services/flex_templates_service/transports/rest.py b/google/cloud/dataflow_v1beta3/services/flex_templates_service/transports/rest.py index ed9312e..9d55045 100644 --- a/google/cloud/dataflow_v1beta3/services/flex_templates_service/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/flex_templates_service/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class FlexTemplatesServiceRestInterceptor: """Interceptor for FlexTemplatesService. @@ -67,12 +63,13 @@ class FlexTemplatesServiceRestInterceptor: .. code-block:: python class MyCustomFlexTemplatesServiceInterceptor(FlexTemplatesServiceRestInterceptor): - def pre_launch_flex_template(request, metadata): + def pre_launch_flex_template(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_launch_flex_template(response): + def post_launch_flex_template(self, response): logging.log(f"Received response: {response}") + return response transport = FlexTemplatesServiceRestTransport(interceptor=MyCustomFlexTemplatesServiceInterceptor()) client = FlexTemplatesServiceClient(transport=transport) diff --git a/google/cloud/dataflow_v1beta3/services/jobs_v1_beta3/transports/rest.py b/google/cloud/dataflow_v1beta3/services/jobs_v1_beta3/transports/rest.py index b151109..18eaf06 100644 --- a/google/cloud/dataflow_v1beta3/services/jobs_v1_beta3/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/jobs_v1_beta3/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class JobsV1Beta3RestInterceptor: """Interceptor for JobsV1Beta3. @@ -67,54 +63,61 @@ class JobsV1Beta3RestInterceptor: .. code-block:: python class MyCustomJobsV1Beta3Interceptor(JobsV1Beta3RestInterceptor): - def pre_aggregated_list_jobs(request, metadata): + def pre_aggregated_list_jobs(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_aggregated_list_jobs(response): + def post_aggregated_list_jobs(self, response): logging.log(f"Received response: {response}") + return response - def pre_check_active_jobs(request, metadata): + def pre_check_active_jobs(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_check_active_jobs(response): + def post_check_active_jobs(self, response): logging.log(f"Received response: {response}") + return response - def pre_create_job(request, metadata): + def pre_create_job(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_create_job(response): + def post_create_job(self, response): logging.log(f"Received response: {response}") + return response - def pre_get_job(request, metadata): + def pre_get_job(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_job(response): + def post_get_job(self, response): logging.log(f"Received response: {response}") + return response - def pre_list_jobs(request, metadata): + def pre_list_jobs(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_list_jobs(response): + def post_list_jobs(self, response): logging.log(f"Received response: {response}") + return response - def pre_snapshot_job(request, metadata): + def pre_snapshot_job(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_snapshot_job(response): + def post_snapshot_job(self, response): logging.log(f"Received response: {response}") + return response - def pre_update_job(request, metadata): + def pre_update_job(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_update_job(response): + def post_update_job(self, response): logging.log(f"Received response: {response}") + return response transport = JobsV1Beta3RestTransport(interceptor=MyCustomJobsV1Beta3Interceptor()) client = JobsV1Beta3Client(transport=transport) diff --git a/google/cloud/dataflow_v1beta3/services/messages_v1_beta3/transports/rest.py b/google/cloud/dataflow_v1beta3/services/messages_v1_beta3/transports/rest.py index 045698a..c6a61c7 100644 --- a/google/cloud/dataflow_v1beta3/services/messages_v1_beta3/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/messages_v1_beta3/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class MessagesV1Beta3RestInterceptor: """Interceptor for MessagesV1Beta3. @@ -67,12 +63,13 @@ class MessagesV1Beta3RestInterceptor: .. code-block:: python class MyCustomMessagesV1Beta3Interceptor(MessagesV1Beta3RestInterceptor): - def pre_list_job_messages(request, metadata): + def pre_list_job_messages(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_list_job_messages(response): + def post_list_job_messages(self, response): logging.log(f"Received response: {response}") + return response transport = MessagesV1Beta3RestTransport(interceptor=MyCustomMessagesV1Beta3Interceptor()) client = MessagesV1Beta3Client(transport=transport) diff --git a/google/cloud/dataflow_v1beta3/services/metrics_v1_beta3/transports/rest.py b/google/cloud/dataflow_v1beta3/services/metrics_v1_beta3/transports/rest.py index 10b73c2..f32c0c9 100644 --- a/google/cloud/dataflow_v1beta3/services/metrics_v1_beta3/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/metrics_v1_beta3/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class MetricsV1Beta3RestInterceptor: """Interceptor for MetricsV1Beta3. @@ -67,26 +63,29 @@ class MetricsV1Beta3RestInterceptor: .. code-block:: python class MyCustomMetricsV1Beta3Interceptor(MetricsV1Beta3RestInterceptor): - def pre_get_job_execution_details(request, metadata): + def pre_get_job_execution_details(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_job_execution_details(response): + def post_get_job_execution_details(self, response): logging.log(f"Received response: {response}") + return response - def pre_get_job_metrics(request, metadata): + def pre_get_job_metrics(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_job_metrics(response): + def post_get_job_metrics(self, response): logging.log(f"Received response: {response}") + return response - def pre_get_stage_execution_details(request, metadata): + def pre_get_stage_execution_details(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_stage_execution_details(response): + def post_get_stage_execution_details(self, response): logging.log(f"Received response: {response}") + return response transport = MetricsV1Beta3RestTransport(interceptor=MyCustomMetricsV1Beta3Interceptor()) client = MetricsV1Beta3Client(transport=transport) diff --git a/google/cloud/dataflow_v1beta3/services/snapshots_v1_beta3/transports/rest.py b/google/cloud/dataflow_v1beta3/services/snapshots_v1_beta3/transports/rest.py index 54cbbab..8f32c4b 100644 --- a/google/cloud/dataflow_v1beta3/services/snapshots_v1_beta3/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/snapshots_v1_beta3/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class SnapshotsV1Beta3RestInterceptor: """Interceptor for SnapshotsV1Beta3. @@ -67,26 +63,29 @@ class SnapshotsV1Beta3RestInterceptor: .. code-block:: python class MyCustomSnapshotsV1Beta3Interceptor(SnapshotsV1Beta3RestInterceptor): - def pre_delete_snapshot(request, metadata): + def pre_delete_snapshot(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_delete_snapshot(response): + def post_delete_snapshot(self, response): logging.log(f"Received response: {response}") + return response - def pre_get_snapshot(request, metadata): + def pre_get_snapshot(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_snapshot(response): + def post_get_snapshot(self, response): logging.log(f"Received response: {response}") + return response - def pre_list_snapshots(request, metadata): + def pre_list_snapshots(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_list_snapshots(response): + def post_list_snapshots(self, response): logging.log(f"Received response: {response}") + return response transport = SnapshotsV1Beta3RestTransport(interceptor=MyCustomSnapshotsV1Beta3Interceptor()) client = SnapshotsV1Beta3Client(transport=transport) diff --git a/google/cloud/dataflow_v1beta3/services/templates_service/transports/rest.py b/google/cloud/dataflow_v1beta3/services/templates_service/transports/rest.py index be3b8e4..a4ed0c1 100644 --- a/google/cloud/dataflow_v1beta3/services/templates_service/transports/rest.py +++ b/google/cloud/dataflow_v1beta3/services/templates_service/transports/rest.py @@ -47,10 +47,6 @@ rest_version=requests_version, ) -# TODO (numeric enums): This file was generated with the option to -# request that the server respond with enums JSON-encoded as -# numbers. The code below does not implement that functionality yet. - class TemplatesServiceRestInterceptor: """Interceptor for TemplatesService. @@ -67,26 +63,29 @@ class TemplatesServiceRestInterceptor: .. code-block:: python class MyCustomTemplatesServiceInterceptor(TemplatesServiceRestInterceptor): - def pre_create_job_from_template(request, metadata): + def pre_create_job_from_template(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_create_job_from_template(response): + def post_create_job_from_template(self, response): logging.log(f"Received response: {response}") + return response - def pre_get_template(request, metadata): + def pre_get_template(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_get_template(response): + def post_get_template(self, response): logging.log(f"Received response: {response}") + return response - def pre_launch_template(request, metadata): + def pre_launch_template(self, request, metadata): logging.log(f"Received request: {request}") return request, metadata - def post_launch_template(response): + def post_launch_template(self, response): logging.log(f"Received response: {response}") + return response transport = TemplatesServiceRestTransport(interceptor=MyCustomTemplatesServiceInterceptor()) client = TemplatesServiceClient(transport=transport)