wizata-dsapi 2.0.0.dev10__tar.gz → 2.0.0.dev11__tar.gz

{wizata_dsapi-2.0.0.dev10/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev10
+Version: 2.0.0.dev11
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.0.0.dev10 → wizata_dsapi-2.0.0.dev11}/wizata_dsapi/__init__.py

@@ -56,4 +56,5 @@ from .edge_module import EdgeModule, EdgeModuleType, WIZATA_MODULES
 from .edge_config import EdgeConsumer, EdgeWriter, EdgeTrigger
 from .dashboard import Dashboard, DashboardType
 from .mobile_asset import MobileAssetTile, MobileAssetTileType, MobileAssetConfig
-from .graylog_log import GraylogLog
+from .graylog_log import GraylogLog
+from .user import User, UserType

wizata_dsapi-2.0.0.dev11/wizata_dsapi/user.py

@@ -0,0 +1,62 @@
+import enum
+import uuid
+
+
+class UserType(enum.Enum):
+    USER = 0
+    API_KEY = 1
+
+
+class User:
+
+    def __init__(self):
+        self.id: uuid.UUID | None = None
+        self.ad_id: uuid.UUID | None = None
+        self.first_name: str | None = None
+        self.last_name: str | None = None
+        self.email: str | None = None
+        self.phone: str | None = None
+        self.user_type: UserType | None = None
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "User":
+        obj = cls()
+        if "id" in data and data["id"] is not None:
+            obj.id = uuid.UUID(str(data["id"]))
+        if "adId" in data and data["adId"] is not None:
+            obj.ad_id = uuid.UUID(str(data["adId"]))
+        obj.first_name = data.get("firstName")
+        obj.last_name = data.get("lastName")
+        obj.email = data.get("email")
+        obj.phone = data.get("phone")
+        raw_type = data.get("userType")
+        if raw_type is not None:
+            obj.user_type = UserType(int(raw_type))
+        return obj
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.id is not None:
+            obj["id"] = str(self.id)
+        if self.ad_id is not None:
+            obj["adId"] = str(self.ad_id)
+        if self.first_name is not None:
+            obj["firstName"] = self.first_name
+        if self.last_name is not None:
+            obj["lastName"] = self.last_name
+        if self.email is not None:
+            obj["email"] = self.email
+        if self.phone is not None:
+            obj["phone"] = self.phone
+        if self.user_type is not None:
+            obj["userType"] = self.user_type.value
+        return obj
+
+    def api_id(self) -> str | None:
+        return str(self.id) if self.id else None
+
+    def api_key(self) -> str | None:
+        return str(self.ad_id) if self.ad_id else None
+
+    def set_id(self, id_value):
+        self.id = uuid.UUID(str(id_value))

wizata_dsapi-2.0.0.dev11/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.0.0.dev11"

{wizata_dsapi-2.0.0.dev10 → wizata_dsapi-2.0.0.dev11/wizata_dsapi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev10
+Version: 2.0.0.dev11
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.0.0.dev10 → wizata_dsapi-2.0.0.dev11}/wizata_dsapi.egg-info/SOURCES.txt

@@ -42,6 +42,7 @@ wizata_dsapi/template.py
 wizata_dsapi/trigger.py
 wizata_dsapi/twin.py
 wizata_dsapi/twinregistration.py
+wizata_dsapi/user.py
 wizata_dsapi/version.py
 wizata_dsapi/wizard_function.py
 wizata_dsapi/wizard_request.py

wizata_dsapi-2.0.0.dev10/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.0.0.dev10"