slots fix

Author: GnarLito

tryhackme/badge.py

@@ -1,7 +1,7 @@
 
 
 class Badge:
-    __slots__ = ("title", "name", "description", "earned", )
+    __slots__ = ("_state", "_to_earn", "title", "name", "description", "earned", )
 
     def __init__(self, state, data):
         self._state = state

tryhackme/game.py

@@ -1,7 +1,7 @@
 from .user import KOTHUser
 
 class KingoftheHill:
-    __slot__ = ("id", "start_time", "finnish_time", "status", "flag_count", "resets", "game_type", "ceator", "users", "King", )
+    __slot__ = ("_state", "id", "start_time", "finnish_time", "status", "flag_count", "resets", "game_type", "ceator", "users", "King", )
 
     def __init__(self, state, data):
         self._state = state

tryhackme/message.py

@@ -1,7 +1,7 @@
 from . import utils
 
 class Message:
-    __slots__ = ("group", "inserted", "user", )
+    __slots__ = ("_state", "group", "inserted", "user", )
 
     def __init__(self, state, group, data):
         self._state = state
@@ -16,7 +16,7 @@ def _from_data(self, data):
 
 # * can only be used on `get_all_message_groups` api call
 class MessageGroup:
-    __slots__ = ("messages", "id", "title", )
+    __slots__ = ("_state", "_users", "messages", "id", "title", )
     def __init__(self, state, data):
         self._state = state
         self.messages = []

tryhackme/module.py

@@ -1,7 +1,7 @@
 
 
 class Module:
-    __slots__ = ("name", "code", "id", "description", "summary", )
+    __slots__ = ("_state", "_rooms", "_prerequisites", "_next_steps", "name", "code", "id", "description", "summary", )
 
     def __init__(self, state, data):
         self._state = state
@@ -16,7 +16,7 @@ def _from_data(self, data):
         self.summary = data.get('summary')
         self._rooms = data.get('rooms')
         self._prerequisites = data.get('prerequisites')
-        self._nextSteps = data.get('nextSteps')
+        self._next_steps = data.get('nextSteps')
 
     @property
     def rooms(self):
@@ -25,6 +25,6 @@ def rooms(self):
     def prerequisites(self):
         return [self._state.store_module(module) for module in self._prerequisites]
     @property
-    def nextSteps(self):
-        return [self._state.store_module(module) for module in self._nextSteps]
+    def next_steps(self):
+        return [self._state.store_module(module) for module in self._next_steps]
 

tryhackme/path.py

@@ -2,7 +2,7 @@
 from . import utils
 
 class Path:
-    __slots__ = ("code", "raw_description", "color", "raw_intro", "type", "public", "room_count", "summary", "difficult", "time_to_complete", )
+    __slots__ = ("_state", "_badges", "_careers", "_modules", "_tasks", "code", "raw_description", "color", "raw_intro", "type", "public", "room_count", "summary", "difficult", "time_to_complete", )
 
     def __init__(self, state, data):
         self._state = state
@@ -47,11 +47,9 @@ def tasks(self):
     @property
     def modules(self):
         return [self._state.store_module(module) for module in self._module]
-
     @property
     def badges(self):
         return [self._state.store_badge(badge) for badge in self._badges]
-    # ? What is this
     @property
-    def careers(self):
+    def careers(self): # ? What is this
         return [career for career in self._careers]

tryhackme/question.py

@@ -1,7 +1,7 @@
 from . import utils
 
 class Question:
-    __slots__ = ("raw_question", "question", "raw_hint", "hint", "number", "raw_description", "description", "extra_points", "correct", "attempts", "submission", "has_answer", )
+    __slots__ = ("_state", "raw_question", "question", "raw_hint", "hint", "number", "raw_description", "description", "extra_points", "correct", "attempts", "submission", "has_answer", )
 
     def __init__(self, state, data):
         self._state = state
@@ -14,7 +14,7 @@ def _from_data(self, data):
         self.hint = utils.HTML_parse(self.raw_hint)
         self.number = data.get("questionNo")
 
-        # * only when valid session is used
+        # * only when authenticated
         self.raw_description = data.get("answerDesc", "")
         self.description = utils.HTML_parse(self.raw_description)
         self.extra_points = data.get("extraPoints", None)

tryhackme/room.py

@@ -4,7 +4,7 @@
 # ? writeups class
 
 class Room:
-    __slots__ = ("name", "id", "title", "description", "created", "published", "users", "type", "public", "difficulty", "freeToUse", "ctf", "tags", "ipType", "simpleRoom", "writeups", "locked", "comingSoon", "views", "certificate", "timeToComplete", "userCompleted", )
+    __slots__ = ("_state", "_creators", "name", "id", "title", "description", "created", "published", "users", "type", "public", "difficulty", "freeToUse", "ctf", "tags", "ipType", "simpleRoom", "writeups", "locked", "comingSoon", "views", "certificate", "timeToComplete", "userCompleted", )
 
     def __init__(self, state, data):
         self._state = state
@@ -65,7 +65,7 @@ def tasks(self):
         if self.freeToUse or self._state.subscribed:
             return [RoomTask(state=self._state, data=task) for task in self._state.http.get_room_tasks(room_code=self.name).get('data')]
         else:
-            return [RoomTask(state=self._state, data=task) for task in self._state.http.get_room_tasks(room_code=self.name, settings={"static": True})]
+            return [RoomTask(state=self._state, data=task) for task in self._state.http.get_room_tasks(room_code=self.name, settings=["static"]).get('data')]
     @property
     def creators(self):
         return [self._state.store_user(username=user.get('username')) for user in self._creators]

tryhackme/serie.py

@@ -1,7 +1,7 @@
 
 
 class Serie:
-    __slots__ = ("id", "code", "name", "description", "difficulty", )
+    __slots__ = ("_state", "_rooms", "id", "code", "name", "description", "difficulty", )
     def __init__(self, state, data):
         self._state = state
 

tryhackme/task.py

@@ -3,7 +3,7 @@
 
 
 class RoomTask:
-    __slots__ = ("raw_title", "raw_description", "description", "type", "number", "created", "deadline", "uploadId", )
+    __slots__ = ("_state", "_questions", "raw_title", "raw_description", "description", "type", "number", "created", "deadline", "uploadId", )
 
     def __init__(self, state, data):
         self._state = state
@@ -32,7 +32,7 @@ def questions(self):
 
 
 class PathTask:
-    __slots__ = ("id", "title", "time", "overview", "outcome", "number", )
+    __slots__ = ("_state", "_module_url", "_rooms", "id", "title", "time", "overview", "outcome", "number", )
 
     def __init__(self, state, data):
         self.state = state
@@ -42,7 +42,7 @@ def __init__(self, state, data):
     def _from_data(self, data):
         self.id = data.get("_id")
         self.title = data.get("title")
-        self._moduleURL = data.get("moduleURL")
+        self._module_url = data.get("moduleURL")
         self.time = int(data.get("time"))
         self.overview = data.get("overview")
         self.outcomes = data.get("outcome")
@@ -54,4 +54,4 @@ def rooms(self):
         return [self._state.get_room(room_code=room.get("code")) for room in self._rooms]
     @property
     def module(self):
-        return self._state.get_module(moduleURL=self._moduleURL)
+        return self._state.get_module(moduleURL=self._module_url)

tryhackme/team.py

@@ -1,7 +1,7 @@
 
 
 class Team:
-    __slots__ = ("name", "captain", "password", "university", )
+    __slots__ = ("_state", "_members", "name", "captain", "password", "university", )
 
     def __init__(self, state, data):
         self._state = state