flespi-sdk 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

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/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/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/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/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.2.dist-info → flespi_sdk-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flespi-sdk
-Version: 0.1.2
+Version: 0.1.3
 Summary: Add your description here
 Author-email: Sergey Shevchik <sergey.shevchik@gmail.com>
 Requires-Python: >=3.11

{flespi_sdk-0.1.2.dist-info → flespi_sdk-0.1.3.dist-info}/RECORD

@@ -1,17 +1,20 @@
 flespi_sdk/__init__.py,sha256=INx-TmKYpKi9SC0-UMqtqKwnMnHK88oXyvh3tTFY668,1462
 flespi_sdk/account.py,sha256=tu4JX4Qh8Sf25BbX4RMvXjWs5DJg2W_LeasaBSYzAzo,2774
 flespi_sdk/cli.py,sha256=6OcRQAu-LLFTYGGY_S0SNTXgHlazr3NaMDqPQ7QLg_E,1346
+flespi_sdk/exception.py,sha256=cffPxxMtUCyPO121XBzSzQ2yxUlJ75e18YaD3ikBZlg,356
 flespi_sdk/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flespi_sdk/modules/flespi_session.py,sha256=TiEUVZMo-JpU7vPqnMhJJX47-kI6UMrDAlptSSpRzTo,682
+flespi_sdk/modules/flespi_session.py,sha256=TdYYfLaXpU4eeKODNCtUYvm0-1Ka5V32MRHc7bojUqo,799
 flespi_sdk/modules/item_with_metadata.py,sha256=_J1dvU_aFg6yTeiKFuQPNMFEZnFkP5Li6NePXpyqRzY,906
 flespi_sdk/modules/items_container.py,sha256=5c8ov35NQ8dVmDGpq68ROaVIj8VtySaE0Rlrk0JUsLc,1440
 flespi_sdk/modules/metadata/__init__.py,sha256=3-FNhSeWFQIlUZH8jnCAVbiJvQI7emAz3I8LRB92VPU,2792
 flespi_sdk/modules/mqtt/__init__.py,sha256=nLUQxrNpJ2E8l-OGyOAE-dfyOpQ4_-bjC-fkUEj_8As,2714
 flespi_sdk/modules/realms/__init__.py,sha256=U3nT0Zwz7Y6knmEsL3jDqxwRntM3GgmMqJUERCJJsTs,1063
-flespi_sdk/modules/realms/realm.py,sha256=TxYZpUJfKY3DnlUXzbu2OaY4nlfdA_PkVcW8N5tXNQc,633
+flespi_sdk/modules/realms/realm.py,sha256=cXhEnfzSiV1ZMJWTYqG90Whon0-qUYmZ_rEf_xPc7Mw,954
+flespi_sdk/modules/realms/realm_home.py,sha256=WICjpFfNqW35iZYzLeWgPgg39ZdBwlI2SUClBWvgxEk,1488
+flespi_sdk/modules/realms/realm_token_params.py,sha256=dCQUAdDtzKoYbvSAatJsAdxomM267sZYC2BuE_O90w8,2835
 flespi_sdk/modules/realms/users/__init__.py,sha256=hHv5R_eySJ_ioQIfYjYHbnbcMm1NhSKBKBpsNILHfbE,1798
 flespi_sdk/modules/realms/users/user.py,sha256=Y6Z3WZuzyl9-IOvKl4_OkBXBT2nIiq8L_dWF6MhD-IY,1249
 flespi_sdk/modules/subaccounts/__init__.py,sha256=PhS91e9FoV3m6VkfrANoq0_Zf-v139VsShGBXoPaJ-I,1044
-flespi_sdk-0.1.2.dist-info/METADATA,sha256=Ij9JZs0jpNBmxRDfLLKwCrLJJOraIX_dgLxUZo23t1c,2267
-flespi_sdk-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flespi_sdk-0.1.2.dist-info/RECORD,,
+flespi_sdk-0.1.3.dist-info/METADATA,sha256=7chNQ96JQ9T-WiFPWa9QJ5VmU1A_JCX1IGxamif0H1M,2267
+flespi_sdk-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flespi_sdk-0.1.3.dist-info/RECORD,,