updated stazioni for python3

This commit is contained in:
putro 2020-04-16 20:39:17 +02:00
parent 8f461db1c0
commit 207565eab2

View File

@ -1,7 +1,7 @@
#!/usr/bin/env python #!/usr/bin/python3
# pylint: skip-file
import os.path import os.path
import tabulate
import pandas as pd import pandas as pd
from sodapy import Socrata from sodapy import Socrata
@ -15,13 +15,16 @@ def get_stazioni(offline=False):
if offline: if offline:
stazioni = pd.read_csv("stazioni.csv") stazioni = pd.read_csv("stazioni.csv")
stazioni_df = pd.DataFrame.from_records(stazioni) stazioni_df = pd.DataFrame.from_records(stazioni)
stazioni_df.drop(columns=["Storico", "Idstazione", "Utm_Nord", "UTM_Est", "lat", "lng", "location", "Limiti amministrativi 2014 delle province di Regione Lombardia", "Limiti amministrativi 2015 delle province di Regione Lombardia"], inplace=True) stazioni_df.drop(columns=["Storico", "Idstazione", "Utm_Nord", "UTM_Est", "lat", "lng", "location",
"Limiti amministrativi 2014 delle province di Regione Lombardia",
"Limiti amministrativi 2015 delle province di Regione Lombardia"], inplace=True)
stazioni_df.columns = [x.lower() for x in stazioni_df.columns] stazioni_df.columns = [x.lower() for x in stazioni_df.columns]
else: else:
client = _connect() client = _connect()
stazioni = client.get("ib47-atvt") stazioni = client.get("ib47-atvt")
stazioni_df = pd.DataFrame.from_records(stazioni) stazioni_df = pd.DataFrame.from_records(stazioni)
stazioni_df.drop(columns=[":@computed_region_6hky_swhk", ":@computed_region_ttgh_9sm5", "utm_est", "utm_nord", "storico", "idstazione", "lat", "lng", "location"], inplace=True) stazioni_df.drop(columns=[":@computed_region_6hky_swhk", ":@computed_region_ttgh_9sm5", "utm_est",
"utm_nord", "storico", "idstazione", "lat", "lng", "location"], inplace=True)
stazioni_df["datastart"] = pd.to_datetime(stazioni_df["datastart"]) stazioni_df["datastart"] = pd.to_datetime(stazioni_df["datastart"])
stazioni_df["datastop"] = pd.to_datetime(stazioni_df["datastop"]) stazioni_df["datastop"] = pd.to_datetime(stazioni_df["datastop"])
return stazioni_df return stazioni_df
@ -33,7 +36,7 @@ def print_stazioni():
stazioni = get_stazioni(offline=True) stazioni = get_stazioni(offline=True)
else: else:
stazioni = get_stazioni() stazioni = get_stazioni()
print(stazioni) print(tabulate.tabulate(stazioni, headers='keys', tablefmt='simple'))
if __name__ == '__main__': if __name__ == '__main__':