def convert_annassign(config: ParserConfig, children: Sequence[Any]) -> Any:
if len(children) == 2:
# Variable annotation only
colon, annotation = children
annotation = annotation.value
equal = None
value = None
elif len(children) == 4:
# Variable annotation and assignment
colon, annotation, equal, value = children
annotation = annotation.value
value = value.value
equal = AssignEqual(
whitespace_before=parse_simple_whitespace(config, equal.whitespace_before),
whitespace_after=parse_simple_whitespace(config, equal.whitespace_after),
)
else:
raise Exception("Invalid parser state!")
return AnnAssignPartial(
annotation=Annotation(
whitespace_before_indicator=parse_simple_whitespace(
config, colon.whitespace_before
),
whitespace_after_indicator=parse_simple_whitespace(
config, colon.whitespace_after
),
annotation=annotation,
),
equal=equal,
value=value,
)
def convert_funcdef_annotation(config: ParserConfig,
children: Sequence[Any]) -> Any:
arrow, typehint = children
return Annotation(
whitespace_before_indicator=parse_parenthesizable_whitespace(
config, arrow.whitespace_before),
whitespace_after_indicator=parse_parenthesizable_whitespace(
config, arrow.whitespace_after),
annotation=typehint.value,
)
def convert_fpdef(config: ParserConfig, children: Sequence[Any]) -> Any:
if len(children) == 1:
# This is just a parameter
(child, ) = children
namenode = Name(child.string)
annotation = None
else:
# This is a parameter with a type hint
name, colon, typehint = children
namenode = Name(name.string)
annotation = Annotation(
whitespace_before_indicator=parse_parenthesizable_whitespace(
config, colon.whitespace_before),
whitespace_after_indicator=parse_parenthesizable_whitespace(
config, colon.whitespace_after),
annotation=typehint.value,
)
return Param(star="", name=namenode, annotation=annotation, default=None)