Merge lp:~inspirated/fbfriendsgraph/testing into lp:fbfriendsgraph

[Kamran Riaz Khan]

[Kamran Riaz Khan (inspirated)]

review: Approve

=== modified file 'fbfriendsgraph.py'
--- fbfriendsgraph.py	2010-02-16 15:03:58 +0000
+++ fbfriendsgraph.py	2010-02-21 16:03:17 +0000
@@ -3,6 +3,7 @@
 mutual connections among them."""
 
 import simplejson
+import time
 import thread
 import threading
 import urlparse
@@ -132,6 +133,16 @@
 				writer = PNGWriter(filename, self.graph, self.mozbrowser)
 				writer.start()
 
+	def get_connections(self, uid, friends):
+		print 'Building connections for', uid
+		ilist = [uid] * len(friends)
+		try:
+			iconnections = self.fb.friends.areFriends(ilist, friends)
+		except Exception:
+			return None
+		else:
+			return iconnections
+
 	def run(self):
 		myfriends = self.fb.friends.get()
 		myfriendsinfo = self.fb.users.getInfo(myfriends, [u'name'])
@@ -144,9 +155,14 @@
 				myfriends.remove(i)
 
 		edges = []
+		reqs = {'limit' : 10, 'time' : time.time(), 'count' : 0}
 		for (num, i) in enumerate(myfriends):
-			ilist = [i] * len(myfriends)
-			iconnections = self.fb.friends.areFriends(ilist, myfriends)
+			iconnections = None
+			while iconnections == None and not self._stopped:
+				iconnections = self.get_connections(i, myfriends)
+
+			if iconnections == None:
+				break
 
 			self.data = (prepare_html(h1 = 'Creating friends connections',
 				h2 = '%.2f%%' % (float(num) / len(myfriends) * 100),
@@ -166,6 +182,18 @@
 					edges.append((name1, name2))
 					gobject.idle_add(self.update)
 
+			curtime = time.time()
+			reqs['count'] = reqs['count'] + 1
+			if reqs['count'] >= reqs['limit'] or curtime - reqs['time'] > 60:
+				delay = 60 - (curtime - reqs['time'])
+				if delay > 0:
+					print 'Made', reqs['count'], 'requests in a minute'
+					print 'Delaying for', delay, 'seconds'
+					time.sleep(delay)
+					print 'Continuing'
+				reqs['count'] = 0
+				reqs['time'] = curtime
+
 		self.data = prepare_html(h1 = 'Friends connections parsed successfully')
 		gobject.idle_add(self.update)