vnai 2.1.7__py3-none-any.whl → 2.1.8__py3-none-any.whl

vnai/scope/state.py

@@ -1,155 +1,222 @@
-import time
-import threading
-import json
-import os
-from datetime import datetime
-from pathlib import Path
-
-class Tracker:
-    _instance = None
-    _lock = threading.Lock()
-
-    def __new__(cls):
-        with cls._lock:
-            if cls._instance is None:
-                cls._instance = super(Tracker, cls).__new__(cls)
-                cls._instance._initialize()
-            return cls._instance
-
-    def _initialize(self):
-        self.metrics = {
-"startup_time": datetime.now().isoformat(),
-"function_calls": 0,
-"api_requests": 0,
-"errors": 0,
-"warnings": 0
-        }
-        self.performance_metrics = {
-"execution_times": [],
-"last_error_time": None,
-"peak_memory": 0
-        }
-        self.privacy_level ="standard"
-        self.home_dir = Path.home()
-        self.project_dir = self.home_dir /".vnstock"
-        self.project_dir.mkdir(exist_ok=True)
-        self.data_dir = self.project_dir /'data'
-        self.data_dir.mkdir(exist_ok=True)
-        self.metrics_path = self.data_dir /"usage_metrics.json"
-        self.privacy_config_path = self.project_dir /'config' /"privacy.json"
-        os.makedirs(os.path.dirname(self.privacy_config_path), exist_ok=True)
-        self._load_metrics()
-        self._load_privacy_settings()
-        self._start_background_collector()
-
-    def _load_metrics(self):
-        if self.metrics_path.exists():
-            try:
-                with open(self.metrics_path,'r') as f:
-                    stored_metrics = json.load(f)
-                for key, value in stored_metrics.items():
-                    if key in self.metrics:
-                        self.metrics[key] = value
-            except:
-                pass
-
-    def _save_metrics(self):
-        try:
-            with open(self.metrics_path,'w') as f:
-                json.dump(self.metrics, f)
-        except:
-            pass
-
-    def _load_privacy_settings(self):
-        if self.privacy_config_path.exists():
-            try:
-                with open(self.privacy_config_path,'r') as f:
-                    settings = json.load(f)
-                    self.privacy_level = settings.get("level","standard")
-            except:
-                pass
-
-    def setup_privacy(self, level=None):
-        privacy_levels = {
-"minimal":"Essential system data only",
-"standard":"Performance metrics and errors",
-"enhanced":"Detailed operation analytics"
-        }
-        if level is None:
-            level ="standard"
-        if level not in privacy_levels:
-            raise ValueError(f"Invalid privacy level: {level}. Choose from {', '.join(privacy_levels.keys())}")
-        self.privacy_level = level
-        with open(self.privacy_config_path,"w") as f:
-            json.dump({"level": level}, f)
-        return level
-
-    def get_privacy_level(self):
-        return self.privacy_level
-
-    def _start_background_collector(self):
-        def collect_metrics():
-            while True:
-                try:
-                    import psutil
-                    current_process = psutil.Process()
-                    memory_info = current_process.memory_info()
-                    memory_usage = memory_info.rss / (1024 * 1024)
-                    if memory_usage > self.performance_metrics["peak_memory"]:
-                        self.performance_metrics["peak_memory"] = memory_usage
-                    self._save_metrics()
-                except:
-                    pass
-                time.sleep(300)
-        thread = threading.Thread(target=collect_metrics, daemon=True)
-        thread.start()
-
-    def record(self, event_type, data=None):
-        if self.privacy_level =="minimal" and event_type !="errors":
-            return True
-        if event_type in self.metrics:
-            self.metrics[event_type] += 1
-        else:
-            self.metrics[event_type] = 1
-        if event_type =="errors":
-            self.performance_metrics["last_error_time"] = datetime.now().isoformat()
-        if event_type =="function_calls" and data and"execution_time" in data:
-            self.performance_metrics["execution_times"].append(data["execution_time"])
-            if len(self.performance_metrics["execution_times"]) > 100:
-                self.performance_metrics["execution_times"] = self.performance_metrics["execution_times"][-100:]
-        if self.metrics["function_calls"] % 100 == 0 or event_type =="errors":
-            self._save_metrics()
-        return True
-
-    def get_metrics(self):
-        avg_execution_time = 0
-        if self.performance_metrics["execution_times"]:
-            avg_execution_time = sum(self.performance_metrics["execution_times"]) / len(self.performance_metrics["execution_times"])
-        output = self.metrics.copy()
-        output.update({
-"avg_execution_time": avg_execution_time,
-"peak_memory_mb": self.performance_metrics["peak_memory"],
-"uptime": (datetime.now() - datetime.fromisoformat(self.metrics["startup_time"])).total_seconds(),
-"privacy_level": self.privacy_level
-        })
-        return output
-
-    def reset(self):
-        self.metrics = {
-"startup_time": datetime.now().isoformat(),
-"function_calls": 0,
-"api_requests": 0,
-"errors": 0,
-"warnings": 0
-        }
-        self.performance_metrics = {
-"execution_times": [],
-"last_error_time": None,
-"peak_memory": 0
-        }
-        self._save_metrics()
-        return True
-tracker = Tracker()
-
-def record(event_type, data=None):
-    return tracker.record(event_type, data)
+# vnai/scope/state.py
+
+import time
+import threading
+import json
+import os
+from datetime import datetime
+from pathlib import Path
+
+class Tracker:
+    """Tracks system state and performance metrics"""
+    
+    _instance = None
+    _lock = threading.Lock()
+    
+    def __new__(cls):
+        with cls._lock:
+            if cls._instance is None:
+                cls._instance = super(Tracker, cls).__new__(cls)
+                cls._instance._initialize()
+            return cls._instance
+    
+    def _initialize(self):
+        """Initialize tracker"""
+        self.metrics = {
+            "startup_time": datetime.now().isoformat(),
+            "function_calls": 0,
+            "api_requests": 0,
+            "errors": 0,
+            "warnings": 0
+        }
+        
+        self.performance_metrics = {
+            "execution_times": [],
+            "last_error_time": None,
+            "peak_memory": 0
+        }
+        
+        self.privacy_level = "standard"
+        
+        # Setup data directory
+        self.home_dir = Path.home()
+        self.project_dir = self.home_dir / ".vnstock"
+        self.project_dir.mkdir(exist_ok=True)
+        self.data_dir = self.project_dir / 'data'
+        self.data_dir.mkdir(exist_ok=True)
+        self.metrics_path = self.data_dir / "usage_metrics.json"
+        self.privacy_config_path = self.project_dir / 'config' / "privacy.json"
+        
+        # Create config directory if it doesn't exist
+        os.makedirs(os.path.dirname(self.privacy_config_path), exist_ok=True)
+        
+        # Load existing metrics
+        self._load_metrics()
+        
+        # Load privacy settings
+        self._load_privacy_settings()
+        
+        # Start background metrics collector
+        self._start_background_collector()
+    
+    def _load_metrics(self):
+        """Load metrics from file"""
+        if self.metrics_path.exists():
+            try:
+                with open(self.metrics_path, 'r') as f:
+                    stored_metrics = json.load(f)
+                    
+                # Update metrics with stored values
+                for key, value in stored_metrics.items():
+                    if key in self.metrics:
+                        self.metrics[key] = value
+            except:
+                pass
+    
+    def _save_metrics(self):
+        """Save metrics to file"""
+        try:
+            with open(self.metrics_path, 'w') as f:
+                json.dump(self.metrics, f)
+        except:
+            pass
+    
+    def _load_privacy_settings(self):
+        """Load privacy settings"""
+        if self.privacy_config_path.exists():
+            try:
+                with open(self.privacy_config_path, 'r') as f:
+                    settings = json.load(f)
+                    self.privacy_level = settings.get("level", "standard")
+            except:
+                pass
+    
+    def setup_privacy(self, level=None):
+        """Configure privacy level for data collection"""
+        privacy_levels = {
+            "minimal": "Essential system data only",
+            "standard": "Performance metrics and errors",
+            "enhanced": "Detailed operation analytics" 
+        }
+        
+        if level is None:
+            # Default level
+            level = "standard"
+        
+        if level not in privacy_levels:
+            raise ValueError(f"Invalid privacy level: {level}. Choose from {', '.join(privacy_levels.keys())}")
+        
+        # Store preference
+        self.privacy_level = level
+        
+        # Store in configuration file
+        with open(self.privacy_config_path, "w") as f:
+            json.dump({"level": level}, f)
+        
+        return level
+    
+    def get_privacy_level(self):
+        """Get current privacy level"""
+        return self.privacy_level
+    
+    def _start_background_collector(self):
+        """Start background metrics collection"""
+        def collect_metrics():
+            while True:
+                try:
+                    import psutil
+                    
+                    # Update peak memory
+                    current_process = psutil.Process()
+                    memory_info = current_process.memory_info()
+                    memory_usage = memory_info.rss / (1024 * 1024)  # MB
+                    
+                    if memory_usage > self.performance_metrics["peak_memory"]:
+                        self.performance_metrics["peak_memory"] = memory_usage
+                        
+                    # Save metrics periodically
+                    self._save_metrics()
+                    
+                except:
+                    pass
+                    
+                time.sleep(300)  # Run every 5 minutes
+        
+        # Start thread
+        thread = threading.Thread(target=collect_metrics, daemon=True)
+        thread.start()
+    
+    def record(self, event_type, data=None):
+        """Record an event"""
+        # Check privacy level
+        if self.privacy_level == "minimal" and event_type != "errors":
+            # In minimal mode, only track errors
+            return True
+            
+        # Update counts
+        if event_type in self.metrics:
+            self.metrics[event_type] += 1
+        else:
+            self.metrics[event_type] = 1
+            
+        # Special handling for errors
+        if event_type == "errors":
+            self.performance_metrics["last_error_time"] = datetime.now().isoformat()
+            
+        # Special handling for function calls with timing data
+        if event_type == "function_calls" and data and "execution_time" in data:
+            # Keep up to 100 latest execution times
+            self.performance_metrics["execution_times"].append(data["execution_time"])
+            if len(self.performance_metrics["execution_times"]) > 100:
+                self.performance_metrics["execution_times"] = self.performance_metrics["execution_times"][-100:]
+                
+        # Save if metrics change significantly
+        if self.metrics["function_calls"] % 100 == 0 or event_type == "errors":
+            self._save_metrics()
+            
+        return True
+    
+    def get_metrics(self):
+        """Get current metrics"""
+        # Calculate derived metrics
+        avg_execution_time = 0
+        if self.performance_metrics["execution_times"]:
+            avg_execution_time = sum(self.performance_metrics["execution_times"]) / len(self.performance_metrics["execution_times"])
+            
+        # Add derived metrics to output
+        output = self.metrics.copy()
+        output.update({
+            "avg_execution_time": avg_execution_time,
+            "peak_memory_mb": self.performance_metrics["peak_memory"],
+            "uptime": (datetime.now() - datetime.fromisoformat(self.metrics["startup_time"])).total_seconds(),
+            "privacy_level": self.privacy_level
+        })
+        
+        return output
+    
+    def reset(self):
+        """Reset metrics"""
+        self.metrics = {
+            "startup_time": datetime.now().isoformat(),
+            "function_calls": 0,
+            "api_requests": 0,
+            "errors": 0,
+            "warnings": 0
+        }
+        
+        self.performance_metrics = {
+            "execution_times": [],
+            "last_error_time": None,
+            "peak_memory": 0
+        }
+        
+        self._save_metrics()
+        return True
+
+# Create singleton instance
+tracker = Tracker()
+
+
+def record(event_type, data=None):
+    """Record an event"""
+    return tracker.record(event_type, data)

{vnai-2.1.7.dist-info → vnai-2.1.8.dist-info}/METADATA

@@ -1,20 +1,20 @@
-Metadata-Version: 2.4
-Name: vnai
-Version: 2.1.7
-Summary: Vnstock Analytics Interface
-Author-email: Vnstock Analytics Team <support@vnstocks.com>
-License: proprietary
-Project-URL: Homepage, https://vnstocks.com
-Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.7
-Description-Content-Type: text/markdown
-Requires-Dist: requests>=2.25.0
-Requires-Dist: psutil>=5.8.0
-Provides-Extra: dev
-Requires-Dist: pytest>=6.0.0; extra == "dev"
-
-# VnAI
+Metadata-Version: 2.4
+Name: vnai
+Version: 2.1.8
+Summary: Vnstock Analytics Interface
+Author-email: Vnstock Analytics Team <support@vnstocks.com>
+License: proprietary
+Project-URL: Homepage, https://vnstocks.com
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.25.0
+Requires-Dist: psutil>=5.8.0
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0.0; extra == "dev"
+
+# VnAI

vnai-2.1.8.dist-info/RECORD

@@ -0,0 +1,16 @@
+vnai/__init__.py,sha256=fZQKQSpvzFX9aum8kjJyZieGQnmEzWIbzLLgpaQvVQM,7996
+vnai/beam/__init__.py,sha256=Ibfpu8ulgVw1E6ew5kaE1Z69V6mGxwYy6SRa0-DFivk,156
+vnai/beam/metrics.py,sha256=TEBgrwrJbL7D3dlBVo9qdq9qEgqQKXef_otNMXqwm6E,8606
+vnai/beam/pulse.py,sha256=SuzA_ZL_FGEHFNrLZOvXbSJSKn8pGsehNwmaADw9DyY,3427
+vnai/beam/quota.py,sha256=dTbtOP82XSzLaNksPlW-2qnrnQgyvMI8KAH3GKPPQqE,21363
+vnai/flow/__init__.py,sha256=xKOTxRtAkDekfJaQtx-zN1-5pxQdF_oAEcsK-DXGM2k,136
+vnai/flow/queue.py,sha256=KcA20u-WiEPVV-knburw86t6ynuKXsv_Kwd-gJE5zT0,4138
+vnai/flow/relay.py,sha256=Ar_GlGhUvTU1w8ByAMxKGjh6468MXTYY0haAGi4WKnE,16647
+vnai/scope/__init__.py,sha256=overJZ_UiEfBRNcSieE1GPU_9X3oS4C5l6JeBaFFVxk,267
+vnai/scope/profile.py,sha256=5DKB8_ImpAruJQ0V_jPs98PVZ1uyr0QyN3zjBbUfPSI,30706
+vnai/scope/promo.py,sha256=EjdzmWuKozSauzcZo7C4K0OAMK7YOj49L1hwOhoezqE,16841
+vnai/scope/state.py,sha256=7u-jH3TKXFo7Y5vUZUMhRM1_2YA9wXuHIat0rvhFgBc,7466
+vnai-2.1.8.dist-info/METADATA,sha256=XpGL3ogyO6rZ9sLg8xi1axISrzZ2boCQ6cPW32Ly2qI,662
+vnai-2.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vnai-2.1.8.dist-info/top_level.txt,sha256=4zI0qZHePCwvgSqXl4420sBcd0VzZn4MEcRsAIFae3k,5
+vnai-2.1.8.dist-info/RECORD,,

vnai-2.1.7.dist-info/RECORD

@@ -1,16 +0,0 @@
-vnai/__init__.py,sha256=IPw89g9Ar7TnXHXq9YVU2Sa1EyCgP-udw7Qbfpi95UU,6226
-vnai/beam/__init__.py,sha256=Of7ydx35u4gTOBCYXxDwTfDAZVyIpbds09dXX0hJh-w,133
-vnai/beam/metrics.py,sha256=p0WzO1a5MvXm98KyyshLzB4FCk-5ERh7JN65IxVB0I0,6441
-vnai/beam/pulse.py,sha256=WTdjJ8mr2fvtPEZcoyzVO9Nv8jt2QPCjOZjCh7CB4Ww,2563
-vnai/beam/quota.py,sha256=6DQdA7qwft1VWsscjpxUGyA59PEGmUY2mnGomj2aDRE,14748
-vnai/flow/__init__.py,sha256=d96OtNAJeYEYYW_3ZZXbGh8NWbVK3cd78lsub9m9--Q,82
-vnai/flow/queue.py,sha256=LBuaiAdfaOfbOL_1oxapEKlNav7qGdUk8BQdfJTxMyA,3322
-vnai/flow/relay.py,sha256=e7H_MSNm0AYwDeY0AeHCptl2vuD1NCRUvmsGWNmQK2c,13515
-vnai/scope/__init__.py,sha256=VVFWdhAVGYJY-4CykgyhjsqoL17WlbRjVk3CZYAN7S4,200
-vnai/scope/profile.py,sha256=R_xM5DyDEZ6ju9CwTAv_iXDHx4Dhqd7b1lCeAmlj-lc,22145
-vnai/scope/promo.py,sha256=lnoZJL5smBc1SF49bkCMMCrgzv5NnrIIAMbSqyTGmYI,13003
-vnai/scope/state.py,sha256=Y8eZNVtG31E7Jv-RtHdDLkRRKbfCK8czbbyKp65065k,5543
-vnai-2.1.7.dist-info/METADATA,sha256=Gw9R_X7XNv055XKhXZeHKOhstdtfBT88greichRFK9w,682
-vnai-2.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vnai-2.1.7.dist-info/top_level.txt,sha256=4zI0qZHePCwvgSqXl4420sBcd0VzZn4MEcRsAIFae3k,5
-vnai-2.1.7.dist-info/RECORD,,