code-loader 1.0.115__tar.gz → 1.0.117__tar.gz

{code_loader-1.0.115 → code_loader-1.0.117}/PKG-INFO

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.115
+Version: 1.0.117
 Summary: 
-Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -14,11 +13,13 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: mixpanel (>=4.10.0,<5.0.0)
 Requires-Dist: numpy (>=1.22.3,<2.0.0) ; python_version >= "3.8" and python_version < "3.11"
 Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_version < "3.13"
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
+Project-URL: Homepage, https://github.com/tensorleap/code-loader
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.115 → code_loader-1.0.117}/code_loader/__init__.py

@@ -2,3 +2,9 @@ from code_loader.leaploader import LeapLoader
 from code_loader.inner_leap_binder import global_leap_binder as leap_binder
 from code_loader.experiment_api.experiment import init_experiment
 from code_loader.experiment_api.client import Client
+
+try:
+    from code_loader.mixpanel_tracker import track_code_loader_loaded
+    track_code_loader_loaded({'event_type': 'module_import'})
+except Exception:
+    pass

{code_loader-1.0.115 → code_loader-1.0.117}/code_loader/leaploader.py

@@ -34,6 +34,16 @@ class LeapLoader(LeapLoaderBase):
         super().__init__(code_path, code_entry_name)
 
         self._preprocess_result_cached = None
+        
+        try:
+            from code_loader.mixpanel_tracker import track_code_loader_loaded
+            track_code_loader_loaded({
+                'event_type': 'leap_loader_instantiated',
+                'code_path': code_path,
+                'code_entry_name': code_entry_name
+            })
+        except Exception:
+            pass
 
     @lru_cache()
     def exec_script(self) -> None:

code_loader-1.0.117/code_loader/mixpanel_tracker.py

@@ -0,0 +1,134 @@
+"""
+Mixpanel tracking utilities for code-loader.
+"""
+import os
+import sys
+import getpass
+import uuid
+from typing import Optional, Dict, Any
+import mixpanel  # type: ignore[import]
+
+
+class MixpanelTracker:
+    """Handles Mixpanel event tracking for code-loader."""
+    
+    def __init__(self, token: str = "f1bf46fb339d8c2930cde8c1acf65491"):
+        self.token = token
+        self.mp = mixpanel.Mixpanel(token)
+        self._user_id: Optional[str] = None
+    
+    def _get_whoami(self) -> str:
+        """Get the current system username (whoami) for device identification.
+        
+        Returns:
+            str: The system username, with fallbacks to environment variables or 'unknown'
+        """
+        if self._user_id is None:
+            try:
+                self._user_id = getpass.getuser()
+            except Exception:
+                # Fallback to environment variables or default
+                self._user_id = os.environ.get('USER', os.environ.get('USERNAME', 'unknown'))
+        return self._user_id or 'unknown'
+    
+       
+    def _get_tensorleap_user_id(self) -> Optional[str]:
+        """Get the TensorLeap user ID from ~/.tensorleap/user_id if it exists."""
+        try:
+            user_id_path = os.path.expanduser("~/.tensorleap/user_id")
+            if os.path.exists(user_id_path):
+                with open(user_id_path, 'r') as f:
+                    user_id = f.read().strip()
+                    if user_id:
+                        return user_id
+        except Exception:
+            pass
+        return None
+    
+    def _get_or_create_device_id(self) -> str:
+        """Get or create a device ID from ~/.tensorleap/device_id file.
+        
+        If the file doesn't exist, creates it with a new UUID.
+        
+        Returns:
+            str: The device ID (UUID string)
+        """
+        try:
+            device_id_path = os.path.expanduser("~/.tensorleap/device_id")
+            
+            # Create directory if it doesn't exist
+            os.makedirs(os.path.dirname(device_id_path), exist_ok=True)
+            
+            if os.path.exists(device_id_path):
+                with open(device_id_path, 'r') as f:
+                    device_id = f.read().strip()
+                    if device_id:
+                        return device_id
+            
+            # Generate new device ID and save it
+            device_id = str(uuid.uuid4())
+            with open(device_id_path, 'w') as f:
+                f.write(device_id)
+            
+            return device_id
+        except Exception:
+            # Fallback to generating a new UUID if file operations fail
+            return str(uuid.uuid4())
+    
+    def _get_distinct_id(self) -> str:
+        """Get the distinct ID for Mixpanel tracking.
+        
+        Priority order:
+        1. TensorLeap user ID (from ~/.tensorleap/user_id)
+        2. Device ID (from ~/.tensorleap/device_id, generated if not exists)
+        """
+        tensorleap_user_id = self._get_tensorleap_user_id()
+        if tensorleap_user_id:
+            return tensorleap_user_id
+        
+        return self._get_or_create_device_id()
+    
+    def track_code_loader_loaded(self, event_properties: Optional[Dict[str, Any]] = None) -> None:
+        """Track code loader loaded event with device identification.
+        
+        Args:
+            event_properties: Optional additional properties to include in the event
+        """
+        try:
+            distinct_id = self._get_distinct_id()
+            
+            tensorleap_user_id = self._get_tensorleap_user_id()
+            whoami = self._get_whoami()
+            device_id = self._get_or_create_device_id()
+            
+            properties = {
+                'whoami': whoami,
+                '$device_id': device_id,  # Always use device_id for $device_id
+                'python_version': f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}",
+                'platform': os.name,
+            }
+            
+            if tensorleap_user_id:
+                properties['user_id'] = tensorleap_user_id
+            
+            if event_properties:
+                properties.update(event_properties)
+            
+            self.mp.track(distinct_id, 'code_loader_loaded', properties)
+        except Exception as e:
+            pass
+
+
+# Global tracker instance
+_tracker = None
+
+
+def get_tracker() -> MixpanelTracker:
+    global _tracker
+    if _tracker is None:
+        _tracker = MixpanelTracker()
+    return _tracker
+
+
+def track_code_loader_loaded(event_properties: Optional[Dict[str, Any]] = None) -> None:
+    get_tracker().track_code_loader_loaded(event_properties)

{code_loader-1.0.115 → code_loader-1.0.117}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.115"
+version = "1.0.117"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"
@@ -20,6 +20,7 @@ numpy = [
 psutil = "^5.9.5"
 requests = "^2.32.3"
 pyyaml = "^6.0.2"
+mixpanel = "^4.10.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.1"