upgini 1.1.204__tar.gz → 1.1.205__tar.gz

{upgini-1.1.204/src/upgini.egg-info → upgini-1.1.205}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.204
+Version: 1.1.205
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers

{upgini-1.1.204 → upgini-1.1.205}/setup.py

@@ -40,7 +40,7 @@ def send_log(msg: str):
 
 
 here = Path(__file__).parent.resolve()
-version = "1.1.204"
+version = "1.1.205"
 try:
     send_log(f"Start setup PyLib version {version}")
     setup(

{upgini-1.1.204 → upgini-1.1.205}/src/upgini/http.py

@@ -886,15 +886,15 @@ class BackendLogHandler(logging.Handler):
         super().__init__(*args, **kwargs)
         self.rest_client = rest_client
         self.track_metrics = None
-        self.hostname = client_ip
+        self.hostname = "0.0.0.0"
+        self.client_ip = client_ip
 
     def emit(self, record: logging.LogRecord) -> None:
         def task():
             try:
                 if self.track_metrics is None or len(self.track_metrics) == 0:
-                    self.track_metrics = get_track_metrics()
-                    if self.hostname is None:
-                        self.hostname = self.track_metrics.get("ip") or "0.0.0.0"
+                    self.track_metrics = get_track_metrics(self.client_ip)
+                    self.hostname = self.track_metrics.get("ip") or "0.0.0.0"
                 text = self.format(record)
                 tags = self.track_metrics
                 tags["version"] = __version__

{upgini-1.1.204 → upgini-1.1.205}/src/upgini/utils/display_utils.py

@@ -264,9 +264,6 @@ def prepare_and_show_report(
 def show_button_download_pdf(source: str, title="\U0001F4CA Download PDF report"):
     from IPython.display import HTML, display
 
-    with open("/Users/nikolaytoroptsev/Downloads/pdf.html", "w") as f:
-        f.write(source)
-
     file_name = f"upgini-report-{uuid.uuid4()}.pdf"
     with open(file_name, "wb") as output:
         pisa.CreatePDF(src=BytesIO(source.encode("UTF-8")), dest=output)

{upgini-1.1.204 → upgini-1.1.205}/src/upgini/utils/track_info.py

@@ -5,6 +5,7 @@ import socket
 import sys
 from getpass import getuser
 from hashlib import sha256
+from typing import Optional
 from uuid import getnode
 
 from requests import get, post
@@ -46,7 +47,7 @@ def _get_execution_ide() -> str:
 
 
 @lru_cache()
-def get_track_metrics() -> dict:
+def get_track_metrics(client_ip: Optional[str]) -> dict:
     # default values
     track = {"ide": _get_execution_ide()}
     ident_res = "https://api.ipify.org"
@@ -93,28 +94,34 @@ def get_track_metrics() -> dict:
         except Exception as e:
             track["err"] = str(e)
             track["visitorId"] = "None"
-        try:
-            from google.colab import output  # type: ignore
-            from IPython.display import Javascript, display
-
-            display(
-                Javascript(
-                    f"""
-                        window.clientIP = fetch("{ident_res}")
-                            .then(response => response.text())
-                            .then(data => data);
-                    """
+        if client_ip:
+            track["ip"] = client_ip
+        else:
+            try:
+                from google.colab import output  # type: ignore
+                from IPython.display import Javascript, display
+
+                display(
+                    Javascript(
+                        f"""
+                            window.clientIP = fetch("{ident_res}")
+                                .then(response => response.text())
+                                .then(data => data);
+                        """
+                    )
                 )
-            )
-            track["ip"] = output.eval_js("window.clientIP", timeout_sec=10)
-        except Exception as e:
-            track["err"] = str(e)
-            track["ip"] = "0.0.0.0"
+                track["ip"] = output.eval_js("window.clientIP", timeout_sec=10)
+            except Exception as e:
+                track["err"] = str(e)
+                track["ip"] = "0.0.0.0"
 
     elif track["ide"] == "binder":
         try:
             if "CLIENT_IP" in os.environ.keys():
-                track["ip"] = os.environ["CLIENT_IP"]
+                if client_ip:
+                    track["ip"] = client_ip
+                else:
+                    track["ip"] = os.environ["CLIENT_IP"]
                 track["visitorId"] = sha256(os.environ["CLIENT_IP"].encode()).hexdigest()
         except Exception as e:
             track["err"] = str(e)
@@ -142,7 +149,10 @@ def get_track_metrics() -> dict:
             track["visitorId"] = "None"
     else:
         try:
-            track["ip"] = get(ident_res, timeout=10).text
+            if client_ip:
+                track["ip"] = client_ip
+            else:
+                track["ip"] = get(ident_res, timeout=10).text
             track["visitorId"] = sha256(str(getnode()).encode()).hexdigest()
         except Exception as e:
             track["err"] = str(e)

{upgini-1.1.204 → upgini-1.1.205/src/upgini.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.204
+Version: 1.1.205
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers