1 files changed, 38 insertions, 0 deletions
diff --git a/pyload/network/Bucket.py b/pyload/network/Bucket.py
new file mode 100644
index 000000000..40d8c8071
--- /dev/null
+++ b/pyload/network/Bucket.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from time import time
+
+# 10kb minimum rate
+MIN_RATE = 10240
+
+class Bucket:
+    def __init__(self):
+        self.rate = 0 # bytes per second, maximum targeted throughput
+        self.tokens = 0
+        self.timestamp = time()
+
+    def __nonzero__(self):
+        return False if self.rate < MIN_RATE else True
+
+    def setRate(self, rate):
+        self.rate = int(rate)
+
+    def consumed(self, amount):
+        """ return the time the process has to sleep, after it consumed a specified amount """
+        if self.rate < MIN_RATE: return 0 #May become unresponsive otherwise
+
+        self.calc_tokens()
+        self.tokens -= amount
+
+        if self.tokens < 0:
+            return -self.tokens/float(self.rate)
+        else:
+            return 0
+
+    def calc_tokens(self):
+        if self.tokens < self.rate:
+            now = time()
+            delta = self.rate * (now - self.timestamp)
+            self.tokens = min(self.rate, self.tokens + delta)
+            self.timestamp = now
+