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
« 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
5"""Astroid hooks for understanding ``boto3.ServiceRequest()``."""
7from astroid import extract_node
8from astroid.manager import AstroidManager
9from astroid.nodes.scoped_nodes import ClassDef
11BOTO_SERVICE_FACTORY_QUALIFIED_NAME = "boto3.resources.base.ServiceResource"
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
25def _looks_like_boto3_service_request(node) -> bool:
26 return node.qname() == BOTO_SERVICE_FACTORY_QUALIFIED_NAME
29AstroidManager().register_transform(
30 ClassDef, service_request_transform, _looks_like_boto3_service_request
31)