From 885db87dea7912fcbcef00dbe295763845b2cff8 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Mon, 13 Apr 2015 01:10:48 -0400 Subject: [PATCH] MAINT: Use logger instead of printing in loader.py Makes it easier to filter logs when they're not desired. --- zipline/data/loader.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/zipline/data/loader.py b/zipline/data/loader.py index 9b6d7d21..24abb5ea 100644 --- a/zipline/data/loader.py +++ b/zipline/data/loader.py @@ -158,10 +158,13 @@ def load_market_data(trading_day=trading_day_nyse, try: saved_benchmarks = pd.Series.from_csv(bm_filepath) except (OSError, IOError): - print(""" -data files aren't distributed with source. -Fetching data from Yahoo Finance. -""".strip()) + logger.info( + "No cache found at {path}. " + "Downloading benchmark data for '{symbol}'.", + symbol=bm_symbol, + path=bm_filepath, + ) + dump_benchmarks(bm_symbol) saved_benchmarks = pd.Series.from_csv(bm_filepath) @@ -202,10 +205,13 @@ Fetching data from Yahoo Finance. try: saved_curves = pd.DataFrame.from_csv(tr_filepath) except (OSError, IOError): - print(""" -data files aren't distributed with source. -Fetching data from {0} -""".format(source).strip()) + logger.info( + "No cache found at {path}. " + "Downloading treasury data from {source}.", + path=tr_filepath, + source=source, + ) + dump_treasury_curves(module, filename) saved_curves = pd.DataFrame.from_csv(tr_filepath)