Skip to content

Commit 7ee4a02

Browse files
committed
BUG: Ensure headers is always defined
1 parent 9f289b5 commit 7ee4a02

File tree

1 file changed

+22
-21
lines changed

1 file changed

+22
-21
lines changed

pandas_datareader/base.py

Lines changed: 22 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -47,15 +47,15 @@ class _BaseReader(object):
4747
_format = "string"
4848

4949
def __init__(
50-
self,
51-
symbols,
52-
start=None,
53-
end=None,
54-
retry_count=3,
55-
pause=0.1,
56-
timeout=30,
57-
session=None,
58-
freq=None,
50+
self,
51+
symbols,
52+
start=None,
53+
end=None,
54+
retry_count=3,
55+
pause=0.1,
56+
timeout=30,
57+
session=None,
58+
freq=None,
5959
):
6060

6161
self.symbols = symbols
@@ -72,6 +72,7 @@ def __init__(
7272
self.pause_multiplier = 1
7373
self.session = _init_session(session, retry_count)
7474
self.freq = freq
75+
self.headers = None
7576

7677
def close(self):
7778
"""Close network session"""
@@ -149,7 +150,7 @@ def _get_response(self, url, params=None, headers=None):
149150
"""
150151

151152
# initial attempt + retry
152-
if headers == None:
153+
if headers is None:
153154
headers = self.headers
154155

155156
pause = self.pause
@@ -227,14 +228,14 @@ class _DailyBaseReader(_BaseReader):
227228
"""Base class for Google / Yahoo daily reader"""
228229

229230
def __init__(
230-
self,
231-
symbols=None,
232-
start=None,
233-
end=None,
234-
retry_count=3,
235-
pause=0.1,
236-
session=None,
237-
chunksize=25,
231+
self,
232+
symbols=None,
233+
start=None,
234+
end=None,
235+
retry_count=3,
236+
pause=0.1,
237+
session=None,
238+
chunksize=25,
238239
):
239240
super(_DailyBaseReader, self).__init__(
240241
symbols=symbols,
@@ -300,7 +301,7 @@ def _in_chunks(seq, size):
300301
"""
301302
Return sequence in 'chunks' of size defined by size
302303
"""
303-
return (seq[pos: pos + size] for pos in range(0, len(seq), size))
304+
return (seq[pos : pos + size] for pos in range(0, len(seq), size))
304305

305306

306307
class _OptionBaseReader(_BaseReader):
@@ -334,7 +335,7 @@ def get_put_data(self, month=None, year=None, expiry=None):
334335
raise NotImplementedError
335336

336337
def get_near_stock_price(
337-
self, above_below=2, call=True, put=False, month=None, year=None, expiry=None
338+
self, above_below=2, call=True, put=False, month=None, year=None, expiry=None
338339
):
339340
"""
340341
***Experimental***
@@ -343,7 +344,7 @@ def get_near_stock_price(
343344
raise NotImplementedError
344345

345346
def get_forward_data(
346-
self, months, call=True, put=False, near=False, above_below=2
347+
self, months, call=True, put=False, near=False, above_below=2
347348
): # pragma: no cover
348349
"""
349350
***Experimental***

0 commit comments

Comments
 (0)