| | |
|---|
| | | body = TypeVar("body", bytes, str) |
|---|
| | | |
|---|
| | | |
|---|
| | | def convert_cookies_to_dict(cookies): |
|---|
| | | """ |
|---|
| | | 将 mitmproxy 的 cookies 对象转换为可 JSON 序列化的字典 |
|---|
| | | |
|---|
| | | :param cookies: mitmproxy cookies 对象 |
|---|
| | | :return: 可序列化的字典 |
|---|
| | | """ |
|---|
| | | result = {} |
|---|
| | | for key, value in cookies.items(): |
|---|
| | | if hasattr(value, 'items'): |
|---|
| | | result[key] = dict(value.items()) |
|---|
| | | else: |
|---|
| | | result[key] = value |
|---|
| | | return result |
|---|
| | | |
|---|
| | | |
|---|
| | | class RequestInfo(BaseModel): |
|---|
| | | url: str |
|---|
| | | body: str |
|---|
| | |
|---|
| | | response_headers=dict(flow.response.headers), |
|---|
| | | response_content=self.get_response(flow.response), |
|---|
| | | body=self.get_body(flow.request), |
|---|
| | | cookies=dict(flow.response.cookies), |
|---|
| | | request_cookies=dict(flow.request.cookies), |
|---|
| | | cookies=convert_cookies_to_dict(flow.response.cookies), |
|---|
| | | request_cookies=convert_cookies_to_dict(flow.request.cookies), |
|---|
| | | ) |
|---|
| | | ) |
|---|
| | | super().__init__(**kwargs) |
|---|
