def convert_import_relative(config: ParserConfig, children: Sequence[Any]) -> Any:
dots = []
dotted_name = None
for child in children:
if isinstance(child, Token):
# Special case for "...", which is part of the grammar
if child.string == "...":
dots.extend(
[
Dot(),
Dot(),
Dot(
whitespace_after=parse_simple_whitespace(
config, child.whitespace_after
)
),
]
)
else:
dots.append(
Dot(
whitespace_after=parse_simple_whitespace(
config, child.whitespace_after
)
)
)
else:
# This should be the dotted name, and we can't get more than
# one, but lets be sure anyway
if dotted_name is not None:
raise Exception("Logic error!")
dotted_name = child
return ImportRelativePartial(relative=tuple(dots), module=dotted_name)
def convert_trailer_attribute(
config: ParserConfig,
children: typing.Sequence[typing.Any]) -> typing.Any:
dot, name = children
return AttributePartial(
dot=Dot(
whitespace_before=parse_parenthesizable_whitespace(
config, dot.whitespace_before),
whitespace_after=parse_parenthesizable_whitespace(
config, dot.whitespace_after),
),
attr=Name(name.string),
)
def convert_dotted_name(config: ParserConfig, children: Sequence[Any]) -> Any:
left, *rest = children
node = Name(left.string)
for dot, right in grouper(rest, 2):
node = Attribute(
value=node,
dot=Dot(
whitespace_before=parse_parenthesizable_whitespace(
config, dot.whitespace_before),
whitespace_after=parse_parenthesizable_whitespace(
config, dot.whitespace_after),
),
attr=Name(right.string),
)
return node