pysmarlaapi 0.2.4__tar.gz → 0.3.0__tar.gz

{pysmarlaapi-0.2.4 → pysmarlaapi-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pysmarlaapi
-Version: 0.2.4
+Version: 0.3.0
 Summary: Swing2Sleep Smarla API
 Author-email: Robin Lintermann <robin.lintermann@explicatis.com>
 Requires-Python: >=3.11.9

{pysmarlaapi-0.2.4 → pysmarlaapi-0.3.0}/pysmarlaapi/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.2.4"
+__version__ = "0.3.0"
 
 from .classes import Connection
 from .federwiege import Federwiege

{pysmarlaapi-0.2.4 → pysmarlaapi-0.3.0}/pysmarlaapi/classes/auth_token.py

@@ -1,3 +1,4 @@
+import base64
 from dataclasses import dataclass
 from typing import Self
 
@@ -23,5 +24,13 @@ class AuthToken:
     def from_string(cls, value) -> Self:
         return AuthToken.from_json(jsonpickle.decode(value))
 
+    @classmethod
+    def from_base64(cls, value) -> Self:
+        token = base64.b64decode(value.encode()).decode()
+        return AuthToken.from_string(token)
+
     def get_string(self) -> str:
         return jsonpickle.encode(self, unpicklable=False)
+
+    def get_base64(self) -> str:
+        return base64.b64encode(self.get_string().encode()).decode()

{pysmarlaapi-0.2.4 → pysmarlaapi-0.3.0}/pysmarlaapi/classes/connection.py

@@ -6,7 +6,7 @@ from . import AuthToken
 
 class Connection:
 
-    def __init__(self, url: str, token: AuthToken = None, token_str=None, token_json=None):
+    def __init__(self, url: str, token: AuthToken = None, token_str=None, token_json=None, token_b64=None):
         self.url = url
         if token is not None:
             self.token = token
@@ -14,6 +14,8 @@ class Connection:
             self.token = AuthToken.from_json(token_json)
         elif token_str is not None:
             self.token = AuthToken.from_string(token_str)
+        elif token_b64 is not None:
+            self.token = AuthToken.from_base64(token_b64)
         else:
             self.token = None
 

{pysmarlaapi-0.2.4 → pysmarlaapi-0.3.0}/tests/test.py

@@ -10,15 +10,15 @@ logging.basicConfig(level=20)
 from pysmarlaapi import Connection, Federwiege
 
 try:
-    from config import AUTH_TOKEN_PERSONAL, HOST
+    from config import AUTH_TOKEN_B64, HOST
 except ImportError:
     print("config.py or mandatory variables missing, please add in root folder...")
-    exit()
+    sys.exit()
 
 loop = asyncio.get_event_loop()
 async_thread = threading.Thread(target=loop.run_forever)
 
-connection = Connection(url=HOST, token_json=AUTH_TOKEN_PERSONAL)
+connection = Connection(url=HOST, token_b64=AUTH_TOKEN_B64)
 
 federwiege = Federwiege(loop, connection)