#!/usr/bin/env python3
# Uses stdlib only
import pytz
import pprint
import datetime
import struct
import requests
import base64


def main():
    request_data = {
        "jsonrpc": "2.0",
        "id": 1,
        "method": "getAccountInfo",
        "params": [
            "SysvarC1ock11111111111111111111111111111111",
            {"encoding": "base64"},
        ],
    }

    url = "https://api.mainnet-beta.solana.com"

    resp = requests.post(
        url,
        headers={
            'Content-Type': 'application/json',
        },
        json=request_data,
    )
    if not resp.ok:
        resp.raise_for_status()

    data, encoding = resp.json()['result']['value']['data']
    raw_bytes = base64.b64decode(data)

    # <QqQQq per: https://docs.rs/solana-program/1.8.3/solana_program/clock/struct.Clock.html
    slot, epoch_start_timestamp, epoch, leader_schedule_epoch, unix_timestamp = struct.unpack(
        "<QqQQq", raw_bytes,
    )
    return {
        "slot": slot,
        "epoch_start_timestamp": datetime.datetime.fromtimestamp(epoch_start_timestamp, pytz.UTC),
        "epoch": epoch,
        "leader_schedule_epoch": leader_schedule_epoch,
        "unix_timestamp": datetime.datetime.fromtimestamp(unix_timestamp, pytz.UTC),
    }


if __name__ == '__main__':
    pprint.pprint(main())