Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/astroid/brain/brain_pathlib.py: 86%

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 

4 

5from __future__ import annotations 

6 

7from collections.abc import Iterator 

8 

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 

13 

14PATH_TEMPLATE = """ 

15from pathlib import Path 

16Path 

17""" 

18 

19 

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 

25 

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 ) 

35 

36 

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()]) 

43 

44 raise UseInferenceDefault 

45 

46 

47def register(manager: AstroidManager) -> None: 

48 manager.register_transform( 

49 nodes.Subscript, 

50 inference_tip(infer_parents_subscript), 

51 _looks_like_parents_subscript, 

52 )