Use custom Response on FlareRequest
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
parent
cb5ecf9a29
commit
068aebdaaf
@ -1,5 +1,5 @@
|
||||
from urllib import parse
|
||||
from requests import RequestException, Session, post
|
||||
from requests import RequestException, Response, Session, post
|
||||
from .config import CLOUDPROXY_ENDPOINT, REQUESTS_TIMEOUT
|
||||
|
||||
|
||||
@ -35,13 +35,16 @@ class FlareRequests(Session):
|
||||
solution = response.json()
|
||||
|
||||
if "solution" in solution:
|
||||
response.status_code = solution["solution"]["status"]
|
||||
response.headers = solution["solution"]["headers"]
|
||||
response.raw = solution["solution"]["response"]
|
||||
response.url = url
|
||||
response.cookies = solution["solution"]["cookies"]
|
||||
resolved = Response()
|
||||
|
||||
return response
|
||||
resolved.raw = solution["solution"]["response"]
|
||||
resolved.status_code = solution["solution"]["status"]
|
||||
resolved.headers = solution["solution"]["headers"]
|
||||
resolved.url = url
|
||||
resolved.reason = solution["status"]
|
||||
resolved.cookies = solution["solution"]["cookies"]
|
||||
|
||||
return resolved
|
||||
|
||||
raise RequestException(response)
|
||||
except RequestException:
|
||||
|
Reference in New Issue
Block a user