report_miner_status.py

import typer

from src.bobcat_client.client import BobcatClient
from src.bobcat_client.timezone_handler import TimezoneHandler, InvalidTimezoneException
from src.google_sheets.client import GoogleSheetsClient

"""
Usage: report_miner_status.py [OPTIONS] MINER_IP_ADDRESS
                              GOOGLE_SHEETS_CREDENTIALS_FILE_PATH
                              GOOGLE_SHEET_ID

Arguments:
  MINER_IP_ADDRESS                [required]
  GOOGLE_SHEETS_CREDENTIALS_FILE_PATH
                                  [required]
  GOOGLE_SHEET_ID                 [required]

Options:
  --worksheet-id INTEGER
  --clear-worksheet / --no-clear-worksheet
                                  [default: False]
  --timezone TEXT
  --help                          Show this message and exit.
"""

cli = typer.Typer()


@cli.command()
def report_status(miner_ip_address: str, google_sheets_credentials_file_path: str, google_sheet_id: str,
                  worksheet_id: int = None, clear_worksheet: bool = False, timezone: str = None):
    google_sheets_client = GoogleSheetsClient(
        google_sheets_service_account_file_path=google_sheets_credentials_file_path)
    bobcat_client = BobcatClient(miner_ip_address=miner_ip_address)
    # For some reason you have to get `sync_status` first otherwise you get a `rate limit exceeded error`
    _sync_status = bobcat_client.get_sync_status()
    _temperature = bobcat_client.get_temperature()
    if timezone is not None:
        if not TimezoneHandler.is_valid_timezone(timezone):
            raise InvalidTimezoneException(timezone_str=timezone)
        _temperature.convert_timestamp_timezone(to_timezone=timezone)
    miner_report = {**_temperature.__dict__, **_sync_status.__dict__}
    google_sheets_client.write_to_sheet(sheet_id=google_sheet_id, data=miner_report, worksheet_id=worksheet_id,
                                        clear_worksheet=clear_worksheet)
    return None


if __name__ == "__main__":
    cli()