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)