Tidied up unused variables

NexInfinite · Aug 4, 2020 · 60c33d6 · 60c33d6
1 parent 48231a4
commit 60c33d6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/Hiven/Websocket/websocket_handler.py b/Hiven/Websocket/websocket_handler.py
@@ -14,11 +14,10 @@
 
 class WebSocket:
     def __init__(self, token, debug=False, output=False, outer=None):
-        self.TOKEN = token
-        self.user = None
         self.restURL = "https://api.hiven.io/v1"
         self.WEBSOCKET = "wss://swarm-dev.hiven.io/socket?encoding=json&compression=text_json"
         self.HEARTBEAT = 0
+        self.TOKEN = token
         self.OUTPUT = output
         self.outer = outer
         if debug:
@@ -33,7 +32,7 @@ def start(self):
     def on_open(self, ws):
         ws.send('{"op": 2, "d":{"token": "' + self.TOKEN + '"}}')
 
-        def run(*args):
+        def run():
             while True:
                 time.sleep(self.HEARTBEAT / 1000)
                 ws.send('{"op": 3}')

diff --git a/Hiven/client.py b/Hiven/client.py
@@ -5,14 +5,10 @@
 
 class Bot:
     def __init__(self, token, debug=False, output=False):
-        self.TOKEN = token
         self.restURL = "https://api.hiven.io/v1"
-        self.WEBSOCKET = "wss://swarm-dev.hiven.io/socket?encoding=json&compression=text_json"
-        self.HEARTBEAT = 0
-        self.OUTPUT = output
-        self.DEBUG = debug
-        self.ws = WebSocket(self.TOKEN, self.DEBUG, self.OUTPUT, self)
+        self.TOKEN = token
         self.user = None
+        self.ws = WebSocket(self.TOKEN, debug, output, self)
 
     def login(self):
         self.ws.start()