client_clock_read.py

#!/usr/bin/env python3
"""
Client for reading the clock time from an EMLite meter.

Configuration:
  - Reads from .env

Usage:
  uv run client_clock_read.py EML1411222333
"""

import argparse
import logging
import os
import sys

from dotenv import load_dotenv

from simt_emlite.mediator.api_core import EmliteMediatorAPI

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)


def load_config() -> dict:
    load_dotenv()

    mediator_server = os.environ.get("MEDIATOR_SERVER")
    if not mediator_server:
        logger.error("MEDIATOR_SERVER not set in .env")
        sys.exit(1)

    return {
        "mediator_server": mediator_server,
    }


def clock_time_read(serial: str):
    """
    Read the current clock time from a meter.

    Args:
        serial: Meter serial number

    Returns:
        datetime object with UTC timezone

    Raises:
        Exception: If connection fails or meter communication fails
    """

    config = load_config()
    mediator_address = config["mediator_server"]
    logger.info(f"Connecting to mediator at {mediator_address}")

    try:
        # Initialize the API client
        client = EmliteMediatorAPI(mediator_address=mediator_address)

        logger.info(f"Reading clock time from meter {serial}")
        
        # Use the high-level API method
        date_obj = client.clock_time_read(serial)

        logger.info(f"Successfully read time: {date_obj.isoformat()}")
        return date_obj

    except Exception as e:
        logger.error(f"Unexpected error: {e}")
        sys.exit(1)


def main():
    parser = argparse.ArgumentParser(description="Read clock time from an EMLite meter")
    parser.add_argument("serial", help="Meter serial number (e.g., EML1411222333)")
    args = parser.parse_args()

    clock_time_read(args.serial)


if __name__ == "__main__":
    main()