Merge pull request #99 from boukeversteegh/release-v2.0.0b1

Release v2.0.0b1

betterproto/compile/importing.py

@@ -86,7 +86,7 @@ def reference_absolute(imports, py_package, py_type):
     string_import = ".".join(py_package)
     string_alias = safe_snake_case(string_import)
     imports.add(f"import {string_import} as {string_alias}")
-    return f'"{string_alias}.{py_type}"'
+    return f"{string_alias}.{py_type}"
 
 
 def reference_sibling(py_type: str) -> str:
@@ -109,10 +109,10 @@ def reference_descendent(
     if string_from:
         string_alias = "_".join(importing_descendent)
         imports.add(f"from .{string_from} import {string_import} as {string_alias}")
-        return f'"{string_alias}.{py_type}"'
+        return f"{string_alias}.{py_type}"
     else:
         imports.add(f"from . import {string_import}")
-        return f'"{string_import}.{py_type}"'
+        return f"{string_import}.{py_type}"
 
 
 def reference_ancestor(
@@ -130,11 +130,11 @@ def reference_ancestor(
         string_alias = f"_{'_' * distance_up}{string_import}__"
         string_from = f"..{'.' * distance_up}"
         imports.add(f"from {string_from} import {string_import} as {string_alias}")
-        return f'"{string_alias}.{py_type}"'
+        return f"{string_alias}.{py_type}"
     else:
         string_alias = f"{'_' * distance_up}{py_type}__"
         imports.add(f"from .{'.' * distance_up} import {py_type} as {string_alias}")
-        return f'"{string_alias}"'
+        return string_alias
 
 
 def reference_cousin(
@@ -157,4 +157,4 @@ def reference_cousin(
         + "__"
     )
     imports.add(f"from {string_from} import {string_import} as {string_alias}")
-    return f'"{string_alias}.{py_type}"'
+    return f"{string_alias}.{py_type}"

betterproto/plugin.py

@@ -329,7 +329,7 @@ def generate_code(request, response):
                                 output["imports"],
                                 method.output_type,
                                 unwrap=False,
-                            ),
+                            ).strip('"'),
                             "client_streaming": method.client_streaming,
                             "server_streaming": method.server_streaming,
                         }

betterproto/templates/template.py.j2

@@ -16,6 +16,10 @@ import betterproto
 import grpclib
 {% endif %}
 
+{% for i in description.imports %}
+{{ i }}
+{% endfor %}
+
 
 {% if description.enums %}{% for enum in description.enums %}
 class {{ enum.py_name }}(betterproto.Enum):
@@ -98,14 +102,14 @@ class {{ service.py_name }}Stub(betterproto.ServiceStub):
             "{{ method.route }}",
             request_iterator,
             {{ method.input }},
-            {{ method.output.strip('"') }},
+            {{ method.output }},
         ):
             yield response
             {% else %}{# i.e. not client streaming #}
         async for response in self._unary_stream(
             "{{ method.route }}",
             request,
-            {{ method.output.strip('"') }},
+            {{ method.output }},
         ):
             yield response
 
@@ -116,20 +120,16 @@ class {{ service.py_name }}Stub(betterproto.ServiceStub):
             "{{ method.route }}",
             request_iterator,
             {{ method.input }},
-            {{ method.output.strip('"') }}
+            {{ method.output }}
         )
             {% else %}{# i.e. not client streaming #}
         return await self._unary_unary(
             "{{ method.route }}",
             request,
-            {{ method.output.strip('"') }}
+            {{ method.output }}
         )
             {% endif %}{# client streaming #}
         {% endif %}
 
     {% endfor %}
 {% endfor %}
-
-{% for i in description.imports %}
-{{ i }}
-{% endfor %}

betterproto/tests/inputs/config.py

@@ -1,6 +1,7 @@
 # Test cases that are expected to fail, e.g. unimplemented features or bug-fixes.
 # Remove from list when fixed.
 xfail = {
+    "import_circular_dependency",
     "oneof_enum",  # 63
     "namespace_keywords",  # 70
     "namespace_builtin_types",  # 53

betterproto/tests/test_get_ref_type.py

@@ -8,22 +8,22 @@ from ..compile.importing import get_type_reference, parse_source_type_name
     [
         (
             ".google.protobuf.Empty",
-            '"betterproto_lib_google_protobuf.Empty"',
+            "betterproto_lib_google_protobuf.Empty",
             "import betterproto.lib.google.protobuf as betterproto_lib_google_protobuf",
         ),
         (
             ".google.protobuf.Struct",
-            '"betterproto_lib_google_protobuf.Struct"',
+            "betterproto_lib_google_protobuf.Struct",
             "import betterproto.lib.google.protobuf as betterproto_lib_google_protobuf",
         ),
         (
             ".google.protobuf.ListValue",
-            '"betterproto_lib_google_protobuf.ListValue"',
+            "betterproto_lib_google_protobuf.ListValue",
             "import betterproto.lib.google.protobuf as betterproto_lib_google_protobuf",
         ),
         (
             ".google.protobuf.Value",
-            '"betterproto_lib_google_protobuf.Value"',
+            "betterproto_lib_google_protobuf.Value",
             "import betterproto.lib.google.protobuf as betterproto_lib_google_protobuf",
         ),
     ],
@@ -67,27 +67,15 @@ def test_referenceing_google_wrappers_unwraps_them(
 @pytest.mark.parametrize(
     ["google_type", "expected_name"],
     [
-        (
+        (".google.protobuf.DoubleValue", "betterproto_lib_google_protobuf.DoubleValue"),
-            ".google.protobuf.DoubleValue",
+        (".google.protobuf.FloatValue", "betterproto_lib_google_protobuf.FloatValue"),
-            '"betterproto_lib_google_protobuf.DoubleValue"',
+        (".google.protobuf.Int32Value", "betterproto_lib_google_protobuf.Int32Value"),
-        ),
+        (".google.protobuf.Int64Value", "betterproto_lib_google_protobuf.Int64Value"),
-        (".google.protobuf.FloatValue", '"betterproto_lib_google_protobuf.FloatValue"'),
+        (".google.protobuf.UInt32Value", "betterproto_lib_google_protobuf.UInt32Value"),
-        (".google.protobuf.Int32Value", '"betterproto_lib_google_protobuf.Int32Value"'),
+        (".google.protobuf.UInt64Value", "betterproto_lib_google_protobuf.UInt64Value"),
-        (".google.protobuf.Int64Value", '"betterproto_lib_google_protobuf.Int64Value"'),
+        (".google.protobuf.BoolValue", "betterproto_lib_google_protobuf.BoolValue"),
-        (
+        (".google.protobuf.StringValue", "betterproto_lib_google_protobuf.StringValue"),
-            ".google.protobuf.UInt32Value",
+        (".google.protobuf.BytesValue", "betterproto_lib_google_protobuf.BytesValue"),
-            '"betterproto_lib_google_protobuf.UInt32Value"',
-        ),
-        (
-            ".google.protobuf.UInt64Value",
-            '"betterproto_lib_google_protobuf.UInt64Value"',
-        ),
-        (".google.protobuf.BoolValue", '"betterproto_lib_google_protobuf.BoolValue"'),
-        (
-            ".google.protobuf.StringValue",
-            '"betterproto_lib_google_protobuf.StringValue"',
-        ),
-        (".google.protobuf.BytesValue", '"betterproto_lib_google_protobuf.BytesValue"'),
     ],
 )
 def test_referenceing_google_wrappers_without_unwrapping(
@@ -107,7 +95,7 @@ def test_reference_child_package_from_package():
     )
 
     assert imports == {"from . import child"}
-    assert name == '"child.Message"'
+    assert name == "child.Message"
 
 
 def test_reference_child_package_from_root():
@@ -115,7 +103,7 @@ def test_reference_child_package_from_root():
     name = get_type_reference(package="", imports=imports, source_type="child.Message")
 
     assert imports == {"from . import child"}
-    assert name == '"child.Message"'
+    assert name == "child.Message"
 
 
 def test_reference_camel_cased():
@@ -125,7 +113,7 @@ def test_reference_camel_cased():
     )
 
     assert imports == {"from . import child_package"}
-    assert name == '"child_package.ExampleMessage"'
+    assert name == "child_package.ExampleMessage"
 
 
 def test_reference_nested_child_from_root():
@@ -135,7 +123,7 @@ def test_reference_nested_child_from_root():
     )
 
     assert imports == {"from .nested import child as nested_child"}
-    assert name == '"nested_child.Message"'
+    assert name == "nested_child.Message"
 
 
 def test_reference_deeply_nested_child_from_root():
@@ -145,7 +133,7 @@ def test_reference_deeply_nested_child_from_root():
     )
 
     assert imports == {"from .deeply.nested import child as deeply_nested_child"}
-    assert name == '"deeply_nested_child.Message"'
+    assert name == "deeply_nested_child.Message"
 
 
 def test_reference_deeply_nested_child_from_package():
@@ -157,7 +145,7 @@ def test_reference_deeply_nested_child_from_package():
     )
 
     assert imports == {"from .deeply.nested import child as deeply_nested_child"}
-    assert name == '"deeply_nested_child.Message"'
+    assert name == "deeply_nested_child.Message"
 
 
 def test_reference_root_sibling():
@@ -193,7 +181,7 @@ def test_reference_parent_package_from_child():
     )
 
     assert imports == {"from ... import package as __package__"}
-    assert name == '"__package__.Message"'
+    assert name == "__package__.Message"
 
 
 def test_reference_parent_package_from_deeply_nested_child():
@@ -205,7 +193,7 @@ def test_reference_parent_package_from_deeply_nested_child():
     )
 
     assert imports == {"from ... import nested as __nested__"}
-    assert name == '"__nested__.Message"'
+    assert name == "__nested__.Message"
 
 
 def test_reference_ancestor_package_from_nested_child():
@@ -217,7 +205,7 @@ def test_reference_ancestor_package_from_nested_child():
     )
 
     assert imports == {"from .... import ancestor as ___ancestor__"}
-    assert name == '"___ancestor__.Message"'
+    assert name == "___ancestor__.Message"
 
 
 def test_reference_root_package_from_child():
@@ -227,7 +215,7 @@ def test_reference_root_package_from_child():
     )
 
     assert imports == {"from ... import Message as __Message__"}
-    assert name == '"__Message__"'
+    assert name == "__Message__"
 
 
 def test_reference_root_package_from_deeply_nested_child():
@@ -237,7 +225,7 @@ def test_reference_root_package_from_deeply_nested_child():
     )
 
     assert imports == {"from ..... import Message as ____Message__"}
-    assert name == '"____Message__"'
+    assert name == "____Message__"
 
 
 def test_reference_unrelated_package():
@@ -245,7 +233,7 @@ def test_reference_unrelated_package():
     name = get_type_reference(package="a", imports=imports, source_type="p.Message")
 
     assert imports == {"from .. import p as _p__"}
-    assert name == '"_p__.Message"'
+    assert name == "_p__.Message"
 
 
 def test_reference_unrelated_nested_package():
@@ -253,7 +241,7 @@ def test_reference_unrelated_nested_package():
     name = get_type_reference(package="a.b", imports=imports, source_type="p.q.Message")
 
     assert imports == {"from ...p import q as __p_q__"}
-    assert name == '"__p_q__.Message"'
+    assert name == "__p_q__.Message"
 
 
 def test_reference_unrelated_deeply_nested_package():
@@ -263,7 +251,7 @@ def test_reference_unrelated_deeply_nested_package():
     )
 
     assert imports == {"from .....p.q.r import s as ____p_q_r_s__"}
-    assert name == '"____p_q_r_s__.Message"'
+    assert name == "____p_q_r_s__.Message"
 
 
 def test_reference_cousin_package():
@@ -271,7 +259,7 @@ def test_reference_cousin_package():
     name = get_type_reference(package="a.x", imports=imports, source_type="a.y.Message")
 
     assert imports == {"from .. import y as _y__"}
-    assert name == '"_y__.Message"'
+    assert name == "_y__.Message"
 
 
 def test_reference_cousin_package_different_name():
@@ -281,7 +269,7 @@ def test_reference_cousin_package_different_name():
     )
 
     assert imports == {"from ...cousin import package2 as __cousin_package2__"}
-    assert name == '"__cousin_package2__.Message"'
+    assert name == "__cousin_package2__.Message"
 
 
 def test_reference_cousin_package_same_name():
@@ -291,7 +279,7 @@ def test_reference_cousin_package_same_name():
     )
 
     assert imports == {"from ...cousin import package as __cousin_package__"}
-    assert name == '"__cousin_package__.Message"'
+    assert name == "__cousin_package__.Message"
 
 
 def test_reference_far_cousin_package():
@@ -301,7 +289,7 @@ def test_reference_far_cousin_package():
     )
 
     assert imports == {"from ...b import c as __b_c__"}
-    assert name == '"__b_c__.Message"'
+    assert name == "__b_c__.Message"
 
 
 def test_reference_far_far_cousin_package():
@@ -311,7 +299,7 @@ def test_reference_far_far_cousin_package():
     )
 
     assert imports == {"from ....b.c import d as ___b_c_d__"}
-    assert name == '"___b_c_d__.Message"'
+    assert name == "___b_c_d__.Message"
 
 
 @pytest.mark.parametrize(