flespi-sdk 0.1.1__tar.gz → 0.1.3__tar.gz

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flespi-sdk
-Version: 0.1.1
+Version: 0.1.3
 Summary: Add your description here
 Author-email: Sergey Shevchik <sergey.shevchik@gmail.com>
 Requires-Python: >=3.11
@@ -84,4 +84,5 @@ await user.metadata.set_value("test-metadata-key", "123")
 print(await user.metadata.get())
 await user.metadata.delete_value("test-metadata-key")
 print(await user.metadata.get())
+await user.update_password("newpassword123456789")
 ```

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/README.md

@@ -75,4 +75,5 @@ await user.metadata.set_value("test-metadata-key", "123")
 print(await user.metadata.get())
 await user.metadata.delete_value("test-metadata-key")
 print(await user.metadata.get())
+await user.update_password("newpassword123456789")
 ```

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/pyproject.toml

@@ -1,22 +1,15 @@
 [project]
 name = "flespi-sdk"
-version = "0.1.1"
+version = "0.1.3"
 description = "Add your description here"
 readme = "README.md"
-authors = [
-    { name = "Sergey Shevchik", email = "sergey.shevchik@gmail.com" }
-]
+authors = [{ name = "Sergey Shevchik", email = "sergey.shevchik@gmail.com" }]
 requires-python = ">=3.11"
-dependencies = [
-    "aiohttp>=3.11.15",
-]
+dependencies = ["aiohttp>=3.11.15"]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [dependency-groups]
-dev = [
-    "mypy>=1.15.0",
-    "ruff>=0.11.2",
-]
+dev = ["mypy>=1.15.0", "ruff>=0.11.2"]

flespi_sdk-0.1.3/src/flespi_sdk/exception.py

@@ -0,0 +1,15 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class FlespiError:
+    code: int
+    reason: str
+    id: int | None = None
+    selector: str | None = None
+
+
+class FlespiException(Exception):
+    def __init__(self, status_code: int, errors: list[dict]):
+        self.status_code = status_code
+        self.errors = [FlespiError(**error) for error in errors]

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/src/flespi_sdk/modules/flespi_session.py

@@ -1,5 +1,7 @@
 from aiohttp import ClientSession
 
+from flespi_sdk.exception import FlespiException
+
 
 class FlespiSession:
     def __init__(self, session: ClientSession, cid: int | None = None):
@@ -16,7 +18,9 @@ class FlespiSession:
 
     async def get_result(self, response):
         if response.status != 200:
-            raise Exception("Status code: " + str(response.status))
+            raise FlespiException(
+                status_code=response.status, errors=(await response.json())["errors"]
+            )
         response_json = await response.json()
         result = response_json["result"]
         return result

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/src/flespi_sdk/modules/realms/realm.py

@@ -1,5 +1,7 @@
 from aiohttp import ClientSession
 
+from flespi_sdk.modules.realms.realm_home import RealmHome
+from flespi_sdk.modules.realms.realm_token_params import RealmTokenParams
 from flespi_sdk.modules.realms.users import Users
 from flespi_sdk.modules.item_with_metadata import ItemWithMetadata
 
@@ -19,3 +21,8 @@ class Realm(ItemWithMetadata):
             client_session=session,
             cid=cid,
         )
+
+        self.home = RealmHome(realm_id=realm_id, session=session, cid=cid)
+        self.token_params = RealmTokenParams(
+            realm_id=realm_id, session=session, cid=cid
+        )

flespi_sdk-0.1.3/src/flespi_sdk/modules/realms/realm_home.py

@@ -0,0 +1,50 @@
+from aiohttp import ClientSession
+from flespi_sdk.modules.flespi_session import FlespiSession
+
+
+class RealmHome(FlespiSession):
+    def __init__(
+        self,
+        realm_id: int,
+        session: ClientSession,
+        cid: int | None = None,
+    ):
+        super().__init__(session, cid)
+        self.realm_id = realm_id
+
+    async def get(self):
+        params = {"fields": "home"}
+        async with self.session.get(
+            f"/platform/realms/{self.realm_id}",
+            params=params,
+            headers=self.get_headers(),
+        ) as response:
+            result = await self.get_result(response)
+            return result[0]["home"]
+
+    async def set(self, data: dict):
+        response = await self.session.put(
+            f"/platform/realms/{self.realm_id}",
+            json={"home": data},
+            headers=self.get_headers(),
+        )
+        return await self.get_result(response)
+
+    async def set_current_subaccount(self):
+        data = {"type": 0}
+        await self.set(data)
+
+    async def set_existing_subaccount(self, subaccount_id: int):
+        data = {"type": 1, "subaccount_id": subaccount_id}
+        await self.set(data)
+
+    async def set_create_new_subaccount(
+        self, subaccount_id: int | None = None, limit_id: int | None = None
+    ):
+        data = {"type": 2}
+        if subaccount_id:
+            data["subaccount_id"] = subaccount_id
+        if limit_id:
+            data["limit_id"] = limit_id
+
+        await self.set(data)

flespi_sdk-0.1.3/src/flespi_sdk/modules/realms/realm_token_params.py

@@ -0,0 +1,82 @@
+from aiohttp import ClientSession
+
+from flespi_sdk.modules.flespi_session import FlespiSession
+
+
+class RealmTokenParams(FlespiSession):
+    def __init__(self, realm_id: int, session: ClientSession, cid: int | None = None):
+        super().__init__(session, cid)
+        self.realm_id = realm_id
+
+    async def get(self):
+        params = {"fields": "token_params"}
+        async with self.session.get(
+            f"platform/realms/{self.realm_id}",
+            params=params,
+            headers=self.get_headers(),
+        ) as response:
+            return (await self.get_result(response))[0]["token_params"]
+
+    async def set(self, data: dict):
+        response = await self.session.put(
+            f"platform/realms/{self.realm_id}",
+            json={"token_params": data},
+            headers=self.get_headers(),
+        )
+        return await self.get_result(response)
+
+    async def set_ttl(self, seconds: int):
+        token_params = await self.get()
+        if token_params["ttl"] == seconds:
+            return
+        token_params["ttl"] = seconds
+        await self.set(token_params)
+
+    async def set_ip_addresses(self, ip_addresses: list[str]):
+        token_params = await self.get()
+        ip_addresses_csv = ",".join(ip_addresses)
+        if "ips" in token_params and token_params["ips"] == ip_addresses_csv:
+            return
+        token_params["ips"] = ip_addresses_csv
+        await self.set(token_params)
+
+    async def allow_all_origins(self):
+        token_params = await self.get()
+        if "origins" in token_params and token_params["origins"] == "*":
+            return
+        token_params["origins"] = "*"
+        await self.set(token_params)
+
+    async def set_origins(self, origins: list[str], allow_flespi_origins: bool = False):
+        token_params = await self.get()
+        token_params["origins"] = origins.copy()
+        if allow_flespi_origins:
+            token_params["origins"].append({"preset": "flespi"})
+        await self.set(token_params)
+
+    async def set_standard_access(self):
+        token_params = await self.get()
+        if token_params["access"]["type"] == 0:
+            return
+        token_params["access"] = {"type": 0}
+        await self.set(token_params)
+
+    async def set_master_access(self):
+        token_params = await self.get()
+        if token_params["access"]["type"] == 1:
+            return
+        token_params["access"] = {"type": 1}
+        await self.set(token_params)
+
+    async def set_acl_access(
+        self, acl: list[dict] = [], templating: bool | None = None
+    ):
+        token_params = await self.get()
+        access = {
+            "type": 2,
+            "acl": acl,
+        }
+        if templating is not None:
+            access["templating"] = templating
+        token_params["access"] = access
+        await self.set(token_params)

{flespi_sdk-0.1.1 → flespi_sdk-0.1.3}/src/flespi_sdk/modules/realms/users/user.py

@@ -23,3 +23,22 @@ class User(ItemWithMetadata):
             session=session,
             cid=cid,
         )
+
+    async def update_password(
+        self,
+        password: str,
+    ) -> None:
+        """
+        Updates the password of the user.
+
+        :param password: The new password for the user.
+        """
+        if len(password) < 16:
+            raise ValueError("Password must be at least 16 characters long")
+
+        async with self.session.put(
+            f"/platform/realms/{self.parent_id}/users/{self.id}",
+            json={"password": password},
+            headers=self.get_headers(),
+        ) as response:
+            await self.get_result(response)