gas_run.py

"""
Run a Discord sidebar bot that shows gas price of the Ethereum blockchain
"""
# Example:
# python3 gas_run.py -s etherscan &

from typing import Tuple
def get_gas_from_etherscan(key: str,
                           verbose: bool = False) -> Tuple[int]:
    """
    Fetch gas from Etherscan API
    """
    import requests
    import time
    r = requests.get('https://api.etherscan.io/api',
                     params={'module':'gastracker',
                             'action':'gasoracle',
                             'apikey':key})
    if r.status_code == 200:
        if verbose:
            print('200 OK')
        data = r.json()['result']
        return int(data['SafeGasPrice']), int(data['ProposeGasPrice']), int(data['FastGasPrice'])
    else:
        if verbose:
            print(r.status_code)
        time.sleep(10)

def main(source, verbose=False):
    import yaml
    import discord
    import asyncio

    # 1. Load config
    filename = 'gas_config.yaml'
    with open(filename) as f:
        config = yaml.load(f, Loader=yaml.Loader)

    # 2. Connect w the bot
    client = discord.Client()

    async def send_update(slow, medium, fast):
        nickname = f'🚶{medium} gwei'
        status = f'⚡{fast} 🐌{slow}'
        await client.get_guild(config['guildId']).me.edit(nick=nickname)
        await client.change_presence(activity=discord.Activity(type=discord.ActivityType.watching,
                                                               name=status))
        await asyncio.sleep(config['updateFreq']) # in seconds

    @client.event
    async def on_ready():
        """
        When discord client is ready
        """
        while True:
            # 3. Fetch gas
            if source == 'etherscan':
                gweiList = get_gas_from_etherscan(config['etherscanKey'],
                                                  verbose=verbose)
            else:
                raise NotImplemented('Unsupported source')
            # 4. Feed it to the bot
            await send_update(*gweiList)

    client.run(config['discordBotKey'])

if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser()

    parser.add_argument('-s', '--source',
                        choices=['etherscan'],
                        default='etherscan',
                        help='select API')

    parser.add_argument('-v', '--verbose',
                        action='store_true', # equiv. default is False
                        help='toggle verbose')
    args = parser.parse_args()
    main(source=args.source,
         verbose=args.verbose)