tikdown.py

#!/usr/local/bin/python3

import os
import re
import json
import tkinter
import requests
import _thread as thread
from tenacity import *

download_dir = "~/Downloads/tiktok"  # 配置下载目录

download_dir = os.path.expanduser(download_dir)  # 展开成真实路径
if not os.path.exists(download_dir):
    os.makedirs(download_dir)

download_headers = {
    "user-agent": "Mozilla/5.0 (Linux; Android 8.0; Pixel 2 Build/OPD3.170816.012) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Mobile Safari/537.36 Edg/87.0.664.66"
}
tiktok_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",
    "authority": "www.tiktok.com",
    "Accept-Encoding": "gzip, deflate",
    "Connection": "keep-alive",
    "Host": "www.tiktok.com",
    "User-Agent": "Mozilla/5.0  (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) coc_coc_browser/86.0.170 Chrome/80.0.3987.170 Safari/537.36",
}
download_list = []
downloading = False


@retry(stop=stop_after_attempt(3), wait=wait_random(min=1, max=2))
def get_tiktok_info(video_id):
    try:
        tiktok_api_link = "https://api.tiktokv.com/aweme/v1/aweme/detail/?aweme_id={}".format(video_id)  # 从TikTok官方API获取部分视频数据
        log("正在请求API链接:", tiktok_api_link)
        response = requests.get(url=tiktok_api_link, headers=download_headers).text
        result = json.loads(response)
        return {
            "success": True,
            "video_title": result["aweme_detail"]["desc"],  # 视频标题,
            "video_url": result["aweme_detail"]["video"]["play_addr"]["url_list"][0],  # 无水印视频链接
            "video_author": result["aweme_detail"]["author"]["nickname"],  # 视频作者
            "video_cover": result["aweme_detail"]["video"]["cover"]["url_list"][0],  # 封面图
        }
    except Exception as e:
        log("网络请求有错误")
        return {"success": False}


def downloader(video_id):
    global downloading

    downloading = True  # 开始下载
    info = get_tiktok_info(video_id)
    if info["success"]:
        video_title = re.sub(r"[\/\\\:\*\?\"\<\>\|]", "", info["video_title"]).replace("\n", "")  # 删除title中的非法字符
        filename = info["video_author"] + " - " + video_title + ".mp4"
        try:
            target_file = os.path.join(download_dir, filename)
            if os.path.exists(target_file):
                log("视频文件已存在: ", target_file)
            else:
                log("开始下载视频: ", filename)
                response = requests.get(info["video_url"], headers=download_headers)
                with open(target_file, "wb") as f:
                    f.write(response.content)
                    f.close()
                    log("视频下载成功")
        except Exception as e:
            log("视频下载失败.")
    downloading = False  # 开始下载
    manage_download_list()  # 继续进行下载列表管理


def manage_download_list():
    if not downloading and len(download_list) > 0:
        thread.start_new_thread(downloader, (download_list.pop(),))  # 没有在下载则起动一个下载线程


def check_clipboard():
    parsed = re.findall("(https?://www.tiktok.com/[^/]+/video/(\d+))", win.clipboard_get())
    if len(parsed) == 0:
        log("剪贴板内无Tiktok分享链接")
    else:
        download_list.insert(0, parsed[0][1])
        log("任务已创建")
    manage_download_list()  # 进行下载列表管理


def log(*args):
    print(*args)  # 暂时把log先输出在console


win = tkinter.Tk()
win.title("Tiktok 下载器")
win.geometry("200x90")
win.wm_attributes("-topmost", "true")
button = tkinter.Button(win, text="点击下载剪贴板链接", bg="#7CCD7C", width=20, height=5, command=check_clipboard).pack()
win.mainloop()