|
| 1 | +from __future__ import annotations |
| 2 | + |
1 | 3 | from pathlib import Path |
2 | | -from src.riskops.grc.assessment import RiskAssessmentEngine |
3 | 4 |
|
4 | | -# Test basique |
5 | | -engine = RiskAssessmentEngine() |
| 5 | +from riskops.grc.assessment import RiskAssessmentEngine |
| 6 | + |
| 7 | + |
| 8 | +def main() -> None: |
| 9 | + """Quick manual check for the GRC risk assessment engine.""" |
| 10 | + engine = RiskAssessmentEngine() |
| 11 | + |
| 12 | + csv_path = Path("02-Matrices/risk_matrix.csv").resolve() |
| 13 | + df = engine.load_csv(csv_path) |
| 14 | + print(f"Loaded {len(df)} risk scenarios") |
6 | 15 |
|
7 | | -# CORRECTION : utiliser le chemin absolu depuis la racine |
8 | | -csv_path = Path("02-Matrices/risk_matrix.csv").resolve() |
9 | | -print(f"Loading from: {csv_path}") |
| 16 | + scored_df = engine.calculate_scores(df) |
| 17 | + print("Calculated risk scores") |
| 18 | + print(scored_df[["Asset", "Threat", "Risk", "RiskLevel"]].head()) |
10 | 19 |
|
11 | | -df = engine.load_csv(csv_path) |
12 | | -print(f"✓ Loaded {len(df)} risk scenarios") |
| 20 | + output = engine.export_csv( |
| 21 | + scored_df, |
| 22 | + Path("/tmp/risk_scored.csv"), |
| 23 | + overwrite=True, |
| 24 | + ) |
| 25 | + print(f"Exported to {output}") |
13 | 26 |
|
14 | | -scored_df = engine.calculate_scores(df) |
15 | | -print(f"✓ Calculated risk scores") |
16 | | -print("\nTop 5 risks:") |
17 | | -print(scored_df[['Asset', 'Threat', 'Risk', 'RiskLevel']].head()) |
18 | 27 |
|
19 | | -# Export |
20 | | -output = engine.export_csv(scored_df, Path("/tmp/risk_scored.csv"), overwrite=True) |
21 | | -print(f"\n✓ Exported to {output}") |
| 28 | +if __name__ == "__main__": |
| 29 | + main() |
0 commit comments