How to use get_current_request_account_id method in localstack

Best Python code snippet using localstack_python

generic_proxy.py

Source:generic_proxy.py

...354    @classmethod355    def get(cls: Type[T], region: str = None, account_id: str = None) -> T:356        cls.initialise()357        region = region or cls.get_current_request_region()358        account_id = account_id or cls.get_current_request_account_id()359        # Class attributes in RegionBackend need two types of data isolation:360        # - Class attributes like `_ACCOUNT_BACKENDS` must refer to the same mutable object across ALL instances.361        # - Class attributes defined in subclasses represent data common across regions but same account IDs362        # To enable this, we keep copies of this class in `_ACCOUNT_CLS` by account IDs and instantiate that363        # copy of the class for different regions and same account ID364        if MULTI_ACCOUNTS and account_id not in cls._ACCOUNTS_CLS:365            # @classmethod descriptors will be part of a cls.__dict__, but they cannot be pickled by deepcopy366            data = {k: v for k, v in cls.__dict__.items() if not inspect.ismethoddescriptor(v)}367            cls_dict = copy.deepcopy(data)368            cls_dict["_ACCOUNTS_CLS"] = cls._ACCOUNTS_CLS369            cls_dict["_ACCOUNT_BACKENDS"] = cls._ACCOUNT_BACKENDS370            cls._ACCOUNTS_CLS[account_id] = type(cls.__name__, cls.__bases__, cls_dict)371        regions = cls.regions()372        backend = regions.get(region)373        if not backend:374            if MULTI_ACCOUNTS:375                backend = cls._ACCOUNTS_CLS[account_id]()376            else:377                backend = cls()378            backend.name = region379            backend.account_id = account_id380            regions[region] = backend381        return regions[region]382    @classmethod383    def initialise(cls):384        if not hasattr(cls, "_ACCOUNT_BACKENDS"):385            cls._ACCOUNT_BACKENDS = defaultdict(lambda: dict())386        if not hasattr(cls, "_ACCOUNTS_CLS"):387            cls._ACCOUNTS_CLS = dict()388        if not hasattr(cls, "REGIONS"):389            cls.REGIONS = dict()390    @classmethod391    def regions(cls: Type[T]) -> Dict[str, T]:392        """Return a map of all regions and backend instances."""393        cls.initialise()394        account_id = cls.get_current_request_account_id()395        return cls._ACCOUNT_BACKENDS[account_id]396    @classmethod397    def get_current_request_region(cls) -> str:398        return aws_stack.get_region()399    @classmethod400    def get_current_request_account_id(cls) -> str:401        return get_aws_account_id()402    @classmethod403    def reset(cls):404        """Reset the (in-memory) state of this service region backend."""405        cls.initialise()406        cls.REGIONS = {}407        account_id = cls.get_current_request_account_id()408        cls._ACCOUNT_BACKENDS[account_id] = {}409# ---------------------410# PROXY LISTENER UTILS411# ---------------------412def append_cors_headers(413    request_headers: Dict = None, response: Union[Response, LambdaResponse] = None414):415    # use this config to disable returning CORS headers entirely (more restrictive security setting)416    if config.DISABLE_CORS_HEADERS:417        return418    # Note: Use "response is None" here instead of "not response"419    headers = {} if response is None else response.headers420    # In case we have LambdaResponse, copy multivalue headers to regular headers, since421    # CaseInsensitiveDict does not support "__contains__" and it's easier to deal with...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.