Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/astroid/brain/brain_boto3.py: 62%

13 statements  

« prev     ^ index     » next       coverage.py v7.2.7, created at 2023-06-07 06:53 +0000

1# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html 

2# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE 

3# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt 

4 

5"""Astroid hooks for understanding ``boto3.ServiceRequest()``.""" 

6 

7from astroid import extract_node 

8from astroid.manager import AstroidManager 

9from astroid.nodes.scoped_nodes import ClassDef 

10 

11BOTO_SERVICE_FACTORY_QUALIFIED_NAME = "boto3.resources.base.ServiceResource" 

12 

13 

14def service_request_transform(node): 

15 """Transform ServiceResource to look like dynamic classes.""" 

16 code = """ 

17 def __getattr__(self, attr): 

18 return 0 

19 """ 

20 func_getattr = extract_node(code) 

21 node.locals["__getattr__"] = [func_getattr] 

22 return node 

23 

24 

25def _looks_like_boto3_service_request(node) -> bool: 

26 return node.qname() == BOTO_SERVICE_FACTORY_QUALIFIED_NAME 

27 

28 

29AstroidManager().register_transform( 

30 ClassDef, service_request_transform, _looks_like_boto3_service_request 

31)