feat: log request cookies

This commit is contained in:
debugtalk
2020-05-22 13:28:56 +08:00
parent 988e7e6e78
commit 41a188df3f
2 changed files with 6 additions and 3 deletions

View File

@@ -41,6 +41,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
# record actual request info
request_headers = dict(resp_obj.request.headers)
request_cookies = dict(resp_obj.request._cookies)
request_body = resp_obj.request.body
try:
request_body = json.loads(request_body)
@@ -57,6 +58,7 @@ def get_req_resp_record(resp_obj: Response) -> ReqRespData:
method=resp_obj.request.method,
url=resp_obj.request.url,
headers=request_headers,
cookies=request_cookies,
body=request_body,
)

View File

@@ -13,6 +13,7 @@ BaseUrl = Union[HttpUrl, Text]
VariablesMapping = Dict[Text, Any]
FunctionsMapping = Dict[Text, Callable]
Headers = Dict[Text, Text]
Cookies = Dict[Text, Text]
Verify = bool
Hook = List[Text]
Export = List[Text]
@@ -53,7 +54,7 @@ class Request(BaseModel):
headers: Headers = {}
req_json: Dict = Field({}, alias="json")
data: Union[Text, Dict[Text, Any]] = ""
cookies: Dict[Text, Text] = {}
cookies: Cookies = {}
timeout: int = 120
allow_redirects: bool = True
verify: Verify = False
@@ -108,13 +109,13 @@ class RequestData(BaseModel):
method: MethodEnum = MethodEnum.GET
url: Url
headers: Headers = {}
# TODO: add cookies
cookies: Cookies = {}
body: Union[Text, bytes, Dict, None] = {}
class ResponseData(BaseModel):
status_code: int
cookies: Dict
cookies: Cookies
encoding: Union[Text, None] = None
headers: Dict
content_type: Text