-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy path18_security.py
More file actions
63 lines (50 loc) · 2.02 KB
/
18_security.py
File metadata and controls
63 lines (50 loc) · 2.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#!/usr/bin/env python3
"""
Step 18: Security Processing (Thin Orchestrator)
This step orchestrates security processing for GNN models.
Architectural Role:
This is a "thin orchestrator" - a minimal script that delegates core functionality
to the corresponding module (src/security/). It handles argument parsing, logging
setup, and calls the actual processing functions from the security module.
Pipeline Flow:
main.py → 18_security.py (this script) → security/ (modular implementation)
How to run:
python src/18_security.py --target-dir input/gnn_files --output-dir output --verbose
python src/main.py # (runs as part of the pipeline)
Expected outputs:
- Security processing results in the specified output directory
- Comprehensive security reports and summaries
- Actionable error messages if dependencies or paths are missing
- Clear logging of all resolved arguments and paths
If you encounter errors:
- Check that security dependencies are installed
- Check that src/security/ contains security modules
- Check that the output directory is writable
- Verify security configuration and requirements
"""
import sys
from pathlib import Path
# Add src to path for imports
sys.path.insert(0, str(Path(__file__).parent))
from utils.pipeline_template import create_standardized_pipeline_script
# Import module function
try:
from security import process_security
except ImportError:
def process_security(target_dir, output_dir, logger=None, **kwargs) -> bool:
"""Recovery security processing when module unavailable."""
import logging
if logger is None:
logger = logging.getLogger(__name__)
logger.warning("Security module not available - using recovery")
return True
run_script = create_standardized_pipeline_script(
"18_security.py",
process_security,
"Security processing for GNN models",
)
def main() -> int:
"""Main entry point for the security step."""
return run_script()
if __name__ == "__main__":
raise SystemExit(main())