date = lambda d: datetime.strptime(d, "%Y-%m-%d")
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--inicio", "-i", type=date, help="Data inicial de captura: yyyy-mm-dd")
parser.add_argument("--fim", "-f", type=date, help="Data final de captura: yyyy-mm-dd")
parser.add_argument("--codigo", "-c", choices=codes, metavar='ICAO', help="Codigo da estação" )
args = parser.parse_args()
station, start, end = args.codigo, args.inicio, args.fim
data = []
i = 0
stations = codes if station == 'all' else [station]
for station in stations:
if station == 'all':
continue
for date in date_generator(start, end):
if not check_day(date, station):
print("Date {} has already been captured from station {} ".format(date, station))
continue
print("Fetching data from {} at {}.".format(station, date))
res = capture(station, date)
print(res)
data.append(res)
if i % 10 == 0:
save(data, schema='Municipio', table='Clima_wu')
data = []
time.sleep(1)
save(data, schema='Municipio', table='Clima_wu')
def test_capture_return_type(self):
station_code = 'SBAF'
date = datetime(2015, 11, 2)
data = capture(station_code, date)
self.assertIsInstance(data, dict)
type=date,
help="Data final de captura: yyyy-mm-dd")
parser.add_argument("--codigo",
"-c",
choices=codes,
metavar='ICAO',
help="Codigo da estação")
args = parser.parse_args()
station, start, end = args.codigo, args.inicio, args.fim
data = []
i = 0
stations = codes if station == 'all' else [station]
for station in stations:
for date in date_generator(start, end):
if not check_day(date, station):
print("Date {} has already been captured from station {} ".format(
date, station))
continue
print("Fetching data from {} at {}.".format(station, date))
res = capture(station, date)
print(res)
data.append(res)
if i % 10 == 0:
save(data, schema='Municipio', table='Clima_wu')
data = []
time.sleep(1)
save(data, schema='Municipio', table='Clima_wu')
