# https://github.com/BYU-Hydroinformatics/geoglows-rest-api/blob/master/demos/geoglows_package.ipynb
# %%
import geoglows
import pandas as pd
import param

# %%
# australia_file = "/../opt/anaconda3/envs/py39/lib/python3.9/site-packages/geoglows/geometry/australia-geoglows-comid_lat_lon_z.pickle"
# australia_regions = pd.read_pickle(australia_file)
# australia_regions.to_csv("geoflows_australia.csv")

# %%
p = param.parameters()

# %%
reach_df = pd.read_csv(
    f"{p.local_base_dir}spatial/geoflows_australia_on_coast.csv",
    index_col="fid",
).drop(["to_keep"], axis=1)
reach_df

# %%
data = pd.DataFrame()
for r in reach_df.COMID:
    data_tmp = geoglows.streamflow.forecast_records(r).rename(
        {"streamflow_m^3/s": r}, axis="columns"
    )
    data = pd.concat([data, data_tmp], axis=1)

data

# %%
data.to_csv(f"{p.local_base_dir}source/streamflow.csv")
# %%