Merge pull request egrcc#57 from liuwons/master

egrcc · egrcc · commit 4f37da079896 · 2016-04-25T14:05:54.000+08:00
添加获取用户头像链接的接口
diff --git a/README.rst b/README.rst
@@ -271,6 +271,8 @@ User 代表一个用户，处理用户相关操作。创建一个 User 对象需
     agree_num = user.get_agree_num()
     # 获取该用户获得的感谢数
     thanks_num = user.get_thanks_num()
+    # 获取该用户的头像url
+    head_img_url = user.get_head_img_url()
     
     # 获取该用户关注的人
     followees = user.get_followees()
@@ -291,6 +293,7 @@ User 代表一个用户，处理用户相关操作。创建一个 User 对象需
     print collections_num # 44
     print agree_num # 46387
     print thanks_num # 11477
+    print head_img_url  # https://pic2.zhimg.com/0626f4164009f291b26a79d96c6962c5_l.jpg
     
     print followees
     # <generator object get_followee at 0x7ffcac3af050>
@@ -615,6 +618,14 @@ zhihu.User ---- 知乎用户操作类
   得到该用户获得的感谢数。
   
   **Returns**： 代表感谢数的 int 型整数
+
+ **get_head_img_url** (scale)
+
+  获取用户头像url。
+
+  **Parameters**： **scale** int 型整数，代表尺寸: 1(25×25), 3(75×75), 4(100×100), 6(150×150), 10(250×250)
+
+  **Returns**： 对应尺寸头像的图片链接, 字符串
  
  **get_asks_num** ()
  
diff --git a/test.py b/test.py
@@ -121,6 +121,8 @@ def user_test(user_url):
     agree_num = user.get_agree_num()
     # 获取该用户获得的感谢数
     thanks_num = user.get_thanks_num()
+    # 获取该用户的头像url
+    head_img_url = user.get_head_img_url()
 
     # 获取该用户关注的人
     followees = user.get_followees()
@@ -142,6 +144,7 @@ def user_test(user_url):
     print collections_num  # 44
     print agree_num  # 46387
     print thanks_num  # 11477
+    print head_img_url  # https://pic2.zhimg.com/0626f4164009f291b26a79d96c6962c5_l.jpg
 
     print followees
     # <generator object get_followee at 0x7ffcac3af050>
diff --git a/zhihu.py b/zhihu.py
@@ -535,6 +535,32 @@ def get_user_id(self):
                 else:
                     return user_id
 
+    def get_head_img_url(self, scale=4):
+        """
+            By liuwons (https://github.com/liuwons)
+            增加获取知乎识用户的头像url
+            scale对应的头像尺寸:
+                1 - 25×25
+                3 - 75×75
+                4 - 100×100
+                6 - 150×150
+                10 - 250×250
+        """
+        scale_list = [1, 3, 4, 6, 10]
+        scale_name = '0s0ml0t000b'
+        if self.user_url == None:
+            print "I'm anonymous user."
+            return None
+        else:
+            if scale not in scale_list:
+                print 'Illegal scale.'
+                return None
+            if self.soup == None:
+                self.parser()
+            soup = self.soup
+            url = soup.find("img", class_="Avatar Avatar--l")["src"]
+            return url[:-5] + scale_name[scale] + url[-4:]
+
     def get_data_id(self):
         """
             By yannisxu (https://github.com/yannisxu)
