#!/usr/bin/env python3
# quick test: download one month (1993-01) to verify credentials + subset
import os, time
import copernicusmarine as cm

USERNAME = "lzhenyuan"
PASSWORD = "Nnn5nnwykl@"
DATASET_ID = "cmems_obs-sl_glo_phy-ssh_my_allsat-demo-l4-duacs-0.125deg_P1D-i"
OUT_DIR = "/data/public/lzy/test/_test_month"
os.makedirs(OUT_DIR, exist_ok=True)

t0 = time.time()
try:
    resp = cm.subset(
        dataset_id=DATASET_ID,
        username=USERNAME,
        password=PASSWORD,
        variables=["sla"],
        start_datetime="1993-01-01T00:00:00Z",
        end_datetime="1993-01-31T23:59:59Z",
        service="arco-geo-series",
        output_directory=OUT_DIR,
        output_filename="test_1993_01_sla.nc",
        file_format="netcdf",
        overwrite=True,
        skip_existing=False,
        disable_progress_bar=False,
    )
    print("OK", resp)
except Exception as e:
    print("ERR", type(e).__name__, e)
print("elapsed", round(time.time()-t0,1), "s")
# list output
for f in os.listdir(OUT_DIR):
    p = os.path.join(OUT_DIR, f)
    print(f, os.path.getsize(p))