bypass arguments in other class methods

bypass_arguments_in_other_class_methods.py

"""
>>> api_instance = BypassApiKey(RemoteServerApi())
>>> api_instance.method_that_need_access_token(1, 2, 3)
[(1, 2, 3), {'access_token': 'test'}]
"""
import functools
ACCESS_TOKEN = 'test'


class RemoteServerApi:
    def method_that_need_access_token(self, *args, **kwargs):
        if 'access_token' in kwargs:
            return [args, kwargs]
        else:
            raise Exception('I will not work without access_token')


def bypass_api_key(func):
    @functools.wraps(func)
    def wrapper_decorator(*args, **kwargs):
        kwargs.update(dict(access_token=ACCESS_TOKEN))
        value = func(*args, **kwargs)
        return value
    return wrapper_decorator


class BypassApiKey:
    def __init__(self, class_instance):
        self.class_instance = class_instance

    def __getattr__(self, item):
        return bypass_api_key(getattr(self.class_instance, item))