import os.path
def get_package_path(package_name: str, site_package_path: str):
# Determine if there is a site-packages folder in site_package_path.
# If not, report an error. If there is, return the absolute path of the site-packages folder.
if not os.path.exists(site_package_path):
raise ValueError(
f"Python environment not found: {site_package_path}. Please check the path."
)
if not os.path.exists(os.path.join(site_package_path, package_name)):
raise ValueError(
f"Package {package_name} not found in {site_package_path}. Please check the package name."
)
return os.path.join(site_package_path, package_name)
# Define the directory names to be filtered
IGNORE_DIRS = {
"__pycache__",
".git",
".idea",
".vscode",
"build",
"dist",
"egg-info",
".pytest_cache",
".mypy_cache",
".tox",
}
# Define the file extensions to be filtered
IGNORE_EXTS = {".pyc", ".pyo", ".pyd", ".so", ".dll", ".egg-info", ".DS_Store"}
def should_ignore(path, is_dir=False):
"""Determine whether this path should be ignored"""
if is_dir:
return os.path.basename(path) in IGNORE_DIRS
else:
return os.path.splitext(path)[1] in IGNORE_EXTS
def list_directory_contents(directory):
"""Recursively list all contents in a directory and return a list instead of printing."""
result = [] # The list to store the results
try:
# Get the base name of the directory, which is used to construct the relative path
base_dir = os.path.basename(os.path.abspath(directory))
# Traverse the directory tree
for root, dirs, files in os.walk(directory):
# Calculate the path of the current sub-directory relative to the input directory
relative_path = os.path.relpath(root, directory)
# If it is the root directory, relative_path will be '.', and special handling is required.
if relative_path == ".":
current_dir = base_dir
else:
current_dir = os.path.join(base_dir, relative_path)
# Filter out unwanted directories (modify dirs before traversing)
dirs[:] = [d for d in dirs if not should_ignore(d, is_dir=True)]
# Add subdirectories to the result list
result.append(f"{current_dir}/")
# Add files in subdirectories to the result list (filter out unwanted files)
for file in files:
if not should_ignore(file):
result.append(f"{os.path.join(current_dir, file)}")
except NotADirectoryError:
return f"Error: '{directory}' is not a valid directory."
except PermissionError:
return f"Error: No permission to access directory '{directory}'."
except FileNotFoundError:
return f"Error: Directory '{directory}' does not exist."
except Exception as e:
return f"An unknown error occurred: {e}"
return result # Return the result list
