jamlib 2.5.0a3.dev0__tar.gz → 2.5.0a3.dev1__tar.gz

{jamlib-2.5.0a3.dev0 → jamlib-2.5.0a3.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jamlib
-Version: 2.5.0a3.dev0
+Version: 2.5.0a3.dev1
 Summary: Simple and universal library for authorization.
 Author-email: Makridenko Adrian <adrianmakridenko@duck.com>, Ksenia Travnikova <kseniatravnikova@duck.com>
 License: MIT License

{jamlib-2.5.0a3.dev0 → jamlib-2.5.0a3.dev1}/jam/ext/flask/extenstions.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-from typing import Any, Optional, Union
+from typing import Any, Optional
 
 from flask import Flask, request
 
@@ -16,18 +16,16 @@ class JamExtension:
 
     def __init__(
         self,
+        jam: Jam,
         app: Optional[Flask] = None,
-        config: Union[str, dict[str, Any]] = "pyproject.toml",
-        pointer: str = "jam",
     ) -> None:
         """Constructor.
 
         Args:
+            jam (jam): Jam instance
             app (Flask | None): Flask app
-            config (str | dict[str, Any]): Jam config
-            pointer (str): Config pointer
         """
-        self._jam = Jam(config, pointer)
+        self._jam = jam
         if app:
             self.init_app(app)
 
@@ -41,22 +39,20 @@ class JWTExtension(JamExtension):
 
     def __init__(
         self,
+        jam: Jam,
         app: Optional[Flask] = None,
-        config: Union[str, dict[str, Any]] = "pyproject.toml",
-        pointer: str = "jam",
         header_name: Optional[str] = "Authorization",
         cookie_name: Optional[str] = None,
     ) -> None:
         """Constructor.
 
         Args:
+            jam (Jam): Jam instance
             app (Flask | None): Flask app
-            config (str | dict[str, Any]): Jam config
-            pointer (str): Config pointer
             header_name (str | None): Header with access token
             cookie_name (str | None): Cookie with access token
         """
-        super().__init__(app, config, pointer)
+        super().__init__(jam, app)
         self.__use_list = getattr(self._jam.module, "list", False)
         self.header = header_name
         self.cookie = cookie_name
@@ -87,3 +83,53 @@ class JWTExtension(JamExtension):
         """Flask app init."""
         app.before_request(self._get_payload)
         app.extensions["jam"] = self._jam
+
+
+class SessionExtension(JamExtension):
+    """Session extension for Jam."""
+
+    def __init__(
+        self,
+        jam: Jam,
+        app: Optional[Flask] = None,
+        header_name: Optional[str] = None,
+        cookie_name: Optional[str] = "sessionId",
+    ) -> None:
+        """Constructor.
+
+        Args:
+            jam (Jam): Jam instance
+            app (Flask | None): Flask app
+            header_name (str | None): Session id header
+            cookie_name (str | None): Session id cookie
+        """
+        super().__init__(jam, app)
+        self.header = header_name
+        self.cookie = cookie_name
+
+    def _get_payload(self) -> Optional[dict[str, Any]]:
+        session_id = None
+        if self.cookie:
+            session_id = request.cookies.get(self.cookie)
+
+        if not session_id and self.header:
+            header = request.headers.get(self.header)
+            if header and header.startswith("Bearer "):
+                session_id = header.split("Bearer ")[1]
+
+        if not session_id:
+            return None
+        try:
+            payload: Optional[dict[str, Any]] = self._jam.get_session(
+                session_id
+            )
+        except Exception as e:
+            logger.warning(str(e))
+            return None
+
+        return payload
+
+    def init_app(self, app: Flask) -> None:
+        """Flask app init."""
+        app.before_request(self._get_payload)
+        app.extensions["jam"] = self._jam

{jamlib-2.5.0a3.dev0 → jamlib-2.5.0a3.dev1}/jamlib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jamlib
-Version: 2.5.0a3.dev0
+Version: 2.5.0a3.dev1
 Summary: Simple and universal library for authorization.
 Author-email: Makridenko Adrian <adrianmakridenko@duck.com>, Ksenia Travnikova <kseniatravnikova@duck.com>
 License: MIT License

{jamlib-2.5.0a3.dev0 → jamlib-2.5.0a3.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "jamlib"
-version = "2.5.0a3.dev0"
+version = "2.5.0a3.dev1"
 description = "Simple and universal library for authorization."
 authors = [
     {name = "Makridenko Adrian",email = "adrianmakridenko@duck.com"},