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)