sso-nebus 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

sso_nebus/service_client.py

@@ -1,12 +1,33 @@
 """Клиент для микросервисного взаимодействия (Client Credentials)"""
 
-from typing import Optional
+import base64
+import json
+from typing import Any, Dict, Optional
 
 from .base import BaseClient
 from .models import TokenResponse, UserInfo
 from .exceptions import TokenError
 
 
+def _decode_jwt_payload(token: str) -> Dict[str, Any]:
+    """
+    Декодирует payload JWT без проверки подписи (только чтение полей).
+    Токен получен от SSO, используется для извлечения claim service_name.
+    """
+    try:
+        parts = token.split(".")
+        if len(parts) != 3:
+            return {}
+        payload_b64 = parts[1]
+        padding = 4 - len(payload_b64) % 4
+        if padding != 4:
+            payload_b64 += "=" * padding
+        payload_bytes = base64.urlsafe_b64decode(payload_b64)
+        return json.loads(payload_bytes.decode("utf-8"))
+    except Exception:
+        return {}
+
+
 class ServiceClient(BaseClient):
     """Клиент для микросервисного взаимодействия с Client Credentials Grant"""
 
@@ -107,6 +128,35 @@ class ServiceClient(BaseClient):
         """Получить текущий access token"""
         return self._access_token
 
+    def get_token_payload(self, access_token: Optional[str] = None) -> Dict[str, Any]:
+        """
+        Получить payload текущего JWT токена (без проверки подписи).
+        Удобно для чтения полей, вшитых SSO (например service_name).
+
+        Args:
+            access_token: Токен (если не указан, используется сохранённый)
+
+        Returns:
+            Словарь с полями payload или пустой словарь при ошибке
+        """
+        token = access_token or self._access_token
+        if not token:
+            return {}
+        return _decode_jwt_payload(token)
+
+    def get_service_name(self, access_token: Optional[str] = None) -> Optional[str]:
+        """
+        Получить название сервиса из JWT токена (поле service_name, вшитое SSO).
+
+        Args:
+            access_token: Токен (если не указан, используется сохранённый)
+
+        Returns:
+            Название сервиса или None, если токена нет или поле отсутствует
+        """
+        payload = self.get_token_payload(access_token)
+        return payload.get("service_name")
+
     async def request_with_auth(
         self,
         method: str,

{sso_nebus-0.1.3.dist-info → sso_nebus-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sso-nebus
-Version: 0.1.3
+Version: 0.1.4
 Summary: Python клиент для взаимодействия с MS Auth Service API
 License: LICENSE
 License-File: LICENSE

{sso_nebus-0.1.3.dist-info → sso_nebus-0.1.4.dist-info}/RECORD

@@ -5,9 +5,9 @@ sso_nebus/exmples/example_admin.py,sha256=eAqkS4J5YyJneP4EPUrBlvNbGlK7rHEO39kCkl
 sso_nebus/exmples/example_service.py,sha256=19NNTmbaRSaWLpDQffGWNDmXkDJXdLOgD2WkV32xUu8,1778
 sso_nebus/exmples/example_user.py,sha256=c_Uf36FDHIXvv_vdqsO5XvXkvu1YOUsqaTO44_8DqVM,1878
 sso_nebus/models.py,sha256=0xIE5RvnxV2sAxO7_GTCes6HlHbdzGwHIjWVe1X7-ks,3024
-sso_nebus/service_client.py,sha256=7A1QSbANebXHpy5k7eJj3tXuwl70TAs61SAv0UEnuwk,5460
+sso_nebus/service_client.py,sha256=ptXKHyoEYUpaU-yHrDG2IGEMp-tKSpLJzvK8syqEG2k,7558
 sso_nebus/user_client.py,sha256=-3pJmvZxJgyEo3moGQvZVAOhJMOIC6NuE8NgJ33Fyfs,12976
-sso_nebus-0.1.3.dist-info/licenses/LICENSE,sha256=dCbOm3zpH8T7vLDC2K7QJLu-LEl2zqaSuyARbqfGsEY,1863
-sso_nebus-0.1.3.dist-info/METADATA,sha256=RcoakbPB37v7orhugBtjyGnDoYRh2PJGTMdn7dLFAwg,8406
-sso_nebus-0.1.3.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-sso_nebus-0.1.3.dist-info/RECORD,,
+sso_nebus-0.1.4.dist-info/licenses/LICENSE,sha256=dCbOm3zpH8T7vLDC2K7QJLu-LEl2zqaSuyARbqfGsEY,1863
+sso_nebus-0.1.4.dist-info/METADATA,sha256=sZwTP8G8neMmje_rHGW2x3-lNCri4cSVtGtuYwhPNNw,8406
+sso_nebus-0.1.4.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+sso_nebus-0.1.4.dist-info/RECORD,,

{sso_nebus-0.1.3.dist-info → sso_nebus-0.1.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.1
+Generator: poetry-core 2.3.0
 Root-Is-Purelib: true
 Tag: py3-none-any