def get_friends(user_id):
"""Return the list of Facebook friends of user `user_id` who are also
using the service.
The friends list is a list of native user IDs (not Facebook user IDs),
each an object of type ObjectId.
A caching layer sits before the database.
"""
result = FriendsCache.get(user_id)
if result is None:
user = yield sundowner.data.users.read(user_id)
try:
fb_friends = user["facebook"]["friends"]["data"]
except KeyError:
# I don't think it's possible that this path in the user's data
# wouldn't exist but if it doesn't, log it so that it can be
# investigated
print "Facebook user with no/unexpected friends data"
fb_friends = []
fb_user_ids = map(itemgetter("id"), fb_friends)
result = yield sundowner.data.users.\
read_native_user_ids_from_facebook_user_ids(fb_user_ids)
FriendsCache.put(user_id, result)
raise tornado.gen.Return(result)
def get_friends(user_id):
"""Return the list of Facebook friends of user `user_id` who are also
using the service.
The friends list is a list of native user IDs (not Facebook user IDs),
each an object of type ObjectId.
A caching layer sits before the database.
"""
result = FriendsCache.get(user_id)
if result is None:
user = yield sundowner.data.users.read(user_id)
try:
fb_friends = user["facebook"]["friends"]["data"]
except KeyError:
# I don't think it's possible that this path in the user's data
# wouldn't exist but if it doesn't, log it so that it can be
# investigated
print "Facebook user with no/unexpected friends data"
fb_friends = []
fb_user_ids = map(itemgetter("id"), fb_friends)
result = yield sundowner.data.users.\
read_native_user_ids_from_facebook_user_ids(fb_user_ids)
FriendsCache.put(user_id, result)
raise tornado.gen.Return(result)
def test_get_tags(self):
# SETUP ----------------------------------------------------------------
# create mock user docs in the database
@tornado.gen.coroutine
def setup_users():
result = []
for _ in range(self.NUM_USERS):
username = self.rand_noun()
user_id = yield sundowner.data.users.create({
"facebook": {
"id": None,
"name": username,
}})
result.append(user_id)
raise tornado.gen.Return(result)
users = IOLoop.instance().run_sync(setup_users)
# one of the users will act as the user issuing the request and another
# subset will act as friends of this user
friends = random.sample(users, self.NUM_FRIENDS+1)
user_id = friends.pop()
# add friends to cache
FriendsCache.put(user_id, friends)
# create mock tags
@tornado.gen.coroutine
def create_tag():
# pick a random lng/lat inside the query radius
# it appears that a lng/lat > (QUERY_RADIUS_RADIANS + .006) will
# fall outside the query radius
qrr = sundowner.data.content._QUERY_RADIUS_RADIANS
lng = random.uniform(self.QUERY_LNG-qrr, self.QUERY_LNG+qrr)
lat = random.uniform(self.QUERY_LAT-qrr, self.QUERY_LAT+qrr)
user_id = random.choice(users)
text = self.rand_noun()
content_id = yield sundowner.data.content.put(
user_id= user_id,
text= text,
url= None,
accuracy= 0,
lng= lng,
lat= lat)
raise tornado.gen.Return(content_id)
for _ in range(self.NUM_TAGS):
IOLoop.instance().run_sync(create_tag)
# TEST -----------------------------------------------------------------
def test():
@tornado.gen.coroutine
def query_content():
result = yield ContentModel.get_nearby(
self.QUERY_LNG, self.QUERY_LAT, user_id)
raise tornado.gen.Return(result)
return IOLoop.instance().run_sync(query_content)
"""
def test_old():
@tornado.gen.coroutine
def query_content():
result = yield ContentModel.get_nearby_old(
self.QUERY_LNG, self.QUERY_LAT)
raise tornado.gen.Return(result)
result = IOLoop.instance().run_sync(query_content)
"""
#print "OLD", timeit.timeit(test_old, number=self.NUM_TRIALS)
#print "NEW", timeit.timeit(test, number=self.NUM_TRIALS)
pprint.pprint(test())
raise Exception
