[BugFix] - Replace python-jose by PyJWT

openbb_platform/core/openbb_core/app/service/hub_service.py

@@ -4,9 +4,7 @@
 from warnings import warn
 
 from fastapi import HTTPException
-from jose import JWTError
-from jose.exceptions import ExpiredSignatureError
-from jose.jwt import decode, get_unverified_header
+from jwt import ExpiredSignatureError, PyJWTError, decode, get_unverified_header
 from openbb_core.app.model.abstract.error import OpenBBError
 from openbb_core.app.model.credentials import Credentials
 from openbb_core.app.model.hub.hub_session import HubSession
@@ -139,7 +137,7 @@ def _get_session_from_platform_token(self, token: str) -> HubSession:
         if not token:
             raise OpenBBError("Platform personal access token not found.")
 
-        self.check_token_expiration(token)
+        self._check_token_expiration(token)
 
         response = post(
             url=self._base_url + "/sdk/login",
@@ -259,7 +257,7 @@ def platform2hub(self, credentials: Credentials) -> HubUserSettings:
         return settings
 
     @staticmethod
-    def check_token_expiration(token: str) -> None:
+    def _check_token_expiration(token: str) -> None:
         """Check token expiration, raises exception if expired."""
         try:
             header_data = get_unverified_header(token)
@@ -271,5 +269,5 @@ def check_token_expiration(token: str) -> None:
             )
         except ExpiredSignatureError as e:
             raise OpenBBError("Platform personal access token expired.") from e
-        except JWTError as e:
+        except PyJWTError as e:
             raise OpenBBError("Failed to decode Platform token.") from e

openbb_platform/core/pyproject.toml

@@ -13,7 +13,6 @@ websockets = "^12.0"
 pandas = ">=1.5.3"
 html5lib = "^1.1"
 fastapi = "^0.104.1"
-python-jose = "^3.3.0"
 uuid7 = "^0.1.0"
 posthog = "^3.3.1"
 python-multipart = "^0.0.7"
@@ -23,6 +22,7 @@ importlib-metadata = "^6.8.0"
 python-dotenv = "^1.0.0"
 aiohttp = "^3.9.5"
 ruff = "^0.1.6"
+pyjwt = "^2.8.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.0.0"

openbb_platform/core/tests/app/service/test_hub_service.py

@@ -5,9 +5,11 @@
 
 
 from pathlib import Path
+from time import time
 from unittest.mock import MagicMock, patch
 
 import pytest
+from jwt import encode
 from openbb_core.app.service.hub_service import (
     Credentials,
     HubService,
@@ -308,3 +310,35 @@ def test_platform2hub():
     assert user_settings.features_keys["API_FRED_KEY"] == "fred"
     assert user_settings.features_keys["benzinga_api_key"] == "benzinga"
     assert "some_api_key" not in user_settings.features_keys
+
+
+@pytest.mark.parametrize(
+    "token, message",
+    [
+        # valid
+        (
+            encode(
+                {"some": "payload", "exp": int(time()) + 100},
+                "secret",
+                algorithm="HS256",
+            ),
+            None,
+        ),
+        # expired
+        (
+            encode(
+                {"some": "payload", "exp": int(time())}, "secret", algorithm="HS256"
+            ),
+            "Platform personal access token expired.",
+        ),
+        # invalid
+        ("invalid_token", "Failed to decode Platform token."),
+    ],
+)
+def test__check_token_expiration(token, message):
+    """Test check token expiration function."""
+    if message:
+        with pytest.raises(OpenBBError, match=message):
+            HubService._check_token_expiration(token)
+    else:
+        HubService._check_token_expiration(token)