Check SMS Balance

import requests

# Configuration
# Get your API key from: https://roycetalk.com/
# Format: XXXX_XXXXXXXXXXXXXXXXXXXXXXXX_
API_URL = "https://roycetalk.com/api/v1/sms-api/balance/"
BEARER_TOKEN = "XXXX_XXXXXXXXXXXXXXXXXXXXXXXX_"

def check_balance():
    """
    Check current SMS balance via RoyceBulkSMS API
    
    Returns:
        dict: API response with balance information
    """
    
    # Request headers with Bearer token
    headers = {
        "Authorization": f"Bearer {BEARER_TOKEN}",
    }
    
    try:
        # Make GET request
        response = requests.get(API_URL, headers=headers)
        
        # Check if request was successful
        if response.status_code == 200:
            result = response.json()
            balance = result['data']
            
            print("✓ Balance retrieved successfully!")
            print(f"Current Balance: {balance['current_balance']} units")
            print(f"SMS Rate: KES {balance['sms_rate']} per unit")
            print(f"Balance Value: KES {balance['balance_value']}")
            print(f"Currency: {balance['currency']}")
            
            return result
            
        elif response.status_code == 403:
            error = response.json()
            print(f"✗ Authentication Error: {error['error']['message']}")
            return None
            
        else:
            print(f"✗ Unexpected Error: {response.status_code}")
            print(response.text)
            return None
            
    except requests.exceptions.RequestException as e:
        print(f"✗ Request failed: {str(e)}")
        return None

# Example usage
if __name__ == "__main__":
    balance_info = check_balance()
    
    if balance_info:
        # Check if balance is sufficient for campaign
        units_needed = 100
        current_balance = balance_info['data']['current_balance']
        
        if current_balance >= units_needed:
            print(f"\n✓ Sufficient balance for {units_needed} messages")
        else:
            shortage = units_needed - current_balance
            print(f"\n✗ Insufficient balance. Need {shortage} more units")