47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import requests
|
||
|
||
from api.http import Headers as BaseHeaders
|
||
from logger import LOGGER
|
||
|
||
|
||
class Headers(BaseHeaders):
|
||
def __init__(self):
|
||
"""
|
||
初始化Headers类
|
||
- `self.get_headers_url`: 获取token的URL地址，根据实际环境修改。
|
||
- `self.timeout`: 请求超时时间，设置为5秒。
|
||
- `interval`: 定时刷新headers的时间间隔，设置为10分钟（60秒 * 10）。
|
||
"""
|
||
self.get_headers_url = None
|
||
self.timeout = 5
|
||
interval = 60 * 10
|
||
super().__init__(interval)
|
||
|
||
def _generate_headers(self):
|
||
"""
|
||
生成请求头的方法，_generate_headers方法名不允许修改
|
||
通过向指定的URL发送GET请求获取token，并将token添加到请求头中
|
||
:return: 请求头字典
|
||
"""
|
||
try:
|
||
# 定义请求参数
|
||
params = {
|
||
'arg1': 'xxx',
|
||
'arg2': 'xxx'
|
||
}
|
||
if self.get_headers_url is not None:
|
||
# 发送GET请求获取token
|
||
resp = requests.get(self.get_headers_url, params=params, timeout=self.timeout)
|
||
if resp.status_code == 200:
|
||
token = resp.text
|
||
headers = {
|
||
'authorization': 'Bearer {}'.format(token)
|
||
}
|
||
return headers
|
||
else:
|
||
LOGGER.error('Get headers failed')
|
||
return None
|
||
except:
|
||
LOGGER.exception('_generate_headers')
|
||
return None
|