Fix generator error with OneOfs
This commit is contained in:
@@ -582,7 +582,8 @@ class ExtensionField(Field):
|
|||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
class OneOf(Field):
|
class OneOf(Field):
|
||||||
def __init__(self, oneof_desc):
|
def __init__(self, struct_name, oneof_desc):
|
||||||
|
self.struct_name = struct_name
|
||||||
self.name = oneof_desc.name
|
self.name = oneof_desc.name
|
||||||
self.ctype = 'union'
|
self.ctype = 'union'
|
||||||
self.fields = []
|
self.fields = []
|
||||||
@@ -655,7 +656,7 @@ class Message:
|
|||||||
|
|
||||||
if hasattr(desc, 'oneof_decl'):
|
if hasattr(desc, 'oneof_decl'):
|
||||||
for f in desc.oneof_decl:
|
for f in desc.oneof_decl:
|
||||||
oneof = OneOf(f)
|
oneof = OneOf(self.name, f)
|
||||||
self.oneofs.append(oneof)
|
self.oneofs.append(oneof)
|
||||||
self.fields.append(oneof)
|
self.fields.append(oneof)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user