Handle 404 for profile image properly

FBTimNim · May 6, 2017 · af27d54 · af27d54
1 parent fcf3d90
commit af27d54
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/restapi/utils/circlise.py b/restapi/utils/circlise.py
@@ -10,10 +10,10 @@
 def getProfilePic(userID):
     """Return compressed version of a user's profile pic."""
     # Get profile image from Facebook.
-    response = request.urlopen('http://graph.facebook.com/' +
-                               str(userID) + '/picture?type=square')
-
-    if response.code != 200:
+    try:
+        response = request.urlopen('http://graph.facebook.com/' +
+                                   str(userID) + '/picture?type=square')
+    except Exception as e:
         return ""
 
     # Make the image a circle.