From de9684e8b7268412ed170ccd30eb7205ddaf6482 Mon Sep 17 00:00:00 2001
From: hzcortex <iamwanghz@gmail.com>
Date: Thu, 2 Jun 2016 16:15:38 +0800
Subject: [PATCH] Update zhihu.py

---
 zhihu.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/zhihu.py b/zhihu.py
index e00e939..8fbeb77 100755
--- a/zhihu.py
+++ b/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)