Merge pull request #13 from offish/v2.0.7

fix pricestf socket crashing

src/tf2_utils/__init__.py

@@ -1,6 +1,6 @@
 __title__ = "tf2-utils"
 __author__ = "offish"
-__version__ = "2.0.6"
+__version__ = "2.0.7"
 __license__ = "MIT"
 
 from .sku import get_sku, get_sku_properties, sku_to_defindex

src/tf2_utils/sockets.py

@@ -2,7 +2,7 @@
 
 import json
 
-from websockets.sync.client import connect
+from websockets.sync.client import ClientConnection, connect
 
 
 class BackpackTFSocket:
@@ -70,8 +70,23 @@ def __init__(
         self.prices_tf = PricesTF()
         self.settings = settings
 
-    def process_message(self, message: str) -> None:
+    def process_message(self, ws: ClientConnection, message: str) -> None:
         data = json.loads(message)
+
+        # our auths are only valid for 10 minutes at a time
+        # pricestf requests us to authenticate again
+        if data.get("type") == "AUTH_REQUIRED":
+            self.prices_tf.request_access_token()
+            ws.send(
+                json.dumps(
+                    {
+                        "type": "AUTH",
+                        "data": {"accessToken": self.prices_tf.access_token},
+                    }
+                )
+            )
+            return
+
         self.callback(data)
 
     def listen(self) -> None:
@@ -86,4 +101,4 @@ def listen(self) -> None:
         ) as websocket:
             while True:
                 message = websocket.recv()
-                self.process_message(message)
+                self.process_message(websocket, message)