From 471fe118a7d21cd042e6c01fe7d5fe99d649e56e Mon Sep 17 00:00:00 2001 From: Hal Simpson <hasimpson@usgs.gov> Date: Fri, 28 Aug 2015 14:08:08 -0600 Subject: [PATCH] Changed to use url code from timeseriesfactory --- geomagio/iaga2002/IAGA2002Factory.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/geomagio/iaga2002/IAGA2002Factory.py b/geomagio/iaga2002/IAGA2002Factory.py index 43c94f767..9abb5faf2 100644 --- a/geomagio/iaga2002/IAGA2002Factory.py +++ b/geomagio/iaga2002/IAGA2002Factory.py @@ -4,7 +4,7 @@ import obspy.core from .. import ChannelConverter from ..TimeseriesFactory import TimeseriesFactory from ..TimeseriesFactoryException import TimeseriesFactoryException -from ..Url import URL +from ..Util import read_url from IAGA2002Parser import IAGA2002Parser from IAGA2002Writer import IAGA2002Writer @@ -35,8 +35,8 @@ class IAGA2002Factory(TimeseriesFactory): def __init__(self, urlTemplate, observatory=None, channels=None, type=None, interval=None): - TimeseriesFactory.__init__(self, observatory, channels, type, interval) - self.urlTemplate = urlTemplate + TimeseriesFactory.__init__(self, observatory, channels, type, + interval, urlTemplate) def get_timeseries(self, starttime, endtime, observatory=None, channels=None, type=None, interval=None): @@ -72,10 +72,9 @@ class IAGA2002Factory(TimeseriesFactory): interval = interval or self.interval days = self._get_days(starttime, endtime) timeseries = obspy.core.Stream() - url = URL(self.urlTemplate) for day in days: - url_id = url.get_url(observatory, day, type, interval) - iagaFile = url.read_url(url_id) + url_id = self._get_url(observatory, day, type, interval) + iagaFile = read_url(url_id) timeseries += self.parse_string(iagaFile) # merge channel traces for multiple days timeseries.merge() @@ -199,10 +198,9 @@ class IAGA2002Factory(TimeseriesFactory): starttime = starttime or stats.starttime endtime = endtime or stats.endtime days = self._get_days(starttime, endtime) - url = URL(self.urlTemplate) for day in days: - day_filename = url.get_file_from_url( - url.get_url(observatory, day, type, interval)) + day_filename = self._get_file_from_url( + self._get_url(observatory, day, type, interval)) day_timeseries = self._get_slice(timeseries, day, interval) with open(day_filename, 'wb') as fh: self.write_file(fh, day_timeseries, channels) -- GitLab