1515
1616DEFAULT_RETRIES = 3
1717
18- DEFAULTstatus_retries = (500 , 502 , 504 )
18+ DEFAULT_STATUS_RETRIES = (500 , 502 , 504 )
1919
2020
2121def retry (f ):
@@ -41,20 +41,12 @@ def wrapper(*args, **kwargs):
4141class Service (DriverService ):
4242 service = "requests"
4343 default_values = {
44- "data" : ""
44+ "data" : "" ,
45+ "retries" : DEFAULT_RETRIES ,
46+ "status_retries" : DEFAULT_STATUS_RETRIES ,
47+ "propagate_headers" : False ,
4548 }
4649 tracer = None
47- retries = DEFAULT_RETRIES
48- status_retries = DEFAULTstatus_retries
49- _propagate_headers = False
50-
51- def __init__ (self , service , * args , ** kwargs ):
52- """Initialization for trace headers propagation"""
53- super ().__init__ (service , * args , ** kwargs )
54- if self .exists_config ():
55- self .retries = self .config .retries or DEFAULT_RETRIES
56- self .status_retries = self .config .status_retries or DEFAULTstatus_retries
57- self ._propagate_headers = self .config .propagate_headers
5850
5951 def requests (self , session : requests .Session ):
6052 """
@@ -94,7 +86,7 @@ def insert_trace_headers(headers: dict) -> dict:
9486 return headers
9587
9688 @staticmethod
97- def propagate_headers (headers : dict ) -> dict :
89+ def set_propagate_headers (headers : dict ) -> dict :
9890 for k , v in request .headers :
9991 if not headers .get (k ):
10092 headers .update ({k : v })
@@ -112,7 +104,7 @@ def _get_headers(self, headers, propagate_headers=False):
112104 headers = {}
113105
114106 if self ._propagate_headers or propagate_headers :
115- headers = self .propagate_headers (headers )
107+ headers = self .set_propagate_headers (headers )
116108 return headers
117109
118110 @staticmethod
@@ -137,8 +129,8 @@ def parse_response(self, response):
137129
138130 try :
139131 data = response .json ()
140- if self .config . data :
141- data = data .get (self .config . data , {})
132+ if self .data :
133+ data = data .get (self .data , {})
142134 return data
143135 except ValueError :
144136 logger .warning ("Response.content is not a valid json {}" .format (response .content ))
0 commit comments