import qcodes as qc
from plottr.data.datadict_storage import DataDict, DDH5Writer

station = qc.Station()
data = DataDict(
    time=dict(unit="s"),
    voltage=dict(unit="V", axes=["time"]),
)

with DDH5Writer(data) as writer:
    writer.add_tag(["tag1", "tag2"])
    writer.backup_file(__file__)
    writer.save_text("note.md", "some note")
    writer.save_dict("station_snapshot.json", station.snapshot())
    writer.add_data(
        time=[1, 2, 3],
        voltage=[4, 5, 6],
    )