feat: substituir guardar_json pelo utils.save_as_json
Mudei um pouco o save_as_json e talvez tenha ficado um pouco nojento porque não sabia muito bem como fazer uso das EVENT_COLS e STATION_COLS de dentro do utils.py
This commit is contained in:
@@ -137,7 +137,7 @@ def main():
|
|||||||
fname = _get_usr_input("Nome do ficheiro a guardar? ")
|
fname = _get_usr_input("Nome do ficheiro a guardar? ")
|
||||||
if fname is None:
|
if fname is None:
|
||||||
fname = "valores.json"
|
fname = "valores.json"
|
||||||
guardar_json(db, fname)
|
utils.save_as_json(db, fname, EVENT_COLS, STATION_COLS)
|
||||||
else:
|
else:
|
||||||
retInfo = "Base de dados não encontrada!"
|
retInfo = "Base de dados não encontrada!"
|
||||||
|
|
||||||
|
|||||||
@@ -9,9 +9,10 @@ from typing import Any
|
|||||||
import pandas as pd
|
import pandas as pd
|
||||||
|
|
||||||
|
|
||||||
def save_as_json(info: dict[str, Any]) -> bool:
|
def save_as_json(df: pd.DataFrame, fname, event_cols, station_cols) -> bool:
|
||||||
with open("test.json", "w") as fp:
|
info = create_dict_struct(df, event_cols, station_cols)
|
||||||
json.dump(info, fp)
|
with open(fname, "w") as fp:
|
||||||
|
json.dump(info, fp, indent=4)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user