Coverage for lintro / ai / enums / risk_level.py: 100%

1"""Risk level enumeration for AI fix suggestions.""" 

2 

3from __future__ import annotations 

4 

5from enum import auto 

6 

7from lintro.enums.hyphenated_str_enum import HyphenatedStrEnum 

8 

9 

10class RiskLevel(HyphenatedStrEnum): 

11 """Risk classification for AI fix suggestions.""" 

12 

13 SAFE_STYLE = auto() 

14 BEHAVIORAL_RISK = auto() 

15 

16 def to_severity_label(self, *, sarif: bool = False) -> str: 

17 """Map risk level to a severity label. 

18 

19 Args: 

20 sarif: When True, return ``"note"`` for safe-style fixes 

21 (SARIF format). When False, return ``"notice"`` 

22 (GitHub Actions annotation format). 

23 

24 Returns: 

25 One of ``"warning"``, ``"note"``, or ``"notice"``. 

26 """ 

27 if self == RiskLevel.SAFE_STYLE: 

28 return "note" if sarif else "notice" 

29 return "warning"