def handle(self, **options):
grower = GrowerInfo.objects.get(pk=1)
ds = FewsJdbcDataSource(grower)
_from = datetime(2013, 1, 24, 9, 30, tzinfo=pytz.utc)
to = datetime(2013, 1, 29, 9, 30, tzinfo=pytz.utc)
# get_rain data
for rain_type in ('min', 'mean', 'max'): # rain types
results = ds.get_rain(rain_type, _from, to)
data = convert_timeseries_to_python(results)
if data:
f = open('rain_%s_data.py' % rain_type, 'w')
f.write(DATA_MODULE_START_TEXT)
for dt, value in data:
f.write(" [%s, %s],\n" % (format_datetime(dt), value))
f.write(DATA_MODULE_END_TEXT)
f.close()
# get_fill data
_from = datetime(2012, 12, 29, 9, 30, tzinfo=pytz.utc)
to = datetime(2013, 1, 29, 9, 30, tzinfo=pytz.utc)
results = ds.get_fill(_from, to)
data = convert_timeseries_to_python(results)
if data:
f = open('fill_data.py', 'w')
f.write(DATA_MODULE_START_TEXT)
for dt, value in data:
f.write(" [%s, %s],\n" % (format_datetime(dt), value))
f.write(DATA_MODULE_END_TEXT)
f.close()
class FewsJdbcDataSourceTest(TestCase):
fixtures = ['jdbc_source.json']
def setUp(self):
self.ds = FewsJdbcDataSource()
def test_connection(self):
fews_data = self.ds.get_fill(w30_0, w30_5)
self.assertGreater(len(fews_data), 10)
