import discord
import asyncio
import os
import signal
import sys

#Set up Client State
CLIENT_TOKEN=os.environ['TOKEN']

#Create Discord client
client = discord.Client()

#Handle shutdown gracefully
def sigterm_handler(signum, frame):
    print("Logging out...") 
    client.logout()
    print('Shutting down...')
    sys.exit(1)

#Register SIGTERM Handler
signal.signal(signal.SIGTERM, sigterm_handler)



@client.event
async def on_ready():
    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('------')

@client.event
async def on_message(message):
    if message.content.startswith('!test'):
        counter = 0
        tmp = await client.send_message(message.channel, 'Calculating messages...')
        async for log in client.logs_from(message.channel, limit=100):
            if log.author == message.author:
                counter += 1

        await client.edit_message(tmp, 'You have {} messages.'.format(counter))
    elif message.content.startswith('!sleep'):
        await asyncio.sleep(5)
        await client.send_message(message.channel, 'Done sleeping')

client.run(CLIENT_TOKEN)
import discord
import asyncio
import os
import signal
import sys

#Set up Client State
CLIENT_TOKEN=os.environ['TOKEN']

#Create Discord client
client = discord.Client()

#Handle shutdown gracefully
async def sigterm_handler(signum, frame):
    print("Logging out...") 
    raise KeyboardInterrupt
    print('Shutting down...')
    sys.exit(1)

#Register SIGTERM Handler
signal.signal(signal.SIGTERM, sigterm_handler)



@client.event
async def on_ready():
    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('------')

@client.event
async def on_message(message):
    if message.content.startswith('!test'):
        counter = 0
        tmp = await client.send_message(message.channel, 'Calculating messages...')
        async for log in client.logs_from(message.channel, limit=100):
            if log.author == message.author:
                counter += 1

        await client.edit_message(tmp, 'You have {} messages.'.format(counter))
    elif message.content.startswith('!sleep'):
        await asyncio.sleep(5)
        await client.send_message(message.channel, 'Done sleeping')

client.run(CLIENT_TOKEN)
Raise KeyboardInterrupt to allow the run to handle logout