Coverage for archdocs/features/http_clients/parser.py: 100%

1import re as py_re 

2import types 

3import typing 

4 

5from archdocs import prefilter, settings 

6from archdocs.features.http_clients.const import HTTPClientEnum, HTTPClientFeatures 

7 

8 

9_CLIENT_PATTERNS: typing.Final = types.MappingProxyType( 

10 { 

11 one_client: py_re.compile(rf"\b(?:from|import)\s+{one_client.value}\b", flags=settings.TYPICAL_RE_FLAGS) 

12 for one_client in HTTPClientEnum 

13 }, 

14) 

15_ASYNC_DETECTION_PATTERN: typing.Final = py_re.compile( 

16 r"\b(?:httpx\.AsyncClient|aiohttp\.ClientSession)\b", 

17 flags=settings.TYPICAL_RE_FLAGS, 

18) 

19_EMPTY_FEATURES: typing.Final = HTTPClientFeatures( 

20 clients_used=frozenset(), 

21 async_used=False, 

22) 

23 

24 

25def find_http_client_features(raw_source: str) -> HTTPClientFeatures: 

26 lowered_source: typing.Final = raw_source.lower() 

27 clients_found: typing.Final = { 

28 one_client.value 

29 for one_client, one_pattern in _CLIENT_PATTERNS.items() 

30 if prefilter.contains_any_literal(lowered_source, (one_client.value,)) and one_pattern.search(raw_source) 

31 } 

32 if not clients_found: 

33 return _EMPTY_FEATURES 

34 return HTTPClientFeatures( 

35 clients_used=frozenset(clients_found), 

36 async_used=bool(_ASYNC_DETECTION_PATTERN.search(raw_source)), 

37 )