wizata-dsapi 1.1.47__tar.gz → 1.2.1__tar.gz

{wizata_dsapi-1.1.47 → wizata_dsapi-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wizata_dsapi
-Version: 1.1.47
+Version: 1.2.1
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-1.1.47 → wizata_dsapi-1.2.1}/wizata_dsapi/__init__.py

@@ -45,4 +45,5 @@ from .pipeline_image import PipelineImage
 from .evaluation import Evaluation
 
 # Streamlit utils
-from .streamlit_utils import get_streamlit_token, get_streamlit_domain
+from .streamlit_utils import (get_streamlit_token, get_streamlit_domain, get_streamlit_from, get_streamlit_to,
+                              get_streamlit_twin_id)

{wizata_dsapi-1.1.47 → wizata_dsapi-1.2.1}/wizata_dsapi/datastore.py

@@ -5,7 +5,7 @@ class DataStore(ApiDto):
 
     @classmethod
     def route(cls):
-        return "datastore"
+        return "datastores"
 
     @classmethod
     def from_dict(cls, data):

wizata_dsapi-1.2.1/wizata_dsapi/streamlit_utils.py

@@ -0,0 +1,68 @@
+import sys
+import uuid
+from typing import Optional
+from uuid import UUID
+
+from .dataframe_toolkit import generate_epoch
+
+
+def get_streamlit_token():
+    if 'streamlit' in sys.modules:
+        st = sys.modules['streamlit']
+        if st.query_params is not None and "auth_token" in st.query_params:
+            auth_token = st.query_params["auth_token"]
+            return auth_token
+    return None
+
+
+def get_streamlit_domain():
+    if 'streamlit' in sys.modules:
+        st = sys.modules['streamlit']
+        if st.query_params is not None and "dsapi" in st.query_params:
+            domain = st.query_params["dsapi"]
+            return domain
+    return None
+
+def get_streamlit_twin_id() -> Optional[UUID]:
+    """
+    return current selected twin_id.
+    :return:
+    """
+    if 'streamlit' in sys.modules:
+        st = sys.modules['streamlit']
+        if st.query_params is not None and "twin_id" in st.query_params:
+            twin_id = uuid.UUID(st.query_params["twin_id"])
+            return twin_id
+    return None
+
+
+def get_streamlit_from() -> Optional[int]:
+    """
+    timestamp representation of from parameter on UI (ms)
+    :return:
+    """
+    if 'streamlit' in sys.modules:
+        st = sys.modules['streamlit']
+        if st.query_params is not None and "from" in st.query_params:
+            param = st.query_params["from"]
+            if 'now' in param:
+                return generate_epoch(param)
+            else:
+                return int(param)
+    return None
+
+
+def get_streamlit_to() -> Optional[int]:
+    """
+    timestamp representation of to parameter on UI (ms)
+    :return:
+    """
+    if 'streamlit' in sys.modules:
+        st = sys.modules['streamlit']
+        if st.query_params is not None and "to" in st.query_params:
+            param = st.query_params["to"]
+            if 'now' in param:
+                return generate_epoch(param)
+            else:
+                return int(param)
+    return None

wizata_dsapi-1.2.1/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "1.2.1"

{wizata_dsapi-1.1.47 → wizata_dsapi-1.2.1}/wizata_dsapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wizata-dsapi
-Version: 1.1.47
+Version: 1.2.1
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-1.1.47/wizata_dsapi/streamlit_utils.py

@@ -1,17 +0,0 @@
-import sys
-
-
-def get_streamlit_token():
-    if 'streamlit' in sys.modules:
-        st = sys.modules['streamlit']
-        if st.query_params is not None and "auth_token" in st.query_params:
-            auth_token = st.query_params["auth_token"]
-            return auth_token
-
-
-def get_streamlit_domain():
-    if 'streamlit' in sys.modules:
-        st = sys.modules['streamlit']
-        if st.query_params is not None and "dsapi" in st.query_params:
-            domain = st.query_params["dsapi"]
-            return domain

wizata_dsapi-1.1.47/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "1.1.47"