BaseballBot/GhostBallBot/discord_client/client.py

42 lines
1.1 KiB
Python
Raw Normal View History

2022-09-24 02:40:35 +00:00
#!/usr/bin/env python3
# Copyright 2022 - c0de <c0de@c0de.dev>
# Licensed under the MIT License (https://opensource.org/licenses/MIT)
import sys
import discord
# Import game functions
2022-10-25 01:00:39 +00:00
sys.path.append("..")
2022-09-24 02:40:35 +00:00
import game
2022-10-25 01:00:39 +00:00
class GhostBallClient(discord.Client):
2022-09-24 02:40:35 +00:00
def __init__(self, *args, **kwargs):
super(GhostBallClient, self).__init__(*args, **kwargs)
with game.Game() as self.game:
self.game.discord = self
2022-09-24 02:40:35 +00:00
async def on_ready(self):
print("Logged on as", self.user)
async def on_message(self, message):
# Don't respond to ourself
if message.author == self.user:
return
# Bot health check
if message.content == "ping":
await message.channel.send("pong")
# Game commands
2022-10-25 01:00:39 +00:00
if message.content.startswith("!"):
2022-09-24 02:40:35 +00:00
firstword = message.content[1:].split()[0]
# Determine if the first word is a command, and run it
for command, function in self.game.commands:
if firstword == command:
self.game.message = message
await function(self.game)