代码之家 › 专栏 › 技术社区 › MvdD

用pre和post方法调用装饰请求模块调用

python-decorators python-3.x python

MvdD · 技术社区 · 6 年前

我有一个班叫 Client requests 与服务交互的模块。它有如下方法:

def get_resource(self, url, headers):
    response = requests.get(url, headers, auth=self.auth)
    return response

现在我想在每次调用requests模块之前和之后调用一些方法。类似于:

def get_resource(self, url, headers):
    self.add_request_header(headers)
    response = requests.get(url, headers, auth=self.auth)
    self.process_response_headers()
    return response

我很难找到一个不用重写的方法 顾客

def get_resource(self, url, headers):
    response = self.__get(url, headers, auth=self.auth)
    return response

def __get(self, headers, auth):
    self.add_request_header(headers)
    response = requests.get(url, headers, auth=self.auth)
    self.process_response_headers()
    return response

但这需要我更改所有的调用站点并复制请求模块的功能。我曾试图使用decorators将这些方法调用添加到请求模块函数中,但一直纠结于如何将self传递给decorator。

我相信在Python中有一种优雅的方法可以做到这一点。

2 回复 | 直到 6 年前

Rahul Raut 6 年前

你可以用猴子补丁。请阅读: Python: Monkeypatching a method of an object

    import requests
    def get(self, url, params=None, **kwargs):
        self.add_request_header(self.headers)
        response = requests.get(url, self.headers, auth=self.auth)
        self.process_response_headers()
    setattr(requests.Session, 'get', requests.Session.get)
    s = requests.Session()

edilio 6 年前

我认为在这种情况下,decorators不会像听起来那么好,而OOP是解决问题的更好方法。您可以使用基类客户端:

class Client(object):
    def __init__(self, auth):
        self.auth = auth

    def add_request_header(self, headers):
        pass

    def process_response_headers(self):
        pass

    def get_resource(self, url, headers):
        self.add_request_header(headers)
        response = requests.get(url, headers, auth=self.auth)
        self.process_response_headers()
        return response

并用其他实现创建另一个子类 add_request_header 和/或 process_response_headers

推荐文章

James Adams · Python:如何在单元(nose)测试期间忽略修饰符?

7 年前

sokeefe · 初始化类中的Decorator变量

7 年前

user1050619 · Python将self传递给decorator

7 年前

oulenz · 如何使新的装饰器在类中可用,而不显式导入它们?

7 年前

Arun Kaliraja Baskaran · 如何在运行时将参数传递给装饰器?

7 年前

user8224771 · python中decorator中的foo=bar(foo)和something=bar(foo)有什么区别?

7 年前

spencer · 为什么有必要将self作为方法装饰器的参数?

7 年前

Jorge Arévalo · 如何检查函数是否用特定的装饰器装饰?

7 年前

jrk0414 · 在python中使用修饰符设置全局变量

8 年前

Brendan Abel · 在函数属性中访问self

8 年前