Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/astroid/brain/brain_pathlib.py: 76%
21 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:53 +0000
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:53 +0000
1# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
2# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
3# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
5from __future__ import annotations
7from collections.abc import Iterator
9from astroid import bases, context, inference_tip, nodes
10from astroid.builder import _extract_single_node
11from astroid.exceptions import InferenceError, UseInferenceDefault
12from astroid.manager import AstroidManager
14PATH_TEMPLATE = """
15from pathlib import Path
16Path
17"""
20def _looks_like_parents_subscript(node: nodes.Subscript) -> bool:
21 if not (
22 isinstance(node.value, nodes.Attribute) and node.value.attrname == "parents"
23 ):
24 return False
26 try:
27 value = next(node.value.infer())
28 except (InferenceError, StopIteration):
29 return False
30 return (
31 isinstance(value, bases.Instance)
32 and isinstance(value._proxied, nodes.ClassDef)
33 and value.qname() == "pathlib._PathParents"
34 )
37def infer_parents_subscript(
38 subscript_node: nodes.Subscript, ctx: context.InferenceContext | None = None
39) -> Iterator[bases.Instance]:
40 if isinstance(subscript_node.slice, nodes.Const):
41 path_cls = next(_extract_single_node(PATH_TEMPLATE).infer())
42 return iter([path_cls.instantiate_class()])
44 raise UseInferenceDefault
47AstroidManager().register_transform(
48 nodes.Subscript,
49 inference_tip(infer_parents_subscript),
50 _looks_like_parents_subscript,
51)