Coverage for cuda / pathfinder / _utils / find_sub_dirs.py: 95.45%

1# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3 

4import functools (empty)

5import os (empty)

6import site (empty)

7import sys (empty)

8from collections.abc import Sequence (empty)

9 

10 

11def find_sub_dirs_no_cache(parent_dirs: Sequence[str], sub_dirs: Sequence[str]) -> list[str]: (empty)

12 results = [] 46 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeTSgPbcdQRf

13 for base in parent_dirs: 46 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeTSgPbcdQRf

14 stack = [(base, 0)] # (current_path, index into sub_dirs) 45 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeSgPbcdQRf

15 while stack: 45 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeSgPbcdQRf

16 current_path, idx = stack.pop() 45 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeSgPbcdQRf

17 if idx == len(sub_dirs): 45 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeSgPbcdQRf

18 if os.path.isdir(current_path): 7 ctx1eSgbcdf

19 results.append(current_path) 7 ctx1eSgbcdf

20 continue 7 ctx1eSgbcdf

21 

22 sub = sub_dirs[idx] 44 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOegPbcdQRf

23 if sub == "*": 44 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOegPbcdQRf

24 try: 2 ctx1ef

25 entries = sorted(os.listdir(current_path)) 2 ctx1ef

26 except OSError: 

27 continue 

28 for entry in entries: 2 ctx1ef

29 entry_path = os.path.join(current_path, entry) 2 ctx1ef

30 if os.path.isdir(entry_path): 2 ctx1ef

31 stack.append((entry_path, idx + 1)) 2 ctx1ef

32 else: 

33 next_path = os.path.join(current_path, sub) 44 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOegPbcdQRf

34 if os.path.isdir(next_path): 44 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOegPbcdQRf

35 stack.append((next_path, idx + 1)) 6 ctx1egbcdf

36 return results 46 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeTSgPbcdQRf

37 

38 

39@functools.cache (empty)

40def find_sub_dirs_cached(parent_dirs: Sequence[str], sub_dirs: Sequence[str]) -> list[str]: (empty)

41 return find_sub_dirs_no_cache(parent_dirs, sub_dirs) 46 ctx1ahijklmnopqrstuvwxyzABCDEFGHIJKLMNOeTSgPbcdQRf

42 

43 

44def find_sub_dirs(parent_dirs: Sequence[str], sub_dirs: Sequence[str]) -> list[str]: (empty)

45 return find_sub_dirs_cached(tuple(parent_dirs), tuple(sub_dirs)) 57 ctx1ahijklUVmWnXopYqrZ012stu3vwxyzABCDEFGHIJKL4MNOeTSgPbcdQRf

46 

47 

48def find_sub_dirs_sys_path(sub_dirs: Sequence[str]) -> list[str]: (empty)

49 return find_sub_dirs(sys.path, sub_dirs) 1 ctx1Q

50 

51 

52def find_sub_dirs_all_sitepackages(sub_dirs: Sequence[str]) -> list[str]: (empty)

53 parent_dirs = list(site.getsitepackages()) 50 ctx1ahijklUVmWnXopYqrZ012stu3vwxyzABCDEFGHIJKL4MNOPbcd

54 if site.ENABLE_USER_SITE: 50 ctx1ahijklUVmWnXopYqrZ012stu3vwxyzABCDEFGHIJKL4MNOPbcd

55 user_site = site.getusersitepackages() 3 ctx1bcd

56 if user_site: 3 ctx1bcd

57 # Determine insertion index based on whether we're in a virtual environment (fixes #1716): 

58 # - In venv (PEP 405): venv site-packages should come first, then user-site-packages, 

59 # then system site-packages. Insert at index 1 (after venv, before system). 

60 # - Not in venv (PEP 370): user-site-packages should come before system site-packages. 

61 # Insert at index 0. 

62 # Detect venv by checking if sys.prefix differs from sys.base_prefix 

63 insert_idx = 1 if sys.prefix != sys.base_prefix else 0 3 ctx1bcd

64 parent_dirs.insert(insert_idx, user_site) 3 ctx1bcd

65 return find_sub_dirs(parent_dirs, sub_dirs) 50 ctx1ahijklUVmWnXopYqrZ012stu3vwxyzABCDEFGHIJKL4MNOPbcd