From 2bf8629e4cf2395654834e91f80ba1b1684d25d9 Mon Sep 17 00:00:00 2001
From: SeaswimmerTheFsh <seaswimmerthefsh@gmail.com>
Date: Thu, 29 Feb 2024 23:31:30 -0500
Subject: [PATCH] fix(pterodactyl): updated typehints

---
 pterodactyl/pterodactyl.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/pterodactyl/pterodactyl.py b/pterodactyl/pterodactyl.py
index 5895563..87ada6b 100644
--- a/pterodactyl/pterodactyl.py
+++ b/pterodactyl/pterodactyl.py
@@ -1,8 +1,11 @@
 import asyncio
 import json
+from typing import Optional
+
 
 import discord
 import websockets
+from pydactyl import PterodactylClient
 from redbot.core import commands
 from redbot.core.bot import Red
 from redbot.core.utils.chat_formatting import box
@@ -17,9 +20,9 @@ class Pterodactyl(commands.Cog):
 
     def __init__(self, bot: Red):
         self.bot = bot
-        self.client = None
-        self.task = None
-        self.websocket = None
+        self.client: Optional[PterodactylClient] = None
+        self.task: Optional[asyncio.Task] = None
+        self.websocket: Optional[websockets.WebSocketClientProtocol] = None
 
     async def cog_load(self) -> None:
         self.task = self.get_task()