betterproto2-compiler 0.0.2__tar.gz → 0.0.3__tar.gz

{betterproto2_compiler-0.0.2 → betterproto2_compiler-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: betterproto2_compiler
-Version: 0.0.2
+Version: 0.0.3
 Summary: Compiler for betterproto2
 Home-page: https://github.com/betterproto/python-betterproto2-compiler
 License: MIT

{betterproto2_compiler-0.0.2 → betterproto2_compiler-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "betterproto2_compiler"
-version = "0.0.2"
+version = "0.0.3"
 description = "Compiler for betterproto2"
 authors = ["Adrien Vannson <adrien.vannson@protonmail.com>", "Daniel G. Taylor <danielgtaylor@gmail.com>"]
 readme = "README.md"
@@ -61,6 +61,10 @@ combine-as-imports = true
 
 # Dev workflow tasks
 
+[tool.poe.tasks.test]
+cmd = "pytest"
+help = "Run tests"
+
 [tool.poe.tasks.generate]
 script = "tests.generate:main"
 help = "Generate test cases"

{betterproto2_compiler-0.0.2 → betterproto2_compiler-0.0.3}/src/betterproto2_compiler/compile/importing.py

@@ -105,9 +105,9 @@ def get_type_reference(
     compiling_google_protobuf = current_package == ["google", "protobuf"]
     importing_google_protobuf = py_package == ["google", "protobuf"]
     if importing_google_protobuf and not compiling_google_protobuf:
-        py_package = ["betterproto", "lib"] + (["pydantic"] if pydantic else []) + py_package
+        py_package = ["betterproto2", "lib"] + (["pydantic"] if pydantic else []) + py_package
 
-    if py_package[:1] == ["betterproto"]:
+    if py_package[:1] == ["betterproto2"]:
         return reference_absolute(imports, py_package, py_type)
 
     if py_package == current_package:

{betterproto2_compiler-0.0.2 → betterproto2_compiler-0.0.3}/src/betterproto2_compiler/plugin/models.py

@@ -46,7 +46,8 @@ from typing import (
     Union,
 )
 
-import betterproto2_compiler
+import betterproto2
+
 from betterproto2_compiler.compile.naming import (
     pythonize_class_name,
     pythonize_field_name,
@@ -182,7 +183,7 @@ class ProtoContentBase:
     source_file: FileDescriptorProto
     typing_compiler: TypingCompiler
     path: List[int]
-    parent: Union["betterproto2_compiler.Message", "OutputTemplate"]
+    parent: Union["betterproto2.Message", "OutputTemplate"]
 
     __dataclass_fields__: Dict[str, object]
 
@@ -420,7 +421,7 @@ class FieldCompiler(ProtoContentBase):
         match_wrapper = re.match(r"\.google\.protobuf\.(.+)Value$", self.proto_obj.type_name)
         if match_wrapper:
             wrapped_type = "TYPE_" + match_wrapper.group(1).upper()
-            if hasattr(betterproto2_compiler, wrapped_type):
+            if hasattr(betterproto2, wrapped_type):
                 return f"betterproto2.{wrapped_type}"
         return None