Coverage for src / ai_lls_lib / common.py: 100%

1"""Common utilities for ai-lls-lib.""" 

2 

3import json 

4from decimal import Decimal 

5 

6 

7class DecimalEncoder(json.JSONEncoder): 

8 """JSON encoder that handles DynamoDB Decimal types.""" 

9 

10 def default(self, obj: object) -> object: 

11 if isinstance(obj, Decimal): 

12 return int(obj) if obj % 1 == 0 else float(obj) 

13 return super().default(obj) 

14 

15 

16def extract_area_code(phone: str) -> str: 

17 """Extract 3-digit area code from a phone number string. 

18 

19 Handles E.164 format (+1XXXXXXXXXX) and raw digits. 

20 Returns 'unknown' if fewer than 3 digits. 

21 """ 

22 digits = "".join(c for c in phone if c.isdigit()) 

23 if digits.startswith("1") and len(digits) >= 4: 

24 return digits[1:4] 

25 if len(digits) >= 3: 

26 return digits[:3] 

27 return "unknown"