# @Author : 王同學
import requests
from bs4 import BeautifulSoup
import os
import tkinter as tk    # GUI
import concurrent.futures
import threading



def get_content(url):
    headers = {'user-agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36'}
    response = requests.get(url,headers)
    response.encoding = response.apparent_encoding  # 自動轉碼
    if response.status_code == 200:
        return response.text


def get_data(response):
    soup = BeautifulSoup(response,'lxml')
    all_li = soup.find(class_="list").find('ul')
    for i in all_li.find_all('li'):
        if i.find('b') is not None:
            title = i.find('b').text
        else:
            title = 'NOT'
        images = i.find('a').find('img').get('src')
        save_images(title,images)



def save_csv():
    pass

def save_images(title,images):
    if not os.path.exists('img'):   # 創(chuàng)建文件夾
        os.mkdir('img')
    headers = {'user-agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36'}
    images_data = requests.get(url=images,headers=headers).content
    with open('img\\' + title + '.jpg', mode='wb')as f:
        f.write(images_data)
        print('正在保存===>: ',title)

    # GUI文本框輸入
    qq.insert(tk.INSERT,"正在保存圖片:" + title + '\n')
    qq.yview_moveto(1)
    qq.update()



def main():
    print('===================已經點擊按鈕===========================')
    for i in range(2,11):
        url = f'http://www.netbian.com/index_{i}.htm'  # 循環(huán)
        qq.insert(tk.INSERT,f'==========================正在保存第{i}頁的圖片=========================='+ '\n')
        qq.update()
        print(f'============================正在保存第{i}頁的數據內容========================')
        response = get_content(url)
        get_data(response)
    qq.insert(tk.INSERT,'=================================保存結束================================')



# 多線程 防止GUI卡死
def process_it():
    it = threading.Thread(target=main)
    it.setDaemon(True)
    it.start()




if __name__ == '__main__':
    # 設置GUI圖形界面
    windoms = tk.Tk()

    windoms.iconbitmap()

    windoms.title('圖片')
    windoms.geometry('500x500+650+300')

    # labal
    text = tk.Label(windoms,text='圖片小程序',font=('華文新魏',20))
    text.place(x=170,y=10)

    # 按鈕
    button = tk.Button(windoms,text='開始下載',font=(20),width=30,height=4,fg='Violet',bd=8,command=process_it)   # 開始下載
    button.place(x=140,y=80)
    button1 = tk.Button(windoms,text='退出',font=(20),width=30,fg='Violet',height=4,bd=8,command=windoms.quit)  # 退出
    button1.place(x=140, y=180)
    # 文本框
    qq = tk.Text(windoms,state='normal',bg='light cyan',fg='DeepPink')
    qq.place(rely=0.6,relheight=0.4)
    # 顯示窗口
    tk.mainloop()