uiprotect 7.14.2__tar.gz → 7.15.1__tar.gz

{uiprotect-7.14.2 → uiprotect-7.15.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uiprotect
-Version: 7.14.2
+Version: 7.15.1
 Summary: Python API for Unifi Protect (Unofficial)
 License: MIT
 Author: UI Protect Maintainers

{uiprotect-7.14.2 → uiprotect-7.15.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uiprotect"
-version = "7.14.2"
+version = "7.15.1"
 license = "MIT"
 description = "Python API for Unifi Protect (Unofficial)"
 authors = [{ name = "UI Protect Maintainers", email = "ui@koston.org" }]

{uiprotect-7.14.2 → uiprotect-7.15.1}/src/uiprotect/api.py

@@ -169,6 +169,7 @@ class BaseApiClient:
     _last_token_cookie: Morsel[str] | None = None
     _last_token_cookie_decode: dict[str, Any] | None = None
     _session: aiohttp.ClientSession | None = None
+    _public_api_session: aiohttp.ClientSession | None = None
     _loaded_session: bool = False
     _cookiename = "TOKEN"
 
@@ -192,6 +193,7 @@ class BaseApiClient:
         api_key: str | None = None,
         verify_ssl: bool = True,
         session: aiohttp.ClientSession | None = None,
+        public_api_session: aiohttp.ClientSession | None = None,
         ws_timeout: int = 30,
         cache_dir: Path | None = None,
         config_dir: Path | None = None,
@@ -218,6 +220,9 @@ class BaseApiClient:
         if session is not None:
             self._session = session
 
+        if public_api_session is not None:
+            self._public_api_session = public_api_session
+
         self._update_url()
 
     def _update_cookiename(self, cookie: SimpleCookie) -> None:
@@ -261,6 +266,15 @@ class BaseApiClient:
 
         return self._session
 
+    async def get_public_api_session(self) -> aiohttp.ClientSession:
+        """Gets or creates current public API client session"""
+        if self._public_api_session is None or self._public_api_session.closed:
+            if self._public_api_session is not None and self._public_api_session.closed:
+                _LOGGER.debug("Public API session was closed, creating a new one")
+            self._public_api_session = aiohttp.ClientSession()
+
+        return self._public_api_session
+
     async def _auth_websocket(self, force: bool) -> dict[str, str] | None:
         """Authenticate for Websocket."""
         if force:
@@ -306,6 +320,12 @@ class BaseApiClient:
             self._session = None
             self._loaded_session = False
 
+    async def close_public_api_session(self) -> None:
+        """Closing and deletes public API client session"""
+        if self._public_api_session is not None:
+            await self._public_api_session.close()
+            self._public_api_session = None
+
     async def _cancel_update_task(self) -> None:
         if self._update_task:
             self._update_task.cancel()
@@ -345,7 +365,11 @@ class BaseApiClient:
         _LOGGER.debug("Request url: %s", request_url)
         if not self._verify_ssl:
             kwargs["ssl"] = False
-        session = await self.get_session()
+
+        if public_api:
+            session = await self.get_public_api_session()
+        else:
+            session = await self.get_session()
 
         for attempt in range(2):
             try:
@@ -777,6 +801,7 @@ class ProtectApiClient(BaseApiClient):
         api_key: str | None = None,
         verify_ssl: bool = True,
         session: aiohttp.ClientSession | None = None,
+        public_api_session: aiohttp.ClientSession | None = None,
         ws_timeout: int = 30,
         cache_dir: Path | None = None,
         config_dir: Path | None = None,
@@ -797,6 +822,7 @@ class ProtectApiClient(BaseApiClient):
             api_key=api_key,
             verify_ssl=verify_ssl,
             session=session,
+            public_api_session=public_api_session,
             ws_timeout=ws_timeout,
             ws_receive_timeout=ws_receive_timeout,
             cache_dir=cache_dir,
@@ -2063,15 +2089,9 @@ class ProtectApiClient(BaseApiClient):
         if not name:
             raise BadRequest("API key name cannot be empty")
 
-        user_id = None
-        if self._last_token_cookie_decode is not None:
-            user_id = self._last_token_cookie_decode.get("userId")
-        if not user_id:
-            raise BadRequest("User ID not available for API key creation")
-
         response = await self.api_request(
             api_path="/proxy/users/api/v2",
-            url=f"/user/{user_id}/keys",
+            url="/user/self/keys",
             method="post",
             json={"name": name},
         )

{uiprotect-7.14.2 → uiprotect-7.15.1}/src/uiprotect/cli/__init__.py

@@ -173,6 +173,7 @@ def main(
     async def update() -> None:
         protect._bootstrap = await protect.get_bootstrap()
         await protect.close_session()
+        await protect.close_public_api_session()
 
     run_async(update())
     ctx.obj = CliContext(protect=protect, output_format=output_format)
@@ -286,6 +287,7 @@ def profile_ws(
         unsub()
         await protect.async_disconnect_ws()
         await protect.close_session()
+        await protect.close_public_api_session()
 
     _setup_logger()
 
@@ -325,6 +327,7 @@ def create_api_key(
     async def callback() -> str:
         api_key = await protect.create_api_key(name)
         await protect.close_session()
+        await protect.close_public_api_session()
         return api_key
 
     _setup_logger()
@@ -340,6 +343,7 @@ def get_meta_info(ctx: typer.Context) -> None:
     async def callback() -> MetaInfo:
         meta = await protect.get_meta_info()
         await protect.close_session()
+        await protect.close_public_api_session()
         return meta
 
     _setup_logger()

{uiprotect-7.14.2 → uiprotect-7.15.1}/src/uiprotect/cli/backup.py

@@ -1064,6 +1064,7 @@ async def _events(
     finally:
         _LOGGER.debug("Cleaning up Protect connection/database...")
         await ctx.protect.close_session()
+        await ctx.protect.close_public_api_session()
         await ctx.db_engine.dispose()
 
 

{uiprotect-7.14.2 → uiprotect-7.15.1}/src/uiprotect/cli/base.py

@@ -36,6 +36,7 @@ def run(ctx: typer.Context, func: Coroutine[Any, Any, T]) -> T:
     async def callback() -> T:
         return_value = await func
         await ctx.obj.protect.close_session()
+        await ctx.obj.protect.close_public_api_session()
         return return_value
 
     try:

{uiprotect-7.14.2 → uiprotect-7.15.1}/src/uiprotect/test_util/__init__.py

@@ -142,6 +142,7 @@ class SampleDataGenerator:
 
         if close_session:
             await self.client.close_session()
+            await self.client.close_public_api_session()
 
         await self.write_json_file("sample_constants", self.constants, anonymize=False)