增加下载头像的功能

zhihu.py

@@ -591,7 +591,31 @@ def get_head_img_url(self, scale=4):
             soup = self.soup
             url = soup.find("img", class_="Avatar Avatar--l")["src"]
             return url[:-5] + scale_name[scale] + url[-4:]
+
+    def download_head_img(self, head_url=None):
+        url = get_head_img_url()
+        r = requests.get(url)
+        user = get_user_id()
+        image_name = user + url[-4:]
+
+        if os.path.exists(os.path.join(os.getcwd(), 'Avatar')):
+            new_dir = raw_input(u'请输入想保存的路径名称')
+            os.mkdir(new_dir)
+            os.chdir(new_dir)
+        else:
+            os.mkdir('Avatar')
+            os.chdir('Avatar')
+
+        with open(image_name, 'wb') as image:
+            image.write(r.content)
 
+        if os.path.isfile(image_name):
+            Logging.success(u'Success Download the Avatar')
+            return True
+        else:
+            Logging.error(u'Error happened')
+            return False
+
     def get_data_id(self):
         """
             By yannisxu (https://github.com/yannisxu)