Skip to content

Commit 09fa753

Browse files
Kurt Forresterbashtage
authored andcommitted
updated _sanitize_dates to return Timestamp to employ self.start.quarter
updated if-statement to utilise if-elif-else style
1 parent 004ee3c commit 09fa753

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

pandas_datareader/wb.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def __init__(self, symbols=None, countries=None,
148148
if freq not in freq_symbols:
149149
msg = 'The frequency `{0}` is not in the accepted list.'.format(freq)
150150
raise ValueError(msg)
151-
151+
152152
self.freq = freq
153153
self.countries = countries
154154
self.errors = errors
@@ -165,12 +165,12 @@ def params(self):
165165
return {'date': '{0}M{1:02d}:{2}M{3:02d}'.format(self.start.year,
166166
self.start.month, self.end.year, self.end.month),
167167
'per_page': 25000, 'format': 'json'}
168-
if self.freq == 'Q':
168+
elif self.freq == 'Q':
169169
return {'date': '{0}Q{1}:{2}Q{3}'.format(self.start.year,
170-
divmod(self.start.month-1,3)[0]+1, self.end.year,
171-
divmod(self.end.month-1,3)[0]+1),'per_page': 25000,
170+
self.start.quarter, self.end.year,
171+
self.end.quarter), 'per_page': 25000,
172172
'format': 'json'}
173-
if self.freq is None or self.freq == 'A':
173+
else:
174174
return {'date': '{0}:{1}'.format(self.start.year, self.end.year),
175175
'per_page': 25000, 'format': 'json'}
176176

@@ -372,10 +372,10 @@ def download(country=None, indicator=None, start=2003, end=2005, freq=None,
372372
373373
end: int
374374
Last year of the data series (inclusive)
375-
375+
376376
freq: str
377-
frequency or periodicity of the data to be retrieved (e.g. 'M' for
378-
monthly, 'Q' for quarterly, and 'A' for annual). None defaults to
377+
frequency or periodicity of the data to be retrieved (e.g. 'M' for
378+
monthly, 'Q' for quarterly, and 'A' for annual). None defaults to
379379
annual.
380380
381381
errors: str {'ignore', 'warn', 'raise'}, default 'warn'

0 commit comments

Comments
 (0)