optimize/__init__.py:

Added support for gzip ticker data files if they exist.
This commit is contained in:
Michael Smith 2018-01-28 21:57:25 +08:00
parent b44adaa5ab
commit f66958c34f

View File

@ -41,15 +41,13 @@ def load_tickerdata_file(datadir, pair, ticker_interval,
) )
gzipfile = file + '.gz' gzipfile = file + '.gz'
# The file does not exist we download it # If the file does not exist we download it when None is returned.
# If file exists, read the file, load the json # If file exists, read the file, load the json
if os.path.isfile(gzipfile): if os.path.isfile(gzipfile):
with gzip.open(gzipfile) as tickerdata: with gzip.open(gzipfile) as tickerdata:
print("Found gzip file. Using that.")
pairdata = json.load(tickerdata) pairdata = json.load(tickerdata)
elif os.path.isfile(file): elif os.path.isfile(file):
with open(file) as tickerdata: with open(file) as tickerdata:
print("Found json file. Using that.")
pairdata = json.load(tickerdata) pairdata = json.load(tickerdata)
else: else:
return None return None