qapi/visit.py: assert tag_member contains a QAPISchemaEnumType

This is true by design, but not presently able to be expressed in the
type system. An assertion helps mypy understand our constraints.

Signed-off-by: John Snow <jsnow@redhat.com>
Reviewed-by: Eduardo Habkost <ehabkost@redhat.com>
Reviewed-by: Cleber Rosa <crosa@redhat.com>
Message-Id: <20201009161558.107041-35-jsnow@redhat.com>
Reviewed-by: Markus Armbruster <armbru@redhat.com>
Signed-off-by: Markus Armbruster <armbru@redhat.com>
This commit is contained in:
John Snow 2020-10-09 12:15:56 -04:00 committed by Markus Armbruster
parent dec44d3d65
commit 554df4f33f
1 changed files with 7 additions and 5 deletions

View File

@ -22,7 +22,7 @@ from .common import (
mcgen, mcgen,
) )
from .gen import QAPISchemaModularCVisitor, ifcontext from .gen import QAPISchemaModularCVisitor, ifcontext
from .schema import QAPISchemaObjectType from .schema import QAPISchemaEnumType, QAPISchemaObjectType
def gen_visit_decl(name, scalar=False): def gen_visit_decl(name, scalar=False):
@ -84,15 +84,17 @@ bool visit_type_%(c_name)s_members(Visitor *v, %(c_name)s *obj, Error **errp)
ret += gen_endif(memb.ifcond) ret += gen_endif(memb.ifcond)
if variants: if variants:
tag_member = variants.tag_member
assert isinstance(tag_member.type, QAPISchemaEnumType)
ret += mcgen(''' ret += mcgen('''
switch (obj->%(c_name)s) { switch (obj->%(c_name)s) {
''', ''',
c_name=c_name(variants.tag_member.name)) c_name=c_name(tag_member.name))
for var in variants.variants: for var in variants.variants:
case_str = c_enum_const(variants.tag_member.type.name, case_str = c_enum_const(tag_member.type.name, var.name,
var.name, tag_member.type.prefix)
variants.tag_member.type.prefix)
ret += gen_if(var.ifcond) ret += gen_if(var.ifcond)
if var.type.name == 'q_empty': if var.type.name == 'q_empty':
# valid variant and nothing to do # valid variant and nothing to do