import os
from tkinter import *
from tkinter import messagebox
from tkinter import ttk
import threading
from doubanSpider.spider import get_subjectID as gID
from doubanSpider.spider import run as run
from doubanSpider.analysis import save_pics

def put_name_to_listb1():
    listb1.delete(0, END)
    movie = gID()
    l3.place(x=80, y=10)
    l3_var.set(f'正在搜索...')
    l3['background'] = 'yellow'
    i = 0
    for name in movie:
        listb1.insert(i,name)
        i += 1
    l3.place(x=80, y=10)
    l3_var.set(f'搜索完毕！')
    l3['background'] = 'green'

def get_content():
    # key_word = e1.get()
    select_path = c1.get()
    print(len(select_path))
    # 1.再判断是否选择了磁盘
    if len(select_path) == 1:
        # 2.判断所选路径是否存在
        if not os.path.exists(select_path):
            # 3.判断是否在列表框选择了用户名
            try:
                # 直接获取选中项目
                global movie_name_selected
                movie_name_selected = listb1.get(listb1.curselection())
                select_path = c1.get() + ':/DoubanContent/' + movie_name_selected + '/'
                l3.place(x=80, y=10)
                l3_var.set(f'正在爬取...')
                l3['background'] = 'yellow'
                run(movie_name_selected,select_path)
                l3.place(x=80, y=10)
                l3_var.set(f'爬取完毕！')
                l3['background'] = 'green'


            # 搜索后，但是没选择用户，会报TclError错误，此except就用来捕获这个异常
            except TclError:
                messagebox.showwarning(title='警告', message='请选择一个电影！')
                l3.place(x=80, y=10)
                l3_var.set(f'请选择一个电影！')
                l3['background'] = 'red'

        # 获取当前选中项目(使用索引)
        else:
            messagebox.showwarning(title='警告', message='请检查路径！')
            l3.place(x=80, y=10)
            l3_var.set(f'请检查路径！')
            l3['background'] = 'red'
    else:
        messagebox.showwarning(title='警告', message='您未选择磁盘!')
        l3.place(x=80, y=10)
        l3_var.set(f'请检查是否选择了磁盘！')
        l3['background'] = 'red'

def get_pics():
    movie_name_selected = listb1.get(listb1.curselection())
    select_path = c1.get() + ':/DoubanContent/' + movie_name_selected + '/'
    l3.place(x=80, y=10)
    l3_var.set(f'正在分析...')
    l3['background'] = 'yellow'
    save_pics(movie_name_selected, select_path)
    l3.place(x=80, y=10)
    l3_var.set(f'分析完毕！')
    l3['background'] = 'green'

def open_disk():
    disk = c1.get()
    big_dir = disk + ':/DoubanContent/'
    if len(disk) == 1:
        try:
            if not os.path.exists(big_dir):
                os.mkdir(big_dir)
            os.startfile(big_dir)
        except:
            messagebox.showwarning(title='警告', message='选中的磁盘不存在！')
            l3.place(x=80, y=10)
            l3_var.set(f'选中的磁盘不存在！')
            l3['background'] = 'red'
    else:
        messagebox.showwarning(title='警告', message='您未选中磁盘！')
        l3.place(x=80, y=10)
        l3_var.set(f'您未选中磁盘！')
        l3['background'] = 'red'

def print_path(event):
    # 要使用完整的路径
    global disk
    disk = c1.get()
    disk_path = c1.get() + ':/'
    if len(disk) == 1:
        if os.path.exists(disk_path):
            messagebox.showinfo(title='提示', message=f'文件将存储到：{disk}:/DoubanContent目录下')
        else:
            messagebox.showerror(title='错误', message='选定磁盘不存在!')
            l3.place(x=80, y=10)
            l3_var.set(f'选中的磁盘不存在！')
            l3['background'] = 'red'
    else:
        messagebox.showwarning(title='警告', message='请先选定磁盘！')
        l3.place(x=80, y=10)
        l3_var.set(f'请先选定磁盘！')
        l3['background'] = 'red'

def window_quit():
    ret = messagebox.askyesno(title='提示', message='是否要退出？')
    if ret:
        window.destroy()
        window.quit()

def escape(event):
    window_quit()


def thread_movie(func, *args):
    t = threading.Thread(target=func, args=args)
    t.setDaemon(True)
    t.start()



if __name__ == '__main__':
    window = Tk()
    width = 600
    height = 240
    screenWidth = window.winfo_screenwidth()  # 获取显示区域的宽度
    screenHeight = window.winfo_screenheight()  # 获取显示区域的高度
    left = (screenWidth - width) / 2
    top = (screenHeight - height) / 2
    window.geometry("%dx%d+%d+%d" % (width, height, left, top))
    window.resizable(0, 0)
    window.title('豆瓣影评简单分析软件')

    l1 = ttk.Label(window, text='运行状态：')
    l1.place(x=13, y=10)
    l3_var = StringVar()
    l3 = ttk.Label(window, background='yellow', textvar=l3_var)
    l3.place(x=80, y=10)
    l3_var.set('还没搜索')

    l1 = ttk.Label(window, text='本软件用于爬取豆瓣电\n影影评信息并简单分析')
    l1.place(x=60, y=60)

    l4 = ttk.Label(window, text='磁盘:')
    l4.place(x=13, y=120,)
    disk_list = ['C', 'D', 'E', 'F', 'G', 'H', 'I']
    c1 = ttk.Combobox(window, justify='center', state='readonly', width=17, value=disk_list)
    c1.bind('<<ComboboxSelected>>', print_path)
    c1.place(x=80, y=120)

    b1 = ttk.Button(window, text='搜索', command=lambda: thread_movie(put_name_to_listb1,), width=7)
    b1.place(x=100, y=160)

    l5 = ttk.Label(window, text='豆瓣新片榜电影列表:')
    l5.place(x=250, y=10)
    lb1_var = StringVar()
    listb1 = Listbox(window, justify='center', listvariable=lb1_var, width=20, height=6)
    listb1.place(x=250, y=40, height=140)

    b2 = ttk.Button(window, text='开始爬取', command=lambda: thread_movie(get_content, ), width=7)
    b2.place(x=290, y=200)

    b3 = ttk.Button(window, text='打开文件夹', width=12, command=open_disk)
    b3.place(x=450, y=120)

    b3 = ttk.Button(window, text='退出', width=7, command=window_quit)
    b3.place(x=470, y=200)

    b4 = ttk.Button(window, text='分析数据', width=10, command=lambda: thread_movie(get_pics, ))
    b4.place(x=455, y=60)

    f1 = ttk.LabelFrame(window)
    f1.place(x=0, y=190)
    l6 = ttk.Label(f1, text='感谢您的使用！', foreground='red')
    l6.pack(anchor="w", fill=X)

    # 绑定esc键---退出
    window.bind('<Escape>', escape)

    # 加入主窗口销毁事件
    window.protocol('WM_DELETE_WINDOW', window_quit)
    window.mainloop()