1. import os
  2. import platform
  3. import subprocess
  4. import json
  7. def filter_isolated_environments(root_dir, min_python_version=3.6):
  8. """
  9. Filters directories for isolated environments (e.g., venv, conda)
  10. with support for older Python versions.
  12. Args:
  13. root_dir (str): The root directory to search.
  14. min_python_version (int): The minimum Python version supported.
  16. Returns:
  17. list: A list of dictionaries, where each dictionary represents
  18. an isolated environment and includes its path and Python version.
  19. Returns an empty list if no suitable environments are found.
  20. """
  22. isolated_environments = []
  24. for root, dirs, files in os.walk(root_dir):
  25. for dir_name in dirs:
  26. potential_env_path = os.path.join(root, dir_name)
  28. # Check for common environment directory names
  29. if dir_name.endswith('venv') or dir_name == 'env' or dir_name == 'conda':
  30. try:
  31. # Check Python version inside the environment
  32. python_version = get_python_version(potential_env_path)
  34. if python_version and int(python_version.split('.')[0]) >= min_python_version:
  35. isolated_environments.append({
  36. 'path': potential_env_path,
  37. 'python_version': python_version
  38. })
  39. except Exception as e:
  40. # Handle potential errors (e.g., permission issues)
  41. print(f"Error processing {potential_env_path}: {e}")
  43. return isolated_environments
  46. def get_python_version(env_path):
  47. """
  48. Retrieves the Python version within a given environment.
  50. Args:
  51. env_path (str): The path to the environment directory.
  53. Returns:
  54. str: The Python version string (e.g., "3.7.12") or None if not found.
  55. """
  56. try:
  57. # Check for pyvenv.cfg (venv)
  58. if os.path.exists(os.path.join(env_path, 'pyvenv.cfg')):
  59. with open(os.path.join(env_path, 'pyvenv.cfg'), 'r') as f:
  60. for line in f:
  61. if "python" in line.lower():
  62. return line.split('=')[1].strip()
  64. # Check for conda environment file
  65. if os.path.exists(os.path.join(env_path, 'conda-meta', 'Python-version')):
  66. with open(os.path.join(env_path, 'conda-meta', 'Python-version'), 'r') as f:
  67. return f.read().strip()
  69. # Check for activate script (more robust)
  70. if os.path.exists(os.path.join(env_path, 'Scripts', 'activate')):
  71. try:
  72. result = subprocess.run(['python', '-c', 'import sys; print(sys.version)'], cwd=env_path, capture_output=True, text=True, check=True)
  73. return result.stdout.strip()
  74. except subprocess.CalledProcessError:
  75. pass # Fallback if python command fails within the environment
  78. return None # Python version not found
  79. except Exception as e:
  80. print(f"Error getting Python version for {env_path}: {e}")
  81. return None
  84. if __name__ == '__main__':
  85. # Example usage:
  86. root_directory = '.' # Replace with your root directory
  87. environments = filter_isolated_environments(root_directory)
  89. if environments:
  90. print(json.dumps(environments, indent=4)) # Print as JSON
  91. else:
  92. print("No suitable isolated environments found.")

Add your comment