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

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

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

{wizata_dsapi-1.1.48 → 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, get_streamlit_from, get_streamlit_to
+from .streamlit_utils import (get_streamlit_token, get_streamlit_domain, get_streamlit_from, get_streamlit_to,
+                              get_streamlit_twin_id)

{wizata_dsapi-1.1.48 → wizata_dsapi-1.2.1}/wizata_dsapi/streamlit_utils.py

@@ -1,4 +1,8 @@
 import sys
+import uuid
+from typing import Optional
+from uuid import UUID
+
 from .dataframe_toolkit import generate_epoch
 
 
@@ -8,6 +12,7 @@ def get_streamlit_token():
         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():
@@ -16,9 +21,22 @@ def get_streamlit_domain():
         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() -> int:
+def get_streamlit_from() -> Optional[int]:
     """
     timestamp representation of from parameter on UI (ms)
     :return:
@@ -31,9 +49,10 @@ def get_streamlit_from() -> int:
                 return generate_epoch(param)
             else:
                 return int(param)
+    return None
 
 
-def get_streamlit_to() -> int:
+def get_streamlit_to() -> Optional[int]:
     """
     timestamp representation of to parameter on UI (ms)
     :return:
@@ -46,3 +65,4 @@ def get_streamlit_to() -> int:
                 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.48 → 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.48
+Version: 1.2.1
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-1.1.48/wizata_dsapi/version.py

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