adapto 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

adapto/predictor.py

@@ -1,24 +1,24 @@
-import numpy as np
-from sklearn.linear_model import LinearRegression
-
-
-class Predictor:
-    def __init__(self):
-        pass
-
-    def train_predictor(self, history):
-        if len(history) < 2:
-            return history[-1] if history else 0
-
-        X = np.arange(len(history)).reshape(-1, 1)
-        y = np.array(history)
-        model = LinearRegression().fit(X, y)
-        return model.predict([[len(history)]])[0]
-
-    def predict(self, cpu_history, memory_history, network_sent_history, network_recv_history):
-        return {
-            "predicted_cpu": self.train_predictor(cpu_history),
-            "predicted_memory": self.train_predictor(memory_history),
-            "predicted_network_sent": self.train_predictor(network_sent_history),
-            "predicted_network_recv": self.train_predictor(network_recv_history)
-        }
+import numpy as np
+from sklearn.linear_model import LinearRegression
+
+
+class Predictor:
+    def __init__(self):
+        pass
+
+    def train_predictor(self, history):
+        if len(history) < 2:
+            return history[-1] if history else 0
+
+        X = np.arange(len(history)).reshape(-1, 1)
+        y = np.array(history)
+        model = LinearRegression().fit(X, y)
+        return model.predict([[len(history)]])[0]
+
+    def predict(self, cpu_history, memory_history, network_sent_history, network_recv_history):
+        return {
+            "predicted_cpu": self.train_predictor(cpu_history),
+            "predicted_memory": self.train_predictor(memory_history),
+            "predicted_network_sent": self.train_predictor(network_sent_history),
+            "predicted_network_recv": self.train_predictor(network_recv_history)
+        }

adapto/scaler.py

@@ -1,112 +1,128 @@
-import psutil
-import time
-import logging
-from collections import deque
-from adapto.predictor import Predictor
-
-# Set up logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-
-
-class AutoScaler:
-    def __init__(self, scale_up_threshold=80, scale_down_threshold=30, memory_threshold=75,
-                 bandwidth_threshold=100000000, min_instances=1, max_instances=10, history_size=10):
-        self.scale_up_threshold = scale_up_threshold
-        self.scale_down_threshold = scale_down_threshold
-        self.memory_threshold = memory_threshold
-        self.bandwidth_threshold = bandwidth_threshold  # in bytes per second
-        self.current_instances = min_instances
-        self.min_instances = min_instances
-        self.max_instances = max_instances
-        self.previous_network = psutil.net_io_counters()
-
-        # Data history for prediction
-        self.cpu_history = deque(maxlen=history_size)
-        self.memory_history = deque(maxlen=history_size)
-        self.network_sent_history = deque(maxlen=history_size)
-        self.network_recv_history = deque(maxlen=history_size)
-
-        self.predictor = Predictor()
-
-    def get_system_metrics(self):
-        cpu_usage = psutil.cpu_percent(interval=1)
-        memory_usage = psutil.virtual_memory().percent
-        load_avg = psutil.getloadavg()[0] if hasattr(psutil, "getloadavg") else 0  # Unix-only
-        disk_usage = psutil.disk_usage('/').percent
-        network_metrics = self.get_network_metrics()
-
-        # Store metrics in history
-        self.cpu_history.append(cpu_usage)
-        self.memory_history.append(memory_usage)
-        self.network_sent_history.append(network_metrics['network_sent'])
-        self.network_recv_history.append(network_metrics['network_recv'])
-
-        return {
-            "cpu_usage": cpu_usage,
-            "memory_usage": memory_usage,
-            "load_avg": load_avg,
-            "disk_usage": disk_usage,
-            **network_metrics
-        }
-
-    def get_network_metrics(self):
-        network_io = psutil.net_io_counters()
-        network_sent = network_io.bytes_sent - self.previous_network.bytes_sent
-        network_recv = network_io.bytes_recv - self.previous_network.bytes_recv
-        self.previous_network = network_io
-
-        return {
-            "network_sent": network_sent,
-            "network_recv": network_recv
-        }
-
-    def predict_future_usage(self):
-        return self.predictor.predict(
-            list(self.cpu_history),
-            list(self.memory_history),
-            list(self.network_sent_history),
-            list(self.network_recv_history)
-        )
-
-    def scale_up(self):
-        if self.current_instances < self.max_instances:
-            self.current_instances += 1
-            logging.info(f"Scaling up: New instance count = {self.current_instances}")
-        else:
-            logging.info("Max instances reached. Cannot scale up further.")
-
-    def scale_down(self):
-        if self.current_instances > self.min_instances:
-            self.current_instances -= 1
-            logging.info(f"Scaling down: New instance count = {self.current_instances}")
-        else:
-            logging.info("Min instances reached. Cannot scale down further.")
-
-    def monitor(self, interval=5):
-        while True:
-            metrics = self.get_system_metrics()
-            predictions = self.predict_future_usage()
-
-            logging.info(
-                f"CPU: {metrics['cpu_usage']}% | Memory: {metrics['memory_usage']}% | Load Avg: {metrics['load_avg']} | Disk: {metrics['disk_usage']}% | Network Sent: {metrics['network_sent']} bytes/s | Network Recv: {metrics['network_recv']} bytes/s")
-            logging.info(
-                f"Predicted CPU: {predictions['predicted_cpu']}% | Predicted Memory: {predictions['predicted_memory']}% | Predicted Network Sent: {predictions['predicted_network_sent']} bytes/s | Predicted Network Recv: {predictions['predicted_network_recv']} bytes/s")
-
-            if (predictions['predicted_cpu'] > self.scale_up_threshold or
-                    predictions['predicted_memory'] > self.memory_threshold or
-                    predictions['predicted_network_sent'] > self.bandwidth_threshold or
-                    predictions['predicted_network_recv'] > self.bandwidth_threshold):
-                self.scale_up()
-            elif (predictions['predicted_cpu'] < self.scale_down_threshold and
-                  predictions['predicted_memory'] < self.memory_threshold / 2 and
-                  predictions['predicted_network_sent'] < self.bandwidth_threshold / 2 and
-                  predictions['predicted_network_recv'] < self.bandwidth_threshold / 2):
-                self.scale_down()
-
-            time.sleep(interval)
-
-
-# Example usage
-if __name__ == "__main__":
-    scaler = AutoScaler()
-    scaler.monitor()
+import psutil
+import time
+import logging
+from collections import deque
+from adapto.predictor import Predictor
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+
+class AutoScaler:
+    def __init__(self, scale_up_threshold=80, scale_down_threshold=30, memory_threshold=75,
+                 bandwidth_threshold=100000000, min_instances=1, max_instances=10, history_size=10, custom_scaling=None):
+        self.scale_up_threshold = scale_up_threshold
+        self.scale_down_threshold = scale_down_threshold
+        self.memory_threshold = memory_threshold
+        self.bandwidth_threshold = bandwidth_threshold  # in bytes per second
+        self.current_instances = min_instances
+        self.min_instances = min_instances
+        self.max_instances = max_instances
+        self.previous_network = psutil.net_io_counters()
+
+        # Optional custom scaling function; should accept (metrics, predictions) and return 'scale_up', 'scale_down', or 'no_change'
+        self.custom_scaling = custom_scaling
+
+        # Data history for prediction
+        self.cpu_history = deque(maxlen=history_size)
+        self.memory_history = deque(maxlen=history_size)
+        self.network_sent_history = deque(maxlen=history_size)
+        self.network_recv_history = deque(maxlen=history_size)
+
+        self.predictor = Predictor()
+
+    def get_system_metrics(self):
+        cpu_usage = psutil.cpu_percent(interval=1)
+        memory_usage = psutil.virtual_memory().percent
+        load_avg = psutil.getloadavg()[0] if hasattr(psutil, "getloadavg") else 0  # Unix-only
+        disk_usage = psutil.disk_usage('/').percent
+        network_metrics = self.get_network_metrics()
+
+        # Store metrics in history
+        self.cpu_history.append(cpu_usage)
+        self.memory_history.append(memory_usage)
+        self.network_sent_history.append(network_metrics['network_sent'])
+        self.network_recv_history.append(network_metrics['network_recv'])
+
+        return {
+            "cpu_usage": cpu_usage,
+            "memory_usage": memory_usage,
+            "load_avg": load_avg,
+            "disk_usage": disk_usage,
+            **network_metrics
+        }
+
+    def get_network_metrics(self):
+        network_io = psutil.net_io_counters()
+        network_sent = network_io.bytes_sent - self.previous_network.bytes_sent
+        network_recv = network_io.bytes_recv - self.previous_network.bytes_recv
+        self.previous_network = network_io
+
+        return {
+            "network_sent": network_sent,
+            "network_recv": network_recv
+        }
+
+    def predict_future_usage(self):
+        return self.predictor.predict(
+            list(self.cpu_history),
+            list(self.memory_history),
+            list(self.network_sent_history),
+            list(self.network_recv_history)
+        )
+
+    def scale_up(self):
+        if self.current_instances < self.max_instances:
+            self.current_instances += 1
+            logging.info(f"Scaling up: New instance count = {self.current_instances}")
+        else:
+            logging.info("Max instances reached. Cannot scale up further.")
+
+    def scale_down(self):
+        if self.current_instances > self.min_instances:
+            self.current_instances -= 1
+            logging.info(f"Scaling down: New instance count = {self.current_instances}")
+        else:
+            logging.info("Min instances reached. Cannot scale down further.")
+
+    def monitor(self, interval=5):
+        while True:
+            metrics = self.get_system_metrics()
+            predictions = self.predict_future_usage()
+
+            logging.info(
+                f"CPU: {metrics['cpu_usage']}% | Memory: {metrics['memory_usage']}% | Load Avg: {metrics['load_avg']} | Disk: {metrics['disk_usage']}% | "
+                f"Network Sent: {metrics['network_sent']} bytes/s | Network Recv: {metrics['network_recv']} bytes/s"
+            )
+            logging.info(
+                f"Predicted CPU: {predictions['predicted_cpu']}% | Predicted Memory: {predictions['predicted_memory']}% | "
+                f"Predicted Network Sent: {predictions['predicted_network_sent']} bytes/s | Predicted Network Recv: {predictions['predicted_network_recv']} bytes/s"
+            )
+
+            # If a custom scaling strategy is provided, use it to decide the action
+            if self.custom_scaling:
+                action = self.custom_scaling(metrics, predictions)
+                if action == 'scale_up':
+                    self.scale_up()
+                elif action == 'scale_down':
+                    self.scale_down()
+            else:
+                # Default scaling strategy
+                if (predictions['predicted_cpu'] > self.scale_up_threshold or
+                    predictions['predicted_memory'] > self.memory_threshold or
+                    predictions['predicted_network_sent'] > self.bandwidth_threshold or
+                    predictions['predicted_network_recv'] > self.bandwidth_threshold):
+                    self.scale_up()
+                elif (predictions['predicted_cpu'] < self.scale_down_threshold and
+                      predictions['predicted_memory'] < self.memory_threshold / 2 and
+                      predictions['predicted_network_sent'] < self.bandwidth_threshold / 2 and
+                      predictions['predicted_network_recv'] < self.bandwidth_threshold / 2):
+                    self.scale_down()
+
+            time.sleep(interval)
+
+
+# Example usage
+if __name__ == "__main__":
+    scaler = AutoScaler()
+    scaler.monitor()

adapto/utils.py

@@ -1,5 +1,5 @@
-import time
-
-def log_event(message):
-    """Log an event with a timestamp."""
+import time
+
+def log_event(message):
+    """Log an event with a timestamp."""
     print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] {message}")

{adapto-0.1.0.dist-info → adapto-0.1.2.dist-info}/LICENSE

@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) 2025 Harshal Mehta
-
-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.
+MIT License
+
+Copyright (c) 2025 Harshal Mehta
+
+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.

{adapto-0.1.0.dist-info → adapto-0.1.2.dist-info}/METADATA

@@ -1,19 +1,21 @@
-Metadata-Version: 2.1
-Name: adapto
-Version: 0.1.0
-Summary: AI-driven auto-scaling library for dynamic resource allocation.
-Home-page: https://github.com/hrshlmeht/adapto
-Author: Harshal Mehta
-Author-email: harshalmehta1998@gmail.com
-License: UNKNOWN
-Platform: UNKNOWN
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
-License-File: LICENSE
-Requires-Dist: psutil
-Requires-Dist: numpy
-
-UNKNOWN
-
+Metadata-Version: 2.2
+Name: adapto
+Version: 0.1.2
+Summary: AI-driven auto-scaling library for dynamic resource allocation.
+Home-page: https://github.com/hrshlmeht/adapto
+Author: Harshal Mehta
+Author-email: harshalmehta1998@gmail.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+License-File: LICENSE
+Requires-Dist: psutil
+Requires-Dist: numpy
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

adapto-0.1.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+adapto/__init__.py,sha256=8SWysOKRVNsD3cOXLOfIdVZx6MxzIqGKEh9Z26QuCyM,30
+adapto/predictor.py,sha256=ze-cMHdf290o3_9D-PQ7yQ8eMz1Utv3E3qTHRUFQiAI,842
+adapto/scaler.py,sha256=qVwnjd6YondkJ_vorinmReXoO0TNf0-CiKUMvDgHYrw,5544
+adapto/utils.py,sha256=6TgDvowlXL__v-OkLSTZF12ZOEqEKCLuNv1LmL70eAY,140
+adapto-0.1.2.dist-info/LICENSE,sha256=B73NNEiUy-CCs4zrOPlOWR39PfoG4zKtK97OnNctjvg,1070
+adapto-0.1.2.dist-info/METADATA,sha256=c42fZm-Mw_GOCTHNpZLqQpT6jv8DaNaIxSwrvqWnUV8,609
+adapto-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+adapto-0.1.2.dist-info/top_level.txt,sha256=IYsgAr6fnEC1R1ztYmK2ZpFNNGV0qRaf3Tvf-R8j0cM,7
+adapto-0.1.2.dist-info/RECORD,,

{adapto-0.1.0.dist-info → adapto-0.1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

adapto-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-adapto/__init__.py,sha256=8SWysOKRVNsD3cOXLOfIdVZx6MxzIqGKEh9Z26QuCyM,30
-adapto/predictor.py,sha256=RqPbSWwppyI1PQGzGZMihpX5YQoyWvN_G-ocHOa2H0g,866
-adapto/scaler.py,sha256=P1f8TDc4mrFpqXffs2FImg9mRRpP0GoYks_3JNyz4O4,4955
-adapto/utils.py,sha256=8elBvOHytKcqT9TnVYcDU8yN1SibprjhNKrnSPgBR9E,144
-adapto-0.1.0.dist-info/LICENSE,sha256=SAzpYNgJ4rxFNBFW55imF0XI2fjW7O7yXqajoK1SBmw,1091
-adapto-0.1.0.dist-info/METADATA,sha256=i2ikB2E6_fVLZgGUmwgC-LCkxr24pEHr2v2lR55vv-c,531
-adapto-0.1.0.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-adapto-0.1.0.dist-info/top_level.txt,sha256=IYsgAr6fnEC1R1ztYmK2ZpFNNGV0qRaf3Tvf-R8j0cM,7
-adapto-0.1.0.dist-info/RECORD,,