#--*coding:utf-8*--
from tkinter import *
from tkinter import ttk
from tkinter import messagebox
import os
import json
import threading
import requests
from PIL import Image,ImageTk

'''
抖音无水印视频下载工具
难点：1.视频接口
 2.进度条
'''
def clean_progressbar():
 # 清空进度条
 fill_line = canvas.create_rectangle(1.5, 1.5, 0, 23, width=0, fill="white")
 x = 500 # 未知变量，可更改
 n = 600 / x # 465是矩形填充满的次数
 for t in range(x):
 n = n + 600 / x
 # 以矩形的长度作为变量值更新
 canvas.coords(fill_line, (0, 0, n, 60))
 window.update()

def download():
 #先清空进度条，再下载
 clean_progressbar()
 real_link=t1.get('0.0',END).split('**')[-1]
 headers2 = {
 'sec-fetch-dest': 'document',
 'user-agent': 'Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.104 Mobile Safari/537.36'
 }
 r2 = requests.get(real_link, headers=headers2, stream=True)
 try:
 os.mkdir(video_path)
 except:
 pass
 file =video_path+ video_title + '.mp4'
 #下载与进度条
 chunk_size = 1024 # 每一块的大小，每次下载块的大小
 file_size = int(r2.headers['Content-Length']) # 提取出来的文件大小为string格式，使用int()强制转化
 fill_line = canvas.create_rectangle(1.5, 1.5, 0, 23, width=0, fill="green")
 raise_data = 600 / (file_size / chunk_size) # 增量大小，600为进度条的长度
 _size = 0 # 已经下载文件的大小
 with open(file, "wb") as f:
 n=0
 for data in r2.iter_content(chunk_size): # inter_content:用于边下载边存硬盘，每次下载chunk_size大小的块
 f.write(data)
 n = n + raise_data
 canvas.coords(fill_line, (0, 0, n, 60))
 window.update()
 #下载完以后清空
 t1.delete('0.0',END)
 t1.insert(END,'{title}.mp4\n下载完成！'.format(title=video_title))

#解析出无水印视频下载地址以及视频标题
def parse(share_link):
 # 先处理短链接获取item_ids
 headers = {
 'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
 'sec-fetch-dest': 'document',
 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.104 Safari/537.36'
 }
 # 使用allow_redirects=False禁止重定向，拿到重定向的url
 r1 = requests.get(share_link, allow_redirects=False, headers=headers)
 item_ids = re.findall(r'video/(\d+)/', r1.text)
 interface_url='https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/"%dx%d+%d+%d" % (width, height, left, top))
 window.resizable(0,0)

 window.iconbitmap('./rely/my_favicon.ico')
 photo=Image.open('./rely/dy_logo.png')
 photo=photo.resize((200,50))
 image=ImageTk.PhotoImage(photo)
 l0=ttk.Label(window,imag=image,justify='center')
 l0.pack()
 f1=ttk.Labelframe(window,text='视频链接地址：')
 f1.place(x=40,y=55)
 e1=ttk.Entry(f1,width=30)
 e1.pack()
 f2=ttk.Labelframe(window,text='信息：')
 f2.place(x=40,y=102)
 t1=Text(f2,height=6,width=30)
 t1.pack()
 t1.insert(END,'{_xing}\n将抖音分享链接地址粘贴在上面输入框中，本程序会自动解析出视频的下载地址并且显示在这里\n{_xing}'.format(_xing='*'*29))
 b1=ttk.Button(window,text='解析',command=lambda :thread_it(pre_parse))
 b1.place(x=40,y=265)
 b2=ttk.Button(window,text='下载',state='disable',command=lambda :thread_it(download))
 b2.place(x=170,y=265)
 b3=ttk.Button(window,text='打开文件夹',command=open_dir)
 b3.place(x=40,y=298)
 b4=ttk.Button(window,text='退出',command=quit_window)
 b4.place(x=170,y=298)

 f3=ttk.Labelframe(window)
 f3.place(x=40,y=329)
 l1=ttk.Label(f3,text=' 敬告：本软件仅供学习交流使用！',width=30,justify='center',foreground='red')
 l1.pack(fill=X)

 f4=ttk.LabelFrame(window,text='下载进度：')
 f4.place(x=40,y=210)
 canvas = Canvas(f4, width=210, height=20,bg="white")
 canvas.pack(side='left')

 l2_var=StringVar()
 l2=ttk.Label(f4,textvariable=l2_var,text='未下载')
 l2.pack(side='left',anchor=S)

 window.mainloop()
 '''
 test_url： https://v.douyin.com/JcxTMj2/
 '''