small fixes

src/commands/arma_get_db.py

@@ -5,6 +5,7 @@
 import os
 import pandas as pd
 
+
 async def get_players():
  """
  Gets the list of players that have played on the Arma server.
@@ -18,9 +19,9 @@ async def get_players():
  url = f"https://{host}/attendance"
 
  headers = {
- 'content-type': "application/json",
- 'x-apikey': api_key,
- 'cache-control': "no-cache"
+ "content-type": "application/json",
+ "x-apikey": api_key,
+ "cache-control": "no-cache",
  }
 
  # Open asynchronous HTTP session and get the collection of players.
@@ -37,11 +38,20 @@ async def get_players():
 
  # Prepare the data for a pandas DataFrame.
  for x, y in enumerate(records):
- new_records.append((y["Profile_Name"], y["Missions_Attended"],
- y["Date_Of_Last_Mission"].split("T")[0], y["Steam_UID"]))
-
- df = pd.DataFrame(new_records, index=None,
- columns=("Name", "Missions Attended", "Last Seen", "Steam UID"))
+ new_records.append(
+ (
+ y["Profile_Name"],
+ y["Missions_Attended"],
+ y["Date_Of_Last_Mission"].split("T")[0],
+ y["Steam_UID"],
+ )
+ )
+
+ df = pd.DataFrame(
+ new_records,
+ index=None,
+ columns=("Name", "Missions Attended", "Last Seen", "Steam UID"),
+ )
 
  return f"```{df.to_string(index=False)}\nTotal Players Tracked: {len(df)}```"
 
@@ -56,14 +66,7 @@ class ArmAGetDB(commands.Cog):
  """
 
  @commands.slash_command(dm_permission=True)
- async def arma(self, inter):
- pass
-
- @arma.sub_command()
- async def get_players(
- self,
- inter: disnake.ApplicationCommandInteraction
- ):
+ async def arma_get_players(self, inter: disnake.ApplicationCommandInteraction):
  """
  Display a list of players that have played on the Arma server.
  """

src/main.py

@@ -49,7 +49,6 @@ async def on_connect():
  # everything below here will only run on the first connect
 
  try:
-
  logging.info("Connected to Discord. Initializing Database.")
  await init_database(get_mongo_uri(), "FUBot")
  except Exception as e:
@@ -67,6 +66,7 @@ async def on_connect():
  bot.load_extension("commands.squad_markup")
  bot.load_extension("commands.ps2_lookup")
  bot.load_extension("commands.arma_upload_map")
+ bot.load_extension("commands.arma_get_db")
  # bot.load_extension("loggers.discord_logger")
  # bot.load_extension("loggers.ps2_outfit_members")
  # bot.load_extension("loggers.ps2_outfit_online_logger")