code-loader 1.0.115__py3-none-any.whl → 1.0.117__py3-none-any.whl

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/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/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.dist-info → code_loader-1.0.117.dist-info}/METADATA

@@ -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.dist-info → code_loader-1.0.117.dist-info}/RECORD

@@ -1,5 +1,4 @@
-LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader/__init__.py,sha256=6MMWr0ObOU7hkqQKgOqp4Zp3I28L7joGC9iCbQYtAJg,241
+code_loader/__init__.py,sha256=outxRQ0M-zMfV0QGVJmAed5qWfRmyD0TV6-goEGAzBw,406
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/contract/datasetclasses.py,sha256=gJsXu4zVAaiBlq6GJwPxfTD2e0gICTtI_6Ir61MRL48,8838
 code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
@@ -22,15 +21,16 @@ code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaS
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=0iHVHxC2NjfH7F0vQFVGy1e0llgKEyUHUHh3DdtqL70,32602
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=wjtk3TflrjJ8Y-OeuedVBD-09ZuOjIKGUjL7sMBU0fQ,41017
-code_loader/leaploader.py,sha256=fxf2CrWPeWfJYVoNg-Lue7RSvosuZO35FjRrDfrny5I,29363
+code_loader/leaploader.py,sha256=rQRK1lyUPSpZiRs6lKUUxVJBCe3grEY_UxEMxSIpuxI,29709
 code_loader/leaploaderbase.py,sha256=lKdw2pd6H9hFsxVmc7jJMoZd_vlG5He1ooqT-cR_yq8,4496
+code_loader/mixpanel_tracker.py,sha256=U20vQXH8G7XIVXxcpQcVEZSuIwwGnyH5RMHXWZZG8HI,4639
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/plot_functions/plot_functions.py,sha256=xg6Gi4myTN9crq6JtyrhYI38HLXjPVJcbnI7CIy8f7w,14625
 code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6LznoQIvi4vpo,657
 code_loader/utils.py,sha256=_j8b60pimoNAvWMRj7hEkkT6C76qES6cZoBFHpXHMxA,2698
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.115.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.115.dist-info/METADATA,sha256=gG-J8pjjV6uGDL4pQ9eBNJ8qPISDUDxMNAaH2Mx4kN4,1048
-code_loader-1.0.115.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.115.dist-info/RECORD,,
+code_loader-1.0.117.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.117.dist-info/METADATA,sha256=du8-T5bFdFlkZj7rC-z9ZczxH_fCPrKjQCAjfKs3QGU,1102
+code_loader-1.0.117.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+code_loader-1.0.117.dist-info/RECORD,,

{code_loader-1.0.115.dist-info → code_loader-1.0.117.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any

code_loader-1.0.115.dist-info/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2021 TensorLeap
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.