Richard Tang
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
check_requirements.py - Batch import checker for quickstart scripts
|
|
|
|
This script checks multiple Python module imports in a single process,
|
|
reducing subprocess spawning overhead significantly on Windows.
|
|
|
|
Usage:
|
|
python scripts/check_requirements.py <module1> <module2> ...
|
|
|
|
Returns:
|
|
JSON object with import status for each module
|
|
Exit code 0 if all imports succeed, 1 if any fail
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
from typing import Dict
|
|
|
|
|
|
def check_imports(modules: list[str]) -> Dict[str, str]:
|
|
"""
|
|
Attempt to import each module and return status.
|
|
|
|
Args:
|
|
modules: List of module names to check
|
|
|
|
Returns:
|
|
Dictionary mapping module name to "ok" or error message
|
|
"""
|
|
results = {}
|
|
|
|
for module_name in modules:
|
|
try:
|
|
# Handle both simple imports and from imports
|
|
if " " in module_name:
|
|
# This shouldn't happen with current usage, but handle it safely
|
|
results[module_name] = "error: invalid module name"
|
|
else:
|
|
# Try to import the module
|
|
__import__(module_name)
|
|
results[module_name] = "ok"
|
|
except ImportError as e:
|
|
results[module_name] = f"error: {str(e)}"
|
|
except Exception as e:
|
|
results[module_name] = f"error: {type(e).__name__}: {str(e)}"
|
|
|
|
return results
|
|
|
|
|
|
def main():
|
|
"""Main entry point."""
|
|
if len(sys.argv) < 2:
|
|
print(json.dumps({"error": "No modules specified"}), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
modules_to_check = sys.argv[1:]
|
|
results = check_imports(modules_to_check)
|
|
|
|
# Print results as JSON
|
|
print(json.dumps(results, indent=2))
|
|
|
|
# Exit with error code if any imports failed
|
|
has_errors = any(status != "ok" for status in results.values())
|
|
sys.exit(1 if has_errors else 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|