# 引入系統類庫用于打開關閉文件
import sys
# 使用文檔解析類庫
from bs4 import BeautifulSoup
# 使用網絡請求類庫
import requests

#圖片保存目錄
path = 'D:/陰陽師' 
#陰陽師壁紙網站
html_doc = "https://yys.163.com/media/picture.html" 

# 請求
requests_html_doc = requests.get(html_doc).text
# 正則匹配所有href地址
regex = re.compile('.*?href="(.*?)2732x2048.jpg" rel="external nofollow" ')
urls = regex.findall(requests_html_doc)

# set集合可以防止下載的圖片重復
result = set()
for i in urls:
    result.add(i)

# 計數器用于圖片命名
num = 0

# 文件路徑、操作模式、編碼  # r''
# 打開文件錄入圖片
f = open(r'result.txt', 'w', encoding='utf-8')
for a in urls:
    try:
        image_data = requests.get(a).content
        image_name = '{}.jpg'.format(num)  # 給每張圖片命名
        save_path = path + '/' + image_name  # 圖片的保存地址
        with open(save_path, 'wb') as f:
            f.write(image_data)
            print(image_name, '=======================>下載成功！！！')
            num = num+1  # 下一張圖片名字序號加一
    except:
        pass
# 關閉文件錄入
f.close()
print("\r\n掃描結果已寫入到result.txt文件中\r\n")

# 引入系統類庫
import sys
# 使用文檔解析類庫
from bs4 import BeautifulSoup
# 使用網絡請求類庫
import urllib.request
path = 'D:/陰陽師'

html_doc = "https://yys.163.com/media/picture.html"
# 獲取請求
req = urllib.request.Request(html_doc)
# 打開頁面
webpage = urllib.request.urlopen(req)

# 讀取頁面內容
html = webpage.read()
# 解析成文檔對象
soup = BeautifulSoup(html, 'html.parser')  # 文檔對象

# 非法URL 1
invalidLink1 = '#'
# 非法URL 2
invalidLink2 = 'javascript:void(0)'
# set集合可以防止下載的圖片連接重復
result = set()
# 計數器用于圖片命名
num = 0
# 查找文檔中所有a標簽
for k in soup.find_all('a'):
    # print(k)
    # 查找href標簽
    link = k.get('href')
    # 過濾沒找到的
    if(link is not None):
        # 過濾非法鏈接
        if link == invalidLink1:
            pass
        elif link == invalidLink2:
            pass
        elif link.find("javascript:") != -1:
            pass
        else:
            result.add(link)

for a in result:
    # 文件路徑、操作模式、編碼  # r''
    f = open(r'result.txt', 'w', encoding='utf-8')
    # image_data = urllib.request.get(url=a).content
    image_data = requests.get(url=a).content
    image_name = '{}.jpg'.format(num)  # 給每張圖片命名
    save_path = path + '/' + image_name  # 圖片的保存地址
    with open(save_path, 'wb') as f:
        f.write(image_data)
        print(image_name, '=======================>下載成功！！！')
        num = num+1  # 下一張圖片名字序號加一
        f.close()

print("\r\n掃描結果已寫入到result.txt文件中\r\n")