import asyncio

from google.protobuf.json_format import MessageToJson
import storage

async def main() -> None:
    devices = ["/dev/nvme0", "/dev/sda"]
    monitor = storage.Monitor(devices)
    await monitor.initialize()

    resp = await monitor.poll()

    for device in resp:
        json = MessageToJson(device, including_default_value_fields=True)
        print(json)

if __name__ == "__main__":
    loop = asyncio.new_event_loop()
    loop.run_until_complete(main())