用python写一个接口自动化请求框架
时间: 2024-06-09 19:05:17 浏览: 155
好的,以下是一个简单的Python接口自动化请求框架的示例代码:
```python
import requests
class ApiRequest:
def __init__(self, base_url):
self.base_url = base_url
def get(self, endpoint, params=None, headers=None):
url = self.base_url + endpoint
response = requests.get(url, params=params, headers=headers)
return response
def post(self, endpoint, data=None, json=None, headers=None):
url = self.base_url + endpoint
response = requests.post(url, data=data, json=json, headers=headers)
return response
def put(self, endpoint, data=None, json=None, headers=None):
url = self.base_url + endpoint
response = requests.put(url, data=data, json=json, headers=headers)
return response
def delete(self, endpoint, headers=None):
url = self.base_url + endpoint
response = requests.delete(url, headers=headers)
return response
```
在这个代码中,我们定义了一个名为`ApiRequest`的类,它接受一个基本URL作为参数,并提供了四种HTTP请求方法:get、post、put和delete。每个方法都需要一个endpoint参数,用于指定要请求的API端点。其他参数将作为请求的参数或头部发送。
现在我们可以创建一个`ApiRequest`对象,并使用它来发送请求:
```python
api = ApiRequest('https://api.example.com')
response = api.get('/users', params={'page': 1})
if response.status_code == 200:
print(response.json())
else:
print('Error:', response.status_code)
```
这个例子中,我们实例化一个`ApiRequest`对象,并使用`get`方法请求了`/users`端点,同时将`page`参数设置为1。如果响应状态码为200,则打印响应的JSON数据,否则打印错误消息。
阅读全文