图集谷-写真集-爬虫-2.1「建议收藏」

# 导入库
import os, time, requests
 
# 请求头
headers = {
  'User-Agent': 'Chrome 10.0'
}
 
# 图片数据列表
pictures = []
 
# 获取图片数据
def get_pic(url, nums):
    for num in range(0, int(nums)):
        url_all = str(url) + f'{str(num)}.jpg'
        res = requests.get(url=url_all, headers=headers)
        pic_content = res.content
        pictures.append(pic_content)
        time.sleep(0.5)
 
# 创建目的文件夹
def get_path(path_name):
    if not os.path.exists(f'C:/Users/liu/Desktop/图集/{path_name}'):
        os.mkdir(f'C:/Users/liu/Desktop/图集/{path_name}')
    else:
        pass
 
# 保存图片数据
def save_data():
    for pic,i in zip(pictures, range(int(Num))):
        picture = open(f'C:/Users/liu/Desktop/图集/{str(Page_Num)}/{str(i)}.jpg', 'wb')
        picture.write(pic)
        picture.close()
# 主函数，调用函数
def main():
    url = f'https://tjg.gzhuibei.com/a/1/{str(Page_Num)}/'    # 传入网页数据
    get_path(path_name=Page_Num)
    get_pic(url=url, nums=Num)
    save_data()
 
# 主程序入口
if __name__ == '__main__':
    Page_Num = input('Enter the Page_name:')    # Page_Num指下文url所代指的数字
    Num = int(input('Enter the Num:')) + 1    # Num为写真集的图片数
    print('Finish!')