| 12345678910111213141516171819202122232425262728293031 |
- # Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
- # For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
- # Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
- """Astroid hooks for understanding ``boto3.ServiceRequest()``."""
- from astroid import extract_node
- from astroid.manager import AstroidManager
- from astroid.nodes.scoped_nodes import ClassDef
- BOTO_SERVICE_FACTORY_QUALIFIED_NAME = "boto3.resources.base.ServiceResource"
- def service_request_transform(node):
- """Transform ServiceResource to look like dynamic classes."""
- code = """
- def __getattr__(self, attr):
- return 0
- """
- func_getattr = extract_node(code)
- node.locals["__getattr__"] = [func_getattr]
- return node
- def _looks_like_boto3_service_request(node) -> bool:
- return node.qname() == BOTO_SERVICE_FACTORY_QUALIFIED_NAME
- AstroidManager().register_transform(
- ClassDef, service_request_transform, _looks_like_boto3_service_request
- )
|
