discordbot.py

import os
from threading import Thread
from discord.ext import commands
from dotenv import load_dotenv
from datamanager import DataManager

load_dotenv()
TOKEN = os.getenv("DISCORD_TOKEN")
GUILD = os.getenv("GUILD_NAME")

class dataQuery(commands.Cog):
  dataManager: DataManager
  def __init__(self, dataManager: DataManager) -> None:
      super().__init__()
      self.dataManager = dataManager

  @commands.command()
  async def antName(self, ctx, name: str):
    print(f"user requested {name} data")
    uid = self.dataManager.getIdFromName(name)
    if uid is not None:
      data = self.dataManager.getDataFromId(uid)
      msg = f"{name} existed but No data found at {uid}"
      if data is not None:
        msg = f'''****Info****
    Name: {name}
    id: {uid}
    alliance: {data[1]}
    power: {data[2]}
    coords: {data[3]}, {data[4]}
    last checked: {data[5]}
        '''
        await ctx.send(msg)
    else:
      await ctx.send(f"unable to find {name}")
    pass
  @commands.command()
  async def antId(self, ctx, id: int):
    results = self.dataManager.getDataFromId(id)
    if results is not None:
      await ctx.send(f"data: {results}")
    else:
      await ctx.send(f"unable to find {id}")
    pass

class DiscordBot(commands.Bot):
  stopped: bool = True
  def __init__(self, command_prefix, dataManager: DataManager):
      super().__init__(command_prefix)
      self.add_cog(dataQuery(dataManager))

  async def on_ready(self):
    print(f"{self.user} has connected")

  def begin(self):
    self.run(TOKEN)

  async def stop(self):
    await self.close()

class DiscordRunner:
  stopped: bool = True
  bot: DiscordBot
  def __init__(self, dataManager: DataManager) -> None:
      self.bot = DiscordBot("!", dataManager)

  def start(self):
    self.stopped = False
    t = Thread(target=self.run)
    t.start()

  def run(self):
    self.bot.begin()
  async def stop(self):
    self.stopped = True
    await self.bot.stop()