diff --git a/app.py b/app.py index 83941d9..6026c70 100755 --- a/app.py +++ b/app.py @@ -32,6 +32,9 @@ # my profile page (r"/profile", handlers.ProfileHandler), + #follow & followers + (r"/follow",handlers.FollowHandler), + # other users profile page (r"/(?P.*)", handlers.UserHandler) ] diff --git a/handlers.py b/handlers.py index f7067c4..7bedb05 100755 --- a/handlers.py +++ b/handlers.py @@ -244,10 +244,22 @@ def get(self): who = self.get_argument("who",False) _type = self.get_argument("type","followers") if who: + _f = {} if _type == "followers": - f = self.db.users.find({"followed":who}) + _f = {"followers":[]} + f = self.db.follow.find({"followed":who},{"_id":0}) + for i in f: + _f["followers"].append(i["follower"]) + _f["count"] = len(_f["followers"]) + if _type == "following": - f = self.db.users.find({"follower":who}) + _f = {"following":[]} + f = self.db.follow.find({"follower":who},{"_id":0}) + for i in f: + _f["following"].append(i["followed"]) + _f["count"] = len(_f["following"]) + + self.write(tornado.escape.json_encode(_f)) @tornado.web.authenticated def post(self):