fix(snippetgen): use f-strings in print statements (#975)

Co-authored-by: Tres Seaver <[email protected]>

Author: busunkim96

packages/gapic-generator/DEVELOPMENT.md

@@ -21,7 +21,7 @@
 from typing import Any, DefaultDict, Dict, Mapping
 from hashlib import sha256
 from collections import OrderedDict, defaultdict
-from gapic.samplegen_utils.utils import coerce_response_name, is_valid_sample_cfg
+from gapic.samplegen_utils.utils import coerce_response_name, is_valid_sample_cfg, render_format_string
 from gapic.samplegen_utils.types import DuplicateSample
 from gapic.samplegen import manifest, samplegen
 from gapic.generator import formatter
@@ -62,6 +62,7 @@ def __init__(self, opts: Options) -> None:
         self._env.filters["sort_lines"] = utils.sort_lines
         self._env.filters["wrap"] = utils.wrap
         self._env.filters["coerce_response_name"] = coerce_response_name
+        self._env.filters["render_format_string"] = render_format_string
 
         # Add tests to determine type of expressions stored in strings
         self._env.tests["str_field_pb"] = utils.is_str_field_pb

packages/gapic-generator/gapic/generator/generator.py

@@ -18,7 +18,7 @@
 import os
 import yaml
 
-from typing import (Generator, Tuple)
+from typing import (Generator, Tuple, List, Union)
 
 from gapic.samplegen_utils import types
 
@@ -28,6 +28,25 @@
 VALID_CONFIG_TYPE = "com.google.api.codegen.samplegen.v1p2.SampleConfigProto"
 
 
+def render_format_string(s: str, expressions: List[str] = []) -> str:
+    """Given string s and a list of expressions, substitute each %s
+    in the string with {exp}.
+
+    Arguments:
+        s (str): The string literal.
+        expressions (Optional[List[str]]): A list of expressions.
+    """
+
+    s = s.replace('\"', '\\\"')
+
+    for exp in expressions:
+        # some expressions will contain references to "$resp"
+        exp = coerce_response_name(exp)
+        s = s.replace("%s", f"{{{exp}}}", 1)
+
+    return s
+
+
 def coerce_response_name(s: str) -> str:
     # In the sample config, the "$resp" keyword is used to refer to the
     # item of interest as received by the corresponding calling form.

packages/gapic-generator/gapic/samplegen_utils/utils.py

@@ -27,10 +27,15 @@
 {% endmacro %}
 
 {% macro print_string_formatting(string_list) %}
+
 {% if string_list|length == 1 %}
-"{{ string_list[0]|replace("%s", "{}")|replace('\"', '\\\"') }}"
+"{{ string_list[0] | render_format_string }}"
+{% elif string_list|length == 2 and string_list[0] == "%s" and string_list[1] == "$resp" %}
+response
 {% else %}
-"{{ string_list[0]|replace("%s", "{}")|replace('\"', '\\\"') }}".format({{ string_list[1:]|map("coerce_response_name")|join(", ") }})
+{# Note: This is the equivalent of render_format_string(string_list[0], string_list[1:] )
+# See https://jinja.palletsprojects.com/en/3.0.x/api/#custom-filters #}
+f"{{ string_list[0] | render_format_string(string_list[1:]) }}"
 {% endif %}
 {% endmacro %}
 

packages/gapic-generator/gapic/templates/examples/feature_fragments.j2

@@ -45,6 +45,6 @@ async def sample_analyze_iam_policy():
     response = await client.analyze_iam_policy(request=request)
 
     # Handle response
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_AnalyzeIamPolicy_async]

packages/gapic-generator/tests/integration/goldens/asset/samples/generated_samples/cloudasset_generated_asset_v1_asset_service_analyze_iam_policy_async.py

@@ -51,6 +51,6 @@ async def sample_analyze_iam_policy_longrunning():
     print("Waiting for operation to complete...")
 
     response = await operation.result()
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_AnalyzeIamPolicyLongrunning_async]

packages/gapic-generator/tests/integration/goldens/asset/samples/generated_samples/cloudasset_generated_asset_v1_asset_service_analyze_iam_policy_longrunning_async.py

@@ -51,6 +51,6 @@ def sample_analyze_iam_policy_longrunning():
     print("Waiting for operation to complete...")
 
     response = operation.result()
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_AnalyzeIamPolicyLongrunning_sync]

packages/gapic-generator/tests/integration/goldens/asset/samples/generated_samples/cloudasset_generated_asset_v1_asset_service_analyze_iam_policy_longrunning_sync.py

@@ -45,6 +45,6 @@ def sample_analyze_iam_policy():
     response = client.analyze_iam_policy(request=request)
 
     # Handle response
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_AnalyzeIamPolicy_sync]

packages/gapic-generator/tests/integration/goldens/asset/samples/generated_samples/cloudasset_generated_asset_v1_asset_service_analyze_iam_policy_sync.py

@@ -42,6 +42,6 @@ async def sample_batch_get_assets_history():
     response = await client.batch_get_assets_history(request=request)
 
     # Handle response
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_BatchGetAssetsHistory_async]

packages/gapic-generator/tests/integration/goldens/asset/samples/generated_samples/cloudasset_generated_asset_v1_asset_service_batch_get_assets_history_async.py

@@ -42,6 +42,6 @@ def sample_batch_get_assets_history():
     response = client.batch_get_assets_history(request=request)
 
     # Handle response
-    print("{}".format(response))
+    print(response)
 
 # [END cloudasset_generated_asset_v1_AssetService_BatchGetAssetsHistory_sync]