sso-nebus 0.1.2__tar.gz → 0.1.4__tar.gz

{sso_nebus-0.1.2 → sso_nebus-0.1.4}/PKG-INFO

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

{sso_nebus-0.1.2 → sso_nebus-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sso-nebus"
-version = "0.1.2"
+version = "0.1.4"
 description = "Python клиент для взаимодействия с MS Auth Service API"
 authors = [
     {name = "Артем Костюченко",email = "kostyuchenko.work@gmail.com"}

{sso_nebus-0.1.2 → sso_nebus-0.1.4}/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"""
 
@@ -59,7 +80,7 @@ class ServiceClient(BaseClient):
         if scope:
             form_data["scope"] = scope
 
-        data = await self.post("auth/token", form_data=form_data)
+        data = await self.post("token", form_data=form_data)
         token_response = TokenResponse(**data)
 
         # Сохраняем токен
@@ -83,7 +104,7 @@ class ServiceClient(BaseClient):
             raise TokenError(
                 "Access token не найден. Вызовите get_access_token() сначала.")
 
-        data = await self.get("auth/me", access_token=access_token)
+        data = await self.get("me", access_token=access_token)
         return UserInfo(**data)
 
     def set_access_token(self, access_token: str):
@@ -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,