Check if betterproto has wrapper support in idiomatic way

2020-06-03 23:53:36 +02:00 · 2020-06-03 23:53:36 +02:00 · 919b0a6a7d
commit 919b0a6a7d
parent 7ecf3fe0e6
1 changed files with 1 additions and 1 deletions
--- 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