Check if betterproto has wrapper support in idiomatic way

boukeversteegh · boukeversteegh · commit 919b0a6a7d2b · 2020-06-04T00:02:28.000+02:00
diff --git a/betterproto/plugin.py b/betterproto/plugin.py
@@ -202,7 +202,7 @@ def generate_code(request, response):
                         )
                         if match_wrapper:
                             wrapped_type = "TYPE_" + match_wrapper.group(1).upper()
-                            if wrapped_type in dir(betterproto):
+                            if hasattr(betterproto, wrapped_type):
                                 field_wraps = f"betterproto.{wrapped_type}"
 
                         map_types = None

Original file line number	Diff line number	Diff line change
`@@ -202,7 +202,7 @@ def generate_code(request, response):`
`202`	`202`	`)`
`203`	`203`	`if match_wrapper:`
`204`	`204`	`wrapped_type = "TYPE_" + match_wrapper.group(1).upper()`
`205`		`- if wrapped_type in dir(betterproto):`
	`205`	`+ if hasattr(betterproto, wrapped_type):`
`206`	`206`	`field_wraps = f"betterproto.{wrapped_type}"`
`207`	`207`
`208`	`208`	`map_types = None`