Update ws.py

Improving get vendors with an argument to get the desired quantity.

Author: diego-G

api/ws.py

@@ -41,7 +41,7 @@ def connectionMade(self):
     def connectionLost(self, reason=connectionDone):
         self.factory.unregister(self)
 
-    def get_vendors(self, message_id):
+    def get_vendors(self, message_id, quantity):
         if message_id in self.factory.outstanding_vendors:
             queried = self.factory.outstanding_vendors[message_id]
         else:
@@ -80,8 +80,13 @@ def handle_response(metadata, node):
                     del self.factory.mserver.protocol.multiplexer.vendors[node.id]
                 self.factory.db.vendors.delete_vendor(node.id.encode("hex"))
                 return False
+              
+        if quantity == 0:
+            pass;
+        else:
+            to_query = to_query[:quantity]
 
-        for node in to_query[:30]:
+        for node in to_query:
             self.factory.mserver.get_user_metadata(node).addCallback(handle_response, node)
 
     def get_moderators(self, message_id):
@@ -298,8 +303,12 @@ def dataReceived(self, payload):
 
             message_id = str(request_json["request"]["id"])
 
+            quantity = 30;
+            if request_json["request"]["quantity"]:
+                quantity = int(request_json["request"]["quantity"])
+
             if request_json["request"]["command"] == "get_vendors":
-                self.get_vendors(message_id)
+                self.get_vendors(message_id, quantity)
 
             if request_json["request"]["command"] == "get_moderators":
                 self.get_moderators(message_id)