我正在写工作,以从API下载一些报告。有一个外部python包,只需使用即可为我完成此操作:
from module import Api
api = Api(token='token')
api.download_data(save_to='file_name.csv')
但是此Api
对于呼叫者IP地址至关重要(某些报告仅适用于特定位置),因此我被迫通过VPN或袜子代理来更改我的位置。
我需要从不同位置下载的报告很少,所以我决定使用代理。
[我提到,我的module
在后台使用requests
库来检索数据。
我如何强制请求库将我的代理用于外部功能?
事实证明,这可以通过老式的装饰器非常简单地实现:
# config.py
HTTP_METHODS = ['get', 'post', 'put', 'delete', 'head', 'patch', 'options']
# helper.py
import requests
from config import HTTP_METHODS
def _add_proxy(http_req, country):
# retrieve proxy settings for the country
proxy = get_proxy(country)
def wrapper(*args, **kwargs):
kwargs['proxies'] = proxy
return http_req(*args, **kwargs)
return wrapper
def patch_requests_with_proxy(country):
for m in HTTP_METHODS:
setattr(requests, m, _add_proxy(getattr(requests, m), country))
我的新脚本的用法将是:
from module import Api
from helper import patch_requests_with_proxy
api = Api(token='token')
patch_requests_with_proxy('ContryName1')
api.download_data(save_to='file_name1.csv')
patch_requests_with_proxy('ContryName2')
api.download_data(save_to='file_name2.csv')
我的秘诀是工作基础,所以我不需要某种unpatch功能。