Skip to content
This repository was archived by the owner on May 16, 2019. It is now read-only.

Commit 35644e8

Browse files
committed
Merge branch 'diego-G-patch-1'
2 parents 72445ef + 8e3a5b6 commit 35644e8

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

api/ws.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,9 @@ def connectionMade(self):
4141
def connectionLost(self, reason=connectionDone):
4242
self.factory.unregister(self)
4343

44-
def get_vendors(self, message_id):
44+
def get_vendors(self, message_id, quantity):
45+
if quantity == 0:
46+
return
4547
if message_id in self.factory.outstanding_vendors:
4648
queried = self.factory.outstanding_vendors[message_id]
4749
else:
@@ -81,7 +83,7 @@ def handle_response(metadata, node):
8183
self.factory.db.vendors.delete_vendor(node.id.encode("hex"))
8284
return False
8385

84-
for node in to_query[:30]:
86+
for node in to_query[:quantity]:
8587
self.factory.mserver.get_user_metadata(node).addCallback(handle_response, node)
8688

8789
def get_moderators(self, message_id):
@@ -298,8 +300,12 @@ def dataReceived(self, payload):
298300

299301
message_id = str(request_json["request"]["id"])
300302

303+
quantity = 30
304+
if request_json["request"]["quantity"]:
305+
quantity = int(request_json["request"]["quantity"])
306+
301307
if request_json["request"]["command"] == "get_vendors":
302-
self.get_vendors(message_id)
308+
self.get_vendors(message_id, quantity)
303309

304310
if request_json["request"]["command"] == "get_moderators":
305311
self.get_moderators(message_id)

0 commit comments

Comments
 (0)