import requests, os
path_a = os.path.abspath('.')
kw = ''
while True:
   kw = input('请输入你要获取的图片(若想结束请输入1)')
   if kw == '1':
       print('已退出，你下载的图片已保存在'+path_a+',请查看！')
       break
   for x in range(0, 1000, 20):
       url = 'https://www.toutiao.com/search_content/?offset='+str(x)+'&format=json&keyword=%s&autoload=true&count=20&cur_tab=3&from=gallery' % kw
       response = requests.get(url)
       data = response.json()['data']
       if not data:
           print('下载'+kw+'图片完毕，请换个关键词继续')
           break
       n = 1  # 记录文章数
       for atlas in data:
           # 创建目录
           title = atlas['title']
           print(atlas)
           try:
               if title not in os.listdir('.'):  # 防止文件名已经存在
                   os.mkdir(title)
           except OSError as e:
               print('文件名出错，创建目录失败，重新创建一个随机名字')
               title = kw + '文件名出错'+str(x)
               if title not in os.listdir('.'):
                   os.mkdir(title)
           k = 1  # 记录下载的图片数
           path = os.path.join(path_a, title)
           # 转进图片目录
           os.chdir(path)
           for image in atlas['image_list']:  # 这个链接获取的图片是小张的，看着不够爽，所以下面替换成大的图片
               image_url = image['url'].replace('list', 'large')  # 改个链接获取大的图片
               atlas = requests.get('http:'+image_url).content
               with open(str(k)+'.jpg', 'wb') as f:  # 把图片写入文件内
                   f.write(atlas)
               print('下载完第%d个文章的%d幅图完成' % (x+n, k))
               k += 1
           n += 1
           # 转出图片目录
           os.chdir(path_a)