pub/python-betterproto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Do not unwrap google.protobuf.Value and unsupported wrapper types

Browse Source
This commit is contained in:
boukeversteegh 2020-05-29 15:36:00 +02:00
parent e8991339e9
commit 53ce1255d3
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
betterproto/plugin.py
View File

@ -1,13 +1,15 @@
#!/usr/bin/env python
from collections import defaultdict
import itertools
import os.path
import re
import stringcase
import sys
import textwrap
from collections import defaultdict
from typing import Dict, List, Optional, Type
from betterproto.casing import safe_snake_case
import betterproto
try:
# betterproto[compiler] specific dependencies
@ -259,11 +261,13 @@ def generate_code(request, response):
field_type = f.Type.Name(f.type).lower()[5:]
field_wraps = ""
if f.type_name.startswith(
".google.protobuf"
) and f.type_name.endswith("Value"):
w = f.type_name.split(".").pop()[:-5].upper()
field_wraps = f"betterproto.TYPE_{w}"
match_wrapper = re.match(
"\\.google\\.protobuf\\.(.+)Value", f.type_name
)
if match_wrapper:
wrapped_type = "TYPE_" + match_wrapper.group(1).upper()
if wrapped_type in dir(betterproto):
field_wraps = f"betterproto.{wrapped_type}"
map_types = None
if f.type == 11: