Replenishing Consumption Account Balance

This script helps you keep a minimum buffer of Units available in your Consumption account.

• It checks your current balance and, if you are below a threshold, it automatically places a buy order for more Units on the market either as a limit order capped at a max price or as a market order for immediate replenishment.

• You can adjust the minimum buffer based on your Consumption needs in every 4 Hour Consumption Window.

Python

import base64
import hashlib
import json
import os
import time

import requests
from nacl.signing import SigningKey

BASE_URL = 'https://trading.api.thegrid.ai'
MIN_BALANCE = 5  # Minimum units to maintain
BUY_QUANTITY = 10
MAX_PRICE = 2.50
MARKET_ID = 'market_788dcbd5-ac68-4c61-acf1-4443beaf2a1c'
ORDER_TYPE = os.environ.get('ORDER_TYPE', 'limit').lower()


class SignatureAuth:
    def __init__(self, private_key_b64, public_key_b64):
        self.private_key = SigningKey(base64.b64decode(private_key_b64)[:32])
        public_key_bytes = base64.b64decode(public_key_b64)
        hash_digest = hashlib.sha256(public_key_bytes).digest()
        self.fingerprint = base64.b64encode(hash_digest).decode().rstrip('=')

    def get_headers(self, method, path, body=''):
        timestamp = str(int(time.time()))
        message = f'{timestamp}{method.upper()}{path}{body}'
        signature = self.private_key.sign(message.encode()).signature
        return {
            'x-thegrid-signature': base64.b64encode(signature).decode(),
            'x-thegrid-timestamp': timestamp,
            'x-thegrid-fingerprint': self.fingerprint,
        }


def build_order_payload(order_type):
    order_type = (order_type or 'limit').lower()
    base = {
        'market_id': MARKET_ID,
        'side': 'buy',
        'type': order_type,
        'quantity': BUY_QUANTITY,
        'time_in_force': 'gtc',
        'client_order_id': f'replenish-{int(time.time())}',
    }

    if order_type == 'limit':
        base['price'] = str(MAX_PRICE)
        return base

    if order_type == 'market':
        return base

    raise ValueError('ORDER_TYPE must be "limit" or "market"')


auth = SignatureAuth(
    os.environ['GRID_TRADING_PRIVATE_KEY'],
    os.environ['GRID_TRADING_PUBLIC_KEY'],
)

# Check current consumption balance
balance_path = '/api/v1/trading/consumption-accounts'
balance_resp = requests.get(f'{BASE_URL}{balance_path}?order_by=created_at', headers=auth.get_headers('GET', balance_path))
balance_resp.raise_for_status()

accounts = balance_resp.json()['data']
current_balance = accounts[0]['available_balance'] if accounts else 0

print(f'Current balance: {current_balance} units')

if current_balance >= MIN_BALANCE:
    print(f'Balance is sufficient (>= {MIN_BALANCE}), no replenishment needed')
else:
    print(f'Balance is low (< {MIN_BALANCE}), placing {ORDER_TYPE} buy order...')

    order_data = build_order_payload(ORDER_TYPE)

    order_path = '/api/v1/trading/orders'
    body = json.dumps(order_data)
    headers = auth.get_headers('POST', order_path, body)
    headers['Content-Type'] = 'application/json'

    order_resp = requests.post(f'{BASE_URL}{order_path}', data=body, headers=headers)
    order_resp.raise_for_status()

    print(f"Order placed ({ORDER_TYPE}): {order_resp.json()['data']['order_id']}")

JavaScript

Go