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()